引入标签搜索

models/elasticsearch.go

@@ -0,0 +1,89 @@
+package models
+
+import (
+	"bytes"
+	"context"
+	"crypto/tls"
+	"encoding/json"
+	"log"
+	"net/http"
+
+	"github.com/elastic/go-elasticsearch/v8"
+)
+
+func elasticsearch_init() (es *elasticsearch.Client) {
+	es, err := elasticsearch.NewClient(elasticsearch.Config{
+		Addresses: []string{
+			Viper.Get("elasticsearch.host").(string),
+		},
+		Username: Viper.Get("elasticsearch.user").(string),
+		Password: Viper.Get("elasticsearch.password").(string),
+		Transport: &http.Transport{
+			TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
+		},
+	})
+	if err != nil {
+		log.Printf("Error creating the client: %s", err)
+		return nil
+	}
+	return es
+}
+
+func ElasticsearchSearch(text string) interface{} {
+	var (
+		r map[string]interface{}
+	)
+
+	// 通过字符串构建查询
+	var buf bytes.Buffer
+	query := map[string]interface{}{
+		"query": map[string]interface{}{
+			"match": map[string]interface{}{
+				"content": text,
+			},
+		},
+	}
+	if err := json.NewEncoder(&buf).Encode(query); err != nil {
+		log.Fatalf("Error encoding query: %s", err)
+	}
+
+	es := elasticsearch_init()
+
+	// Perform the search request.
+	res, err := es.Search(
+		es.Search.WithContext(context.Background()),
+		es.Search.WithIndex("news"),
+		es.Search.WithBody(&buf),
+		es.Search.WithTrackTotalHits(true),
+		es.Search.WithPretty(),
+	)
+	if err != nil {
+		log.Fatalf("Error getting response: %s", err)
+	}
+	defer res.Body.Close()
+
+	// Check response status
+	if res.IsError() {
+		log.Fatalf("Error: %s", res.String())
+	}
+
+	// Deserialize the response into a map.
+	if err := json.NewDecoder(res.Body).Decode(&r); err != nil {
+		log.Fatalf("Error parsing the response body: %s", err)
+	}
+
+	// Print the response status, number of results, and request duration.
+	log.Printf(
+		"[%s] %d hits; took: %dms",
+		res.Status(),
+		int(r["hits"].(map[string]interface{})["total"].(map[string]interface{})["value"].(float64)),
+		int(r["took"].(float64)),
+	)
+
+	// Print the ID and document source for each hit.
+	for _, hit := range r["hits"].(map[string]interface{})["hits"].([]interface{}) {
+		log.Printf(" * ID=%s, %s", hit.(map[string]interface{})["_id"], hit.(map[string]interface{})["_source"])
+	}
+
+	return r
+}

models/milvus.go

@@ -20,9 +20,10 @@ func (m *MilvusConnection) Init() (err error) {
 	log.Println("Milvus connection init")
 	host := Viper.Get("milvus.host").(string)
 	port := Viper.Get("milvus.port").(int)
-	address := fmt.Sprintf("%s:%d", host, port)
-	log.Println("Milvus address:", address)
-	m.Client, err = client.NewGrpcClient(context.Background(), address)
+	m.Client, err = client.NewGrpcClient(
+		context.Background(),
+		fmt.Sprintf("%s:%d", host, port),
+	)
 	if err != nil {
 		log.Println("Milvus connection failed:", err)
 		return