Go整合ElasticSearch

go整合elasticsearch

  1. 基于docker搭建开发环境

在开发之前我们首先需要借助docker来构建我们的开发环境,先创建一个文件名称为docker-compose.yaml, 里面写入下面的内容:

---
  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 启动容器,之后在浏览器中分别验证es和kibana的运行状态

验证es:http://localhost:9200/

验证kibana:http://localhost:5601

  1. 检查客户端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)

}
  1. 索引相关操作
  • 创建索引
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"

	// 准备文档数据
	doc := Document{
		Title: "Document Title",
		Body:  "This is the body of the document.",
	}

	// 将文档数据序列化为JSON字节
	data, err := json.Marshal(doc)
	if err != nil {
		log.Fatalf("Error marshaling document: %s", err)
	}

	// 创建PUT请求
	req := esapi.IndexRequest{
		Index:      indexName,
		DocumentID: documentID,
		Body:       bytes.NewReader(data),
	}

	// 发送PUT请求
	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())

参考资料

  1. https://cloud.tencent.com/developer/article/1911255

你可能感兴趣的:(elasticsearch,elasticsearch,golang)