Go整合ElasticSearch的示例代碼

更新時(shí)間：2023年07月28日 14:13:57 作者：飽腹百科

這篇文章主要介紹了Go整合ElasticSearch的相關(guān)知識(shí)，本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下

go整合elasticsearch

基于docker搭建開發(fā)環(huán)境

在開發(fā)之前我們首先需要借助docker來構(gòu)建我們的開發(fā)環(huán)境，先創(chuàng)建一個(gè)文件名稱為docker-compose.yaml, 里面寫入下面的內(nèi)容：

---
  version: "3"
  services:
    elasticsearch:
      image: docker.elastic.co/elasticsearch/elasticsearch:7.10.0
      container_name: es01
      environment:
        - node.name=es01
        - cluster.name=docker-cluster
        - bootstrap.memory_lock=true
        - "ES_JAVA_OPTS=-Xms512m -Xmx512m"
        - discovery.type=single-node
      ulimits:
        memlock:
          soft: -1
          hard: -1
      volumes:
        - esdata:/usr/share/elasticsearch/data
      ports:
        - 9200:9200
    kibana:
      image: docker.elastic.co/kibana/kibana:7.10.0
      ports:
        - 5601:5601
      depends_on:
        - elasticsearch
  volumes:
    esdata:
      driver: local

使用docker-compose up -d 啟動(dòng)容器，之后在瀏覽器中分別驗(yàn)證es和kibana的運(yùn)行狀態(tài)

驗(yàn)證es:http://localhost:9200/

驗(yàn)證kibana：http://localhost:5601

檢查客戶端api

package main
import (
	"fmt"
	"github.com/elastic/go-elasticsearch/v7"
)
func main() {
	es, err := elasticsearch.NewDefaultClient()
	if err != nil {
		fmt.Println(err)
		return
	}
	res, err := es.Info()
	if err != nil {
		fmt.Println(err)
		return
	}
	defer res.Body.Close()
	fmt.Println(res)
}

索引相關(guān)操作

創(chuàng)建索引

package main
import (
	"context"
	"fmt"
	"github.com/elastic/go-elasticsearch/v7"
	"log"
)
func main() {
	cfg := elasticsearch.Config{
		Addresses: []string{
			"http://localhost:9200",
		},
	}
	es, err := elasticsearch.NewClient(cfg)
	if err != nil {
		log.Fatalf("Error creating the client: %s", err)
	}
	indexName := "test_20230726"
	res, err := es.Indices.Create(
		indexName,
		es.Indices.Create.WithContext(context.Background()),
		es.Indices.Create.WithPretty())
	if err != nil {
		log.Fatalf("Error creating the index: %s", err)
	}
	defer res.Body.Close()
	fmt.Println(res.String())
}

刪除索引

package main
import (
	"context"
	"fmt"
	"github.com/elastic/go-elasticsearch/v7"
	"log"
)
func main() {
	cfg := elasticsearch.Config{
		Addresses: []string{
			"http://localhost:9200",
		},
	}
	es, err := elasticsearch.NewClient(cfg)
	if err != nil {
		log.Fatalf("Error creating the client: %s", err)
	}
	indexName := "test_20230726"
	res, err := es.Indices.Delete(
		[]string{indexName},
		es.Indices.Delete.WithContext(context.Background()),
		es.Indices.Delete.WithIgnoreUnavailable(true),
		es.Indices.Delete.WithPretty(),
	)
	if err != nil {
		log.Fatalf("Error deleting the index: %s", err)
	}
	defer res.Body.Close()
	fmt.Println(res.String())
}

修改索引

package main
import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"github.com/elastic/go-elasticsearch/v7"
	"github.com/elastic/go-elasticsearch/v7/esapi"
	"log"
)
func main() {
	cfg := elasticsearch.Config{
		Addresses: []string{
			"http://localhost:9200",
		},
	}
	es, err := elasticsearch.NewClient(cfg)
	if err != nil {
		log.Fatalf("Error creating the client: %s", err)
	}
	indexName := "your_index_name"
	documentID := "your_document_id"
	// 準(zhǔn)備文檔數(shù)據(jù)
	doc := Document{
		Title: "Document Title",
		Body:  "This is the body of the document.",
	}
	// 將文檔數(shù)據(jù)序列化為JSON字節(jié)
	data, err := json.Marshal(doc)
	if err != nil {
		log.Fatalf("Error marshaling document: %s", err)
	}
	// 創(chuàng)建PUT請(qǐng)求
	req := esapi.IndexRequest{
		Index:      indexName,
		DocumentID: documentID,
		Body:       bytes.NewReader(data),
	}
	// 發(fā)送PUT請(qǐng)求
	res, err := req.Do(context.Background(), es)
	if err != nil {
		log.Fatalf("Error indexing document: %s", err)
	}
	defer res.Body.Close()
	fmt.Println(res.String())
}

查詢索引列表

package main
import (
	"encoding/json"
	"fmt"
	"github.com/elastic/go-elasticsearch/v7"
	"log"
)
func main() {
	cfg := elasticsearch.Config{
		Addresses: []string{
			"http://localhost:9200",
		},
	}
	es, err := elasticsearch.NewClient(cfg)
	if err != nil {
		log.Fatalf("Error creating the client: %s", err)
	}
	res, err := es.Indices.Get([]string{"_all"})
	if err != nil {
		log.Fatalf("Error getting indices: %s", err)
	}
	defer res.Body.Close()
	if res.IsError() {
		log.Fatalf("Error response: %s", res.String())
	}
	var result map[string]interface{}
	if err := json.NewDecoder(res.Body).Decode(&result); err != nil {
		log.Fatalf("Error parsing the response body: %s", err)
	}
	indices, ok := result["test_20230726"].(map[string]interface{})
	if !ok {
		log.Fatalf("Invalid indices format in the response")
	}
	for index := range indices {
		fmt.Println(index)
	}
}

插入文檔

package main
import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"github.com/elastic/go-elasticsearch/v7"
	"github.com/elastic/go-elasticsearch/v7/esapi"
	"log"
)
type Document struct {
	Title string `json:"title"`
	Body  string `json:"body"`
}
func main() {
	cfg := elasticsearch.Config{
		Addresses: []string{
			"http://localhost:9200",
		},
	}
	es, err := elasticsearch.NewClient(cfg)
	if err != nil {
		log.Fatalf("Error creating the client: %s", err)
	}
	indexName := "test_20230726"
	documentID := "20230726"
	doc := Document{
		Title: "Document Title",
		Body:  "This is the body of the document.",
	}
	data, err := json.Marshal(doc)
	if err != nil {
		log.Fatalf("Error marshaling document: %s", err)
	}
	req := esapi.IndexRequest{
		Index:      indexName,
		DocumentID: documentID,
		Body:       bytes.NewReader(data),
	}
	res, err := req.Do(context.Background(), es)
	if err != nil {
		log.Fatalf("Error indexing document: %s", err)
	}
	defer res.Body.Close()
	fmt.Println(res.String())