mirror of
https://github.com/yv1ing/MollyBlog.git
synced 2025-09-16 14:53:45 +08:00
37 lines
880 B
Go
37 lines
880 B
Go
package mApp
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/huichen/wukong/engine"
|
|
"github.com/huichen/wukong/types"
|
|
)
|
|
|
|
func (ma *MApp) loadPostIndex() {
|
|
cachePath := fmt.Sprintf("%s/persistent", ma.Config.CachePath)
|
|
ma.searcher = nil
|
|
ma.searcher = &engine.Engine{}
|
|
|
|
ma.searcher.Init(types.EngineInitOptions{
|
|
UsePersistentStorage: true,
|
|
PersistentStorageFolder: cachePath,
|
|
StopTokenFile: "data/search/stop_tokens.txt",
|
|
SegmenterDictionaries: "data/search/dictionary.txt",
|
|
IndexerInitOptions: &types.IndexerInitOptions{
|
|
IndexType: types.LocationsIndex,
|
|
},
|
|
})
|
|
|
|
for _, post := range ma.Posts {
|
|
postFile, _ := os.OpenFile(post.HtmlPath, os.O_RDONLY, 0666)
|
|
postData, _ := io.ReadAll(postFile)
|
|
|
|
ma.searcher.IndexDocument(post.Index, types.DocumentIndexData{Content: string(postData)}, false)
|
|
_ = postFile.Close()
|
|
}
|
|
|
|
ma.searcher.FlushIndex()
|
|
}
|