go es

安装go package:

1
go get github.com/olivere/elastic

 

开撸:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
/**
* es demo
* author:JetWu
* date:2020.06.10
 */
package es
 
import (
    "context"
    "errors"
    "fmt"
    "log"
    "reflect"
 
    "encoding/json"
 
    "github.com/olivere/elastic"
)
 
type Student struct {
    Name         string   `json:"name"`
    Age          int      `json:"age"`
    City         string   `json:"city"`
    Interests    []string `json:"interests"`
    Introduction string   `json:"introduction"`
}
 
var client *elastic.Client
var host = "http://127.0.0.1:9200/"
 
/**
* 连接es
 */
func init() {
    log.Println("Connecting to elasticsearch server...")
    //连接es
    var err error
    client, err = elastic.NewSimpleClient(elastic.SetURL(host))
    if err != nil {
        log.Println("Connected failed: ", err)
        return
    }
    log.Println("Connected succeed!")
    //测试es连接
    info, _, err := client.Ping(host).Do(context.Background())
    if err != nil {
        log.Println("Ping elasticsearch server failed: ", err)
        return
    }
    log.Println("elasticsearch version: ", info.Version.Number)
}
 
/**
* 打印查询语句
 */
func PrintQuery(src interface{}) {
    data, err := json.MarshalIndent(src, "", "")
    if err != nil {
        log.Println("PrintQuery Error: ", err)
        return
    }
    fmt.Println("PrintQuery:\n", string(data))
}
 
/**
* 搜索
 */
func Search(keyword string) []*Student {
    queryStr := `{
        "bool": {
            "should": [
                {
                    "match": {"name":"%s"}
                },
                {
                    "match": {"city":"%s"}
                },
                {
                    "match": {"interests":"%s"}
                },
                {
                    "match": {"introduction":"%s"}
                }
            ]
        }
    }`
    rawQuery := elastic.NewRawStringQuery(fmt.Sprintf(queryStr, keyword, keyword, keyword, keyword))
    PrintQuery(rawQuery)
 
    searchResult, err := client.Search("school").Type("student").
        Query(rawQuery).From(0).Size(10).Do(context.Background())
    if err != nil {
        log.Println("Search Error: ", err)
        return nil
    }
    fmt.Printf("Query took %d milliseconds, total: %d\n", searchResult.TookInMillis, searchResult.Hits.TotalHits)
 
    var students []*Student
    var stuType Student
    for _, item := range searchResult.Each(reflect.TypeOf(stuType)) {
        if stu, ok := item.(Student); ok {
            students = append(students, &stu)
        }
    }
    return students
}
 
/**
* 添加/修改
 */
func Add(indexId string, stu *Student) error {
    if stu == nil {
        return errors.New("Add Error: data empty!")
    }
    ret, err := client.Index().Index("school").Type("student").
        Id(indexId).BodyJson(stu).Do(context.Background())
    if err != nil {
        log.Println("Add Error: ", err)
        return err
    }
    fmt.Printf("Indexed %s to index %s, type %s\n", ret.Id, ret.Index, ret.Type)
    return nil
}
 
/**
* 删除
 */
func Remove(indexId string) error {
    ret, err := client.Delete().Index("school").Type("student").
        Id(indexId).Do(context.Background())
    if err != nil {
        log.Println("Remove Error: ", err)
        return err
    }
    fmt.Printf("delete result: %d, %s\n", ret.Status, ret.Result)
    return nil
}

  

posted @   疯一样的狼人  阅读(754)  评论(0编辑  收藏  举报
编辑推荐:
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· AI与.NET技术实操系列(五):向量存储与相似性搜索在 .NET 中的实现
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
点击右上角即可分享
微信分享提示