From fc2f92dbe9fd242bd72019ae45a393a3a4eb782f Mon Sep 17 00:00:00 2001 From: tsaolun Date: Tue, 15 Aug 2023 15:54:29 +0800 Subject: [PATCH] =?UTF-8?q?=09=E4=BF=AE=E6=94=B9=EF=BC=9A=20=20=20=20=20bo?= =?UTF-8?q?t/adapters/storage/memorystorage.go?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- bot/adapters/storage/memorystorage.go | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/bot/adapters/storage/memorystorage.go b/bot/adapters/storage/memorystorage.go index 70be23b..7d03130 100644 --- a/bot/adapters/storage/memorystorage.go +++ b/bot/adapters/storage/memorystorage.go @@ -27,6 +27,15 @@ var ( generatedStopWordsFile = "../etc/stopwords.txt" ) +func init() { + // 获取当前文件的绝对路径 + dir, _ := os.Getwd() + dictFile = path.Join(dir, dictFile) + idfFile = path.Join(dir, idfFile) + stopWordsFile = path.Join(dir, stopWordsFile) + generatedStopWordsFile = path.Join(dir, generatedStopWordsFile) +} + type ( keyChunk struct { offfset int @@ -45,12 +54,6 @@ type ( func RestoreMemoryStorage(decoder *gob.Decoder) (*memoryStorage, error) { var segmenter jiebago.Segmenter - // 获取当前文件的绝对路径 - dir, _ := os.Getwd() - dictFile = path.Join(dir, dictFile) - idfFile = path.Join(dir, idfFile) - stopWordsFile = path.Join(dir, stopWordsFile) - generatedStopWordsFile = path.Join(dir, generatedStopWordsFile) logx.Must(segmenter.LoadDictionary(dictFile)) var extracter analyse.TagExtracter logx.Must(extracter.LoadDictionary(dictFile))