下载依赖：
go get -u github.com/olivere/elastic/v7 复制代码
网络异常，图片无法展示
|
Init.go
package main import ( "context" "fmt" "github.com/olivere/elastic/v7" "log" ) var ctx = context.Background() var Url = "http://127.0.0.1:9200" var esClient *elastic.Client var index = "student" //结构体 type Student struct { Id int `json:"id"` Name string `json:"name"` Age int `json:"age"` Address string `json:"address"` } // 定义一些变量，mapping为定制的index字段类型 // number_of_replicas备份数 ， number_of_shards分片数 const mapping = ` { "settings":{ "number_of_shards": 1, "number_of_replicas": 0 }, "mappings":{ "properties":{ "name":{ "type":"keyword" }, "address":{ "type":"text" }, "age":{ "type":"long" }, "id":{ "type":"long" } } } }` // 初始化es连接 func init() { client, err := elastic.NewClient( elastic.SetURL(Url), ) if err != nil { log.Fatal("es 连接失败:", err) } // ping通服务端，并获得服务端的es版本,本实例的es版本为version 7.16.2 info, code, err := client.Ping(Url).Do(ctx) if err != nil { panic(err) } fmt.Println("Elasticsearch call code:", code, " version:", info.Version.Number) esClient = client // fmt.Println("es连接成功") } 复制代码
Add.go
package main import ( "fmt" "log" ) //添加数据 func AddDoc(index string, data interface{}) (bool, error) { // 添加索引 _, err := addIndex(index) if err != nil { log.Fatal("创建索引失败", err) } // 添加doc 先获取index再为index添加index res, err := esClient.Index(). Index(index). BodyJson(data). Do(ctx) if err != nil { return false, err } fmt.Println("添加数据成功:", res) return true, nil } //添加数据 指定id func AddDocById(id string, index string, data interface{}) (bool, error) { // 添加索引 _, err := addIndex(index) if err != nil { log.Fatal("创建索引失败", err) } // 添加doc 先获取index再为index添加index res, err := esClient.Index(). Index(index). BodyJson(data). Id(id). Do(ctx) if err != nil { return false, err } fmt.Println("添加数据成功:", res) return true, nil } // 添加索引 // 并在7.x后强制要求只能有一个类型，就是 _doc ---> /{index}/_doc/{id} func addIndex(index string) (bool, error) { // 创建index前，先查看es引擎中是否存在自己想要创建的索引index exists, err := esClient.IndexExists(index).Do(ctx) if err != nil { fmt.Println("存在索引:", err) return true, nil } if !exists { // 如果不存在，就创建 BodyString将索引的配置指定为字符串。 createIndex, err := esClient.CreateIndex(index).BodyString(mapping).Do(ctx) if err != nil { return false, err } if !createIndex.Acknowledged { return false, err } } return true, nil } 复制代码
select.go
package main import ( "encoding/json" "fmt" "github.com/olivere/elastic/v7" "strings" ) // 查询数据 func query(index string, field string, filter elastic.Query, sort string, page int, limit int) (*elastic.SearchResult, error) { // 分页数据处理 isAsc := true if sort != "" { sortSlice := strings.Split(sort, " ") sort = sortSlice[0] if sortSlice[1] == "desc" { isAsc = false } } // 查询位置处理 if page <= 1 { page = 1 } fsc := elastic.NewFetchSourceContext(true) // 返回字段处理 if field != "" { fieldSlice := strings.Split(field, ",") if len(fieldSlice) > 0 { for _, v := range fieldSlice { fsc.Include(v) } } } // 开始查询位置 fromStart := (page - 1) * limit res, err := esClient.Search(). Index(index). FetchSourceContext(fsc). Query(filter). Sort(sort, isAsc). From(fromStart). Size(limit). Pretty(true). Do(ctx) if err != nil { return nil, err } return res, nil } //index:"index" 索引 //field:"name,age" 要查询的字段 //filter:*TermQuery 查询规则 //sort:"age asc" 排序规则 //page:0 页数 //limit:10 条目数量 func QueryDoc(index string, field string, filter elastic.Query, sort string, page int, limit int) (interface{}, error) { res, err := query(index, field, filter, sort, page, limit) strD, _ := json.Marshal(res) if err != nil { fmt.Println("失败:", err) return nil, nil } fmt.Println("执行完成") return string(strD), nil } 复制代码
Update.go
package main import ( "errors" "fmt" "github.com/olivere/elastic/v7" ) // 条件更新文档 func UpdateDoc(index string, filter elastic.Query, data map[string]interface{}) (bool, error) { // 修改数据组装 if len(data) < 0 { return false, errors.New("修改参数不正确") } scriptStr := "" for k := range data { scriptStr += "ctx._source." + k + " = params." + k + ";" } script := elastic.NewScript(scriptStr).Params(data) res, err := esClient.UpdateByQuery(index). Query(filter). Script(script). Do(ctx) if err != nil { return false, err } fmt.Println("添加数据成功:", res) return true, nil } 复制代码
delete.go
package main import ( "fmt" "github.com/olivere/elastic/v7" ) // 删除文档 func DeleteDoc(index string, filter elastic.Query) (bool, error) { res, err := esClient.DeleteByQuery(). Query(filter). Index(index). Do(ctx) if err != nil { return false, err } fmt.Println("删除信息：", res) return true, nil } 复制代码
Main.go
package main import ( "fmt" "github.com/olivere/elastic/v7" ) func main() { //deleteDoc() //add() //update() } func add() { stu := Student{ Id: 2, Address: "北京", Name: "小红", Age: 10, } //res, err := AddDoc(index, stu) res, err := AddDocById("2", index, stu) fmt.Println(res) fmt.Println(err) } func selectDoc() { index_ := index sort := "age asc" page := 0 limit := 10 field := "id,name,age,address" filter := elastic.NewTermQuery("name", "名称") res, err := QueryDoc(index_, field, filter, sort, page, limit) fmt.Println(res) fmt.Println(err) } func deleteDoc() { query := elastic.NewTermQuery("name", "小明") DeleteDoc(index, query) } func update() { index_ := index filter := elastic.NewTermQuery("name", "小明") data := make(map[string]interface{}) data["address"] = "上海" data["age"] = 120 res, err := UpdateDoc(index_, filter, data) if err != nil { fmt.Println("失败:", err) } else { fmt.Println("成功:", res) } }
