如何使用Golang进行Elasticsearch的查询操作-Golang-PHP中文网

如何使用Golang进行Elasticsearch的查询操作

PHPz

发布： 2023-04-11 09:13:36

原创

2179人浏览过

随着大数据时代的到来，数据的存储和查询需求也不断增加。elasticsearch 是目前较为流行的一个分布式搜索引擎，提供了相对简单易用的 restful api。而 golang 作为一门高效的编程语言，被越来越多的开发者所喜爱。这篇文章将介绍如何使用 golang 进行 elasticsearch 的查询操作。

一、依赖库安装

在 Golang 中，我们需要使用一个第三方库来进行 Elasticsearch 相关操作。推荐使用官方提供的 github.com/elastic/go-elasticsearch 库。

要安装这个库，只需要在终端中运行以下命令即可：

go get github.com/elastic/go-elasticsearch

登录后复制

如果你的电脑无法访问 github.com，请参考以下步骤：

立即学习“go语言免费学习笔记（深入）”；

1.访问 https://github.com/elastic/go-elasticsearch ，下载 zip 文件到本地。

2.将 zip 文件解压到某个目录下。

3.将解压后的目录移动至您的工程目录下的 GOPATH 目录。

4.在终端运行以下命令：

cd $GOPATH/go-elasticsearch
go install

登录后复制

这个过程可能比较耗时，请耐心等待。

二、建立 Elasticsearch 连接

PPT.CN,PPTCN,PPT.CN是什么,PPT.CN官网,PPT.CN如何使用

一键操作，智能生成专业级PPT

查看详情

要进行 Elasticsearch 查询，我们需要先建立连接。在 Golang 中，首先需要引入 github.com/elastic/go-elasticsearch 库，然后使用 NewDefaultClient 方法即可建立连接。

import (
    "fmt"
    "github.com/elastic/go-elasticsearch"
    "log"
)

func main() {
    cfg := elasticsearch.Config{
        Addresses: []string{
            "http://localhost:9200",
        },
    }
    es, err := elasticsearch.NewClient(cfg)
    if err != nil {
        log.Fatalf("连接 Elasticsearch 失败：%s", err)
    }
    fmt.Println("连接 Elasticsearch 成功")
}

登录后复制

这里我们指定 Elasticsearch 的地址为 http://localhost:9200，如果您的 Elasticsearch 运行在其他地址上，请修改该地址即可。

三、查询 Elasticsearch 数据

建立连接后，就可以进行 Elasticsearch 的查询操作了。我们可以通过 Golang 中的 http 库发送 HTTP 请求，并接收响应内容，即可完成 Elasticsearch 的查询操作。

以查询索引 test_index 下 message 字段包含 hello 字符串的所有数据为例：

import (
    "bytes"
    "encoding/json"
    "fmt"
    "github.com/elastic/go-elasticsearch"
    "github.com/elastic/go-elasticsearch/esapi"
    "io/ioutil"
    "log"
    "net/http"
    "strings"
)

func main() {
    cfg := elasticsearch.Config{
        Addresses: []string{
            "http://localhost:9200",
        },
    }
    es, err := elasticsearch.NewClient(cfg)
    if err != nil {
        log.Fatalf("连接 Elasticsearch 失败：%s", err)
    }
    fmt.Println("连接 Elasticsearch 成功")

    var (
        r map[string]interface{}
        b bytes.Buffer
    )

    query := map[string]interface{}{
        "query": map[string]interface{}{
            "match": map[string]interface{}{
                "message": "hello",
            },
        },
    }

    if err := json.NewEncoder(&b).Encode(query); err != nil {
        log.Fatalf("无法编码查询：%s", err)
    }

    req, _ := http.NewRequest("GET", "/test_index/_search", &b)
    req.Header.Add("Content-Type", "application/json")

    res, err := es.Perform(req)
    if err != nil {
        log.Fatalf("查询 Elasticsearch 失败：%s", err)
    }

    defer res.Body.Close()

    if res.IsError() {
        var r map[string]interface{}
        if err := json.NewDecoder(res.Body).Decode(&r); err != nil {
            log.Fatalf("响应错误：%s", err)
        } else {
            // 响应错误信息
            log.Fatalf("响应错误：%s", r["error"].(map[string]interface{})["reason"])
        }
    }

    if err := json.NewDecoder(res.Body).Decode(&r); err != nil {
        log.Fatalf("响应结果解析失败：%s", err)
    }

    results := r["hits"].(map[string]interface{})["hits"].([]interface{})
    fmt.Printf("共找到 %d 条匹配结果：\n", len(results))
    for _, result := range results {
        message := result.(map[string]interface{})["_source"].(map[string]interface{})["message"].(string)
        fmt.Printf("%s\n", message)
    }
}

登录后复制

这里我们首先定义了一个查询条件，即 message 字段包含 hello 字符串。然后使用 Golang 的 http 库创建了一个 HTTP 请求，并将该查询条件放在了请求体中。接着使用 es.Perform 方法发送请求，并接收响应结果。

如果响应结果出错，我们可以通过解析 JSON 数据，得到错误信息。如果响应结果没有出错，我们将查询结果打印在终端中。

需要注意的是，这里我们使用了 GET 方法发送了一个查询请求。实际上，Elasticsearch 支持多种不同的查询请求方式，包括 GET，POST，PUT 等。具体查询方式和语法，请参考 Elasticsearch 官方文档。

本文介绍的方法基于 Elasticsearch 的 RESTful API。除此之外，Elasticsearch 还提供了一种更加灵活高效的查询方式，即使用其官方提供的 Golang 库 github.com/olivere/elastic。如果您有更高效的查询需求，可以考虑使用这个库。

总之，在 Golang 中使用 Elasticsearch 进行查询是非常方便快捷的。只需要几行代码，即可实现强大的数据查询功能。建议开发者们多多关注 Elasticsearch 相关技术，提高自己的数据存储和查询能力。

以上就是如何使用Golang进行Elasticsearch的查询操作的详细内容，更多请关注php中文网其它相关文章！