Golang操作elasticsearch(一)

使用第三方包:olivere github。总结一下olivere操作ES的常用功能
说明:以下例子用到的 es address: "http://127.0.0.1:9200"

1. 获取Es客户端

    func GetEsClient() *elastic.Client {
    file := "./eslog.log"
    logFile, _ := os.OpenFile(file, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0766) // 应该判断error,此处简略
    client,err:=elastic.NewClient(
    elastic.SetURL("http://127.0.0.1:9200/"),
    //docker
    elastic.SetSniff(false),
    elastic.SetInfoLog(log.New(logFile, "ES-INFO: ", 0)),
    elastic.SetTraceLog(log.New(logFile, "ES-TRACE: ", 0)),
    elastic.SetErrorLog(log.New(logFile, "ES-ERROR: ", 0)),
    )
    if err!=nil{
    return nil
    }
    return client
    }
   

2. 查看某文档是否存在,给定文档ID查询

    func IsDocExists(id int,index string) bool {
    client := GetEsClient()
    defer client.Stop()
    exist,_ := client.Exists().Index(index).Id(strconv.Itoa(id)).Do(context.Background())
    if !exist{
    log.Println("ID may be incorrect! ",id)
    return false
    }
    return true
    }
   

3. 获取文档

    func GetDoc(id int,index string)( *elastic.GetResult ,error){
    client := GetEsClient()
    defer client.Stop()
    if !IsDocExists(id,index){
    return nil,fmt.Errorf("id不存在")
    }
    esResponse,err := client.Get().Index(index).Id( strconv.Itoa(id) ).Do(context.Background())
    if err != nil {
    return nil,err
    }
    return esResponse,nil
    }
   

4. 添加文档

    func AddDoc(id int,doc string,index string)( *elastic.IndexResponse,error){
    client := GetEsClient()
    defer client.Stop()
    if IsDocExists(id,index){
    return nil,fmt.Errorf("id不存在")
    }
    rsp ,err:=client.Index().Index(index).Id( strconv.Itoa(id) ).BodyJson(doc).Do(context.Background())
    if err!=nil{
    return nil,err
    }
    return rsp,nil
    }
   

5. 更新文档

    func UpdateDoc(updateField *map[string]interface{},id int,index string)(*elastic.UpdateResponse,error){
    client := GetEsClient()
    defer client.Stop()
    if !IsDocExists(id,index){
    return nil,fmt.Errorf("id不存在")
    }
    rsp,err:=client.Update().Index(index).Id(strconv.Itoa(id)).Doc(updateField).Do(context.Background())
    if err != nil{
    fmt.Println(err)
    return nil,err
    }
    return rsp,nil
    }
   

6. 删除文档

    func DeleteDoc(id int,index string)(*elastic.DeleteResponse, error){
    client := GetEsClient()
    defer client.Stop()
    rsp,err:=client.Delete().Index(index).Id(strconv.Itoa(id)).Do(context.Background())
    if err != nil{
    return nil,err
    }
    return rsp,nil
    }

本文来自:简书

感谢作者:洛杉矶银河

查看原文:Golang操作elasticsearch(一)

• 编辑
• 预览