forked from go-ego/riot
/
main.go
62 lines (49 loc) · 1.21 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
package main
import (
"log"
"github.com/go-ego/gse"
"github.com/go-ego/riot"
"github.com/go-ego/riot/types"
)
var (
searcher = riot.Engine{}
searcher1 = riot.Engine{}
searcher2 = riot.Engine{}
)
func main() {
searcher2.Init(types.EngineOpts{
Using: 1,
})
defer searcher2.Close()
log.Println("searcher2------------------...")
gseSegmenter := gse.Segmenter{}
gseSegmenter.LoadDict("zh")
searcher.WithGse(gseSegmenter).Init(
types.EngineOpts{
Using: 1,
})
log.Println("searcher----------------...")
searcher1.WithGse(gseSegmenter).Init(
types.EngineOpts{
Using: 1,
})
data := types.DocData{Content: `I wonder how, I wonder why
, I wonder where they are`}
data1 := types.DocData{Content: "所以, 你好, 再见"}
data2 := types.DocData{Content: "没有理由"}
searcher.Index(1, data)
searcher.Index(2, data1)
searcher.Index(4, data1)
searcher.Index(3, data2)
searcher.Flush()
req := types.SearchReq{Text: "你好"}
search := searcher.Search(req)
log.Println("search...", search)
searcher1.Index(1, data)
searcher1.Index(2, data1)
searcher1.Index(4, data1)
searcher1.Index(3, data2)
searcher1.Flush()
search1 := searcher1.Search(req)
log.Println("search1...", search1)
}