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
    }

有疑问加站长微信联系（非本文作者）