20251209

archive/bbs-go/server/internal/pkg/search/search_test.go

@@ -0,0 +1,76 @@
+package search_test
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+	"testing"
+	"time"
+
+	"github.com/blevesearch/bleve/v2"
+)
+
+type TopicDocument struct {
+	Id         int64    `json:"id"`
+	NodeId     int64    `json:"nodeId"`
+	UserId     int64    `json:"userId"`
+	Nickname   string   `json:"nickname"`
+	Title      string   `json:"title"`
+	Content    string   `json:"content"`
+	Tags       []string `json:"tags"`
+	Recommend  bool     `json:"recommend"`
+	Status     int      `json:"status"`
+	CreateTime int64    `json:"createTime"`
+}
+
+func TestBleve(t *testing.T) {
+	// 打开或创建索引
+	index, err := bleve.Open("topic_index")
+	if err == bleve.ErrorIndexPathDoesNotExist {
+		indexMapping := bleve.NewIndexMapping()
+		index, err = bleve.New("topic_index", indexMapping)
+		// index, err = bleve.NewUsing("topic_index", indexMapping, scorch.Name, scorch.Name, nil)
+		if err != nil {
+			log.Fatal(err)
+		}
+	} else if err != nil {
+		log.Fatal(err)
+	}
+
+	// 添加文档到索引
+	doc := TopicDocument{
+		Id:         1,
+		NodeId:     1,
+		UserId:     1,
+		Nickname:   "user1",
+		Title:      "Example Title",
+		Content:    "This is an example content.",
+		Tags:       []string{"example", "test"},
+		Recommend:  true,
+		Status:     1,
+		CreateTime: time.Now().Unix(),
+	}
+	err = index.Index(fmt.Sprintf("%d", doc.Id), doc)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	// 执行查询
+	query := bleve.NewMatchQuery("example") // 查询示例关键词 "example"
+	searchRequest := bleve.NewSearchRequest(query)
+	searchRequest.Fields = []string{"id", "title", "content", "createTime"}
+	searchResult, err := index.Search(searchRequest)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	// 输出查询结果
+	for _, hit := range searchResult.Hits {
+		var topicDoc TopicDocument
+		err := json.Unmarshal(hit.Fields["title"].(json.RawMessage), &topicDoc)
+		if err != nil {
+			log.Fatal(err)
+		}
+		fmt.Printf("Title: %s, Content: %s\n", topicDoc.Title, topicDoc.Content)
+	}
+}