golang elasticsearch Client的使用详解

elasticsearch 的client ，通过 NewClient 建立连接，通过 NewClient 中的 Set.URL设置访问的地址，SetSniff设置集群

获得连接后，通过 Index 方法插入数据，插入后可以通过 Get 方法获得数据（最后的测试用例中会使用 elasticsearch client 的Get 方法）

func Save(item interface{}) {
    client, err := elastic.NewClient(
  elastic.SetURL("http://192.168.174.128:9200/"),
  // Must turn off sniff in docker
  elastic.SetSniff(false),
 )
 
 if err != nil {
  panic(err)
 }
 
 resp, err := client.Index().
  Index("dating_profile").
  Type("zhenai").
  BodyJson(item).
  Do(context.Background()) //contex需要context 包
 if err != nil {
  panic(err)
 }
 
 fmt.Printf("%+v", resp)
 
}

测试程序，自行定义一个数据结构 Profile 进行测试

func TestSave(t *testing.T) {
 profile := model.Profile{
  Age:        34,
  Height:     162,
  Weight:     57,
  Income:     "3001-5000元",
  Gender:     "女",
  Name:       "安静的雪",
  XingZuo:    "牡羊座",
  Occupation: "人事/行政",
  Marriage:   "离异",
  House:      "已购房",
  Hukou:      "山东菏泽",
  Education:  "大学本科",
  Car:        "未购车",
 }
 
 Save(profile)
}

go test 成功

通过 Get 方法查看数据是否存在elasticsearch 中

我们在test中panic，在函数中讲错误返回。在从elastisearch中取出存入的数据，与我们定义的数据进行比较，

所以save中需要将插入数据的Id返回出来

func Save(item interface{}) (id string, err error) {
 client, err := elastic.NewClient(
  elastic.SetURL("http://192.168.174.128:9200/"),
  // Must turn off sniff in docker
  elastic.SetSniff(false),
 )
 
 if err != nil {
  return "", err
 }
 
 resp, err := client.Index().
  Index("dating_profile").
  Type("zhenai").
  BodyJson(item).
  Do(context.Background())
 if err != nil {
  return "", err
 }
 
 return resp.Id, nil
 
}

测试用例

package persist
 
import (
 "context"
 "encoding/json"
 "my_crawler_single/model"
 "testing"
 
 elastic "gopkg.in/olivere/elastic.v5"
)
 
func TestSave(t *testing.T) {
 expected := model.Profile{
  Age:        34,
  Height:     162,
  Weight:     57,
  Income:     "3001-5000元",
  Gender:     "女",
  Name:       "安静的雪",
  XingZuo:    "牡羊座",
  Occupation: "人事/行政",
  Marriage:   "离异",
  House:      "已购房",
  Hukou:      "山东菏泽",
  Education:  "大学本科",
  Car:        "未购车",
 }
 
 id, err := Save(expected)
 if err != nil {
  panic(err)
 }
 
 client, err := elastic.NewClient(
  elastic.SetURL("http://192.168.174.128:9200/"),
  elastic.SetSniff(false),
 )
 if err != nil {
  panic(err)
 }
 
 resp, err := client.Get().
  Index("dating_profile").
  Type("zhenai").
  Id(id). //查找指定id的那一条数据
  Do(context.Background())
 if err != nil {
  panic(err)
 }
 
 t.Logf("%+v", resp)
 //从打印得知，数据在resp.Source中，从rest client的截图也可以知道
 
 var actual model.Profile
 //查看 *resp.Source 可知其数据类型为[]byte
 err = json.Unmarshal(*resp.Source, &actual)
 if err != nil {
  panic(err)
 }
 
 if actual != expected {
  t.Errorf("got %v;expected %v", actual, expected)
 }
}

补充：go-elasticsearch: Elastic官方的Go语言客户端

说明

Elastic官方鼓励在项目中尝试用这个包，但请记住以下几点：

这个项目的工作还在进行中，并非所有计划的功能和Elasticsearch官方客户端中的标准（故障重试，节点自动发现等）都实现了。

API稳定性无法保证。尽管公共API的设计非常谨慎，但它们可以根据进一步的探索和用户反馈以不兼容的方式进行更改。

客户端的目标是Elasticsearch 7.x版本。后续将添加对6.x和5.x版本API的支持。

安装

用go get安装这个包：

go get -u github.com/elastic/go-elasticsearch

或者将这个包添加到go.mod文件：

require github.com/elastic/go-elasticsearch v0.0.0

或者克隆这个仓库：

git clone https://github.com/elastic/go-elasticsearch.git \u0026amp;\u0026amp; cd go-elasticsearch

一个完整的示例：

mkdir my-elasticsearch-app \u0026amp;\u0026amp; cd my-e0026quot;],        e[\u0026quot;error\u0026quot;].(map[string]interface{})[\u0026quot;reason\u0026quot;],      )    }  }  if err := json.NewDecoder(res.Body).Decode(\u0026amp;r); err != nil {    log.Fatalf(\u0026quot;Error parsing the response body: %s\u0026quot;, err)  }  // Print the response status, number of results, and request duration.  log.Printf(    \u0026quot;[%s] %d hits; took: %dms\u0026quot;,    res.Status(),    int(r[\u0026quot;hits\u0026quot;].(map[string]interface{})[\u0026quot;total\u0026quot;].(map[string]interface{})[\u0026quot;value\u0026quot;].(float64)),    int(r[\u0026quot;took\u0026quot;].(float64)),  )  // Print the ID and document source for each hit.  for _, hit := range r[\u0026quot;hits\u0026quot;].(map[string]interface{})[\u0026quot;hits\u0026quot;].([]interface{}) {    log.Printf(\u0026quot; * ID=%s, %s\u0026quot;, hit.(map[string]interface{})[\u0026quot;_id\u0026quot;], hit.(map[string]interface{})[\u0026quot;_source\u0026quot;])  }  log.Println(strings.Repeat(\u0026quot;=\u0026quot;, 37))}// ~~~~~~~\u0026gt; Elasticsearch 7.0.0-SNAPSHOT// [200 OK] updated; version=1// [200 OK] updated; version=1// -------------------------------------// [200 OK] 2 hits; took: 7ms//  * ID=1, map[title:Test One]//  * ID=2, map[title:Test Two]// =====================================

如上述示例所示，esapi包允许通过两种不同的方式调用Elasticsearch API：通过创建结构（如IndexRequest），并向其传递上下文和客户端来调用其Do（）方法，或者通过客户端上可用的函数（如WithIndex（））直接调用其上的Search（）函数。更多信息请参阅包文档。

estransport包处理与Elasticsearch之间的数据传输。目前，这个实现只占据很小的空间：它只在已配置的集群端点上进行循环。后续将添加更多功能：重试失败的请求，忽略某些状态代码，自动发现群集中的节点等等。

Examples

_examples文件夹包含许多全面的示例，可帮助你上手使用客户端，包括客户端的配置和自定义，模拟单元测试的传输，将客户端嵌入自定义类型，构建查询，执行请求和解析回应。

许可证

遵循Apache License 2.0版本。

以上为个人经验，希望能给大家一个参考，也希望大家多多支持社区。如有错误或未考虑完全的地方，望不吝赐教。