From 6527c07b9051b8b4cce270b77233b8e917c6fb23 Mon Sep 17 00:00:00 2001 From: tsaolun Date: Tue, 15 Aug 2023 16:21:20 +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 | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/bot/adapters/storage/memorystorage.go b/bot/adapters/storage/memorystorage.go index 0e0d9b2..4120352 100644 --- a/bot/adapters/storage/memorystorage.go +++ b/bot/adapters/storage/memorystorage.go @@ -2,6 +2,7 @@ package storage import ( "encoding/gob" + "flag" "fmt" "log" "math" @@ -26,12 +27,13 @@ var ( idfFile = "./etc/idf.txt" stopWordsFile = "./etc/stop_words.txt" generatedStopWordsFile = "./etc/stopwords.txt" + EtcDir = flag.String("etc", "cli", "the etc directory to look for corpora files") ) func init() { // 获取当前文件的绝对路径 dir, _ := os.Getwd() - dictFile = path.Join(dir, dictFile) + dictFile = path.Join(dir, *EtcDir, dictFile) idfFile = path.Join(dir, idfFile) stopWordsFile = path.Join(dir, stopWordsFile) generatedStopWordsFile = path.Join(dir, generatedStopWordsFile)