Golang Elasticsearch客户端操作：elastic库使用指南

elastic是一个强大的Go语言Elasticsearch客户端库，提供了丰富的API来与Elasticsearch交互。下面我将详细介绍其使用方法。

1. 安装与初始化

首先安装elastic库：

go get github.com/olivere/elastic/v7

初始化客户端：

package main

import (
	"context"
	"fmt"
	"log"

	"github.com/olivere/elastic/v7"
)

func main() {
	// 创建客户端
	client, err := elastic.NewClient(
		elastic.SetURL("http://localhost:9200"), // 设置ES地址
		elastic.SetSniff(false),                 // 禁用嗅探器（适用于Docker环境）
	)
	if err != nil {
		log.Fatal(err)
	}
	defer client.Stop()

	// 检查ES是否运行
	info, code, err := client.Ping("http://localhost:9200").Do(context.Background())
	if err != nil {
		log.Fatal(err)
	}
	fmt.Printf("Elasticsearch returned with code %d and version %s\n", code, info.Version.Number)
}

2. 索引操作

创建索引

// 创建索引
createIndex, err := client.CreateIndex("my_index").Do(context.Background())
if err != nil {
    log.Fatal(err)
}
if !createIndex.Acknowledged {
    fmt.Println("创建索引未被确认")
}

检查索引是否存在

exists, err := client.IndexExists("my_index").Do(context.Background())
if err != nil {
    log.Fatal(err)
}
fmt.Printf("索引存在: %v\n", exists)

删除索引

deleteIndex, err := client.DeleteIndex("my_index").Do(context.Background())
if err != nil {
    log.Fatal(err)
}
if !deleteIndex.Acknowledged {
    fmt.Println("删除索引未被确认")
}

3. 文档操作

添加文档

type User struct {
    Name string `json:"name"`
    Age  int    `json:"age"`
}

// 添加文档
user := User{Name: "张三", Age: 30}
put, err := client.Index().
    Index("my_index").
    Id("1").          // 设置文档ID
    BodyJson(user).
    Do(context.Background())
if err != nil {
    log.Fatal(err)
}
fmt.Printf("文档ID %s 添加到索引 %s\n", put.Id, put.Index)

获取文档

// 获取文档
get, err := client.Get().
    Index("my_index").
    Id("1").
    Do(context.Background())
if err != nil {
    log.Fatal(err)
}
if get.Found {
    var user User
    err := json.Unmarshal(get.Source, &user)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("获取到文档: %+v\n", user)
}

更新文档

// 更新文档
update, err := client.Update().
    Index("my_index").
    Id("1").
    Doc(map[string]interface{}{"age": 31}).
    Do(context.Background())
if err != nil {
    log.Fatal(err)
}
fmt.Printf("文档版本变为 %d\n", update.Version)

删除文档

// 删除文档
delete, err := client.Delete().
    Index("my_index").
    Id("1").
    Do(context.Background())
if err != nil {
    log.Fatal(err)
}
fmt.Printf("文档删除: %s\n", delete.Result)

4. 搜索操作

简单搜索

// 简单搜索
termQuery := elastic.NewTermQuery("name", "张三")
searchResult, err := client.Search().
    Index("my_index").
    Query(termQuery).
    From(0).Size(10).
    Pretty(true).
    Do(context.Background())
if err != nil {
    log.Fatal(err)
}

fmt.Printf("查询耗时 %d 毫秒, 结果总数: %d\n", 
    searchResult.TookInMillis, 
    searchResult.TotalHits())

for _, hit := range searchResult.Hits.Hits {
    var user User
    err := json.Unmarshal(hit.Source, &user)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("用户 %s 年龄 %d\n", user.Name, user.Age)
}

组合查询

// 组合查询
boolQuery := elastic.NewBoolQuery()
boolQuery.Must(elastic.NewMatchQuery("name", "张三"))
boolQuery.Filter(elastic.NewRangeQuery("age").Gt(25))

searchResult, err := client.Search().
    Index("my_index").
    Query(boolQuery).
    Sort("age", true). // 按年龄升序
    From(0).Size(10).
    Do(context.Background())
// 处理结果同上

聚合查询

// 聚合查询
aggs := elastic.NewTermsAggregation().Field("age")

searchResult, err := client.Search().
    Index("my_index").
    Query(elastic.NewMatchAllQuery()).
    Aggregation("ages", aggs).
    Size(0). // 不返回具体文档
    Do(context.Background())

if agg, found := searchResult.Aggregations.Terms("ages"); found {
    for _, bucket := range agg.Buckets {
        fmt.Printf("年龄 %s 有 %d 人\n", bucket.Key, bucket.DocCount)
    }
}

5. 批量操作

// 批量操作
bulkRequest := client.Bulk()

users := []User{
    {Name: "李四", Age: 25},
    {Name: "王五", Age: 28},
    {Name: "赵六", Age: 32},
}

for i, user := range users {
    req := elastic.NewBulkIndexRequest().
        Index("my_index").
        Id(fmt.Sprintf("%d", i+2)).
        Doc(user)
    bulkRequest = bulkRequest.Add(req)
}

bulkResponse, err := bulkRequest.Do(context.Background())
if err != nil {
    log.Fatal(err)
}
fmt.Printf("批量操作耗时 %d 毫秒\n", bulkResponse.Took)

6. 使用技巧与注意事项

1. 连接池管理：elastic客户端内置连接池，建议复用客户端实例

2. 上下文控制：所有操作都需要传入context，便于超时和取消控制

3. 错误处理：Elasticsearch可能返回HTTP错误，需要检查err和响应状态

4. 版本兼容：确保客户端版本与Elasticsearch服务器版本兼容

5. 性能优化：

   • 批量操作代替单条操作
   • 合理设置分页大小
   • 使用过滤器(filter)代替查询(query)提高性能

elastic库功能强大，以上只是基本用法。实际使用时，建议参考官方文档了解更多高级特性。