Update main.go

Author: fumiama

plugin/wordcount/main.go

@@ -11,27 +11,30 @@ import (
 	"sync"
 	"time"
 
+	"github.com/go-ego/gse"
+	"github.com/golang/freetype"
+	"github.com/sirupsen/logrus"
+	"github.com/tidwall/gjson"
+	"github.com/wcharczuk/go-chart/v2"
+
 	"github.com/FloatTech/floatbox/binary"
 	fcext "github.com/FloatTech/floatbox/ctxext"
 	"github.com/FloatTech/floatbox/file"
 	ctrl "github.com/FloatTech/zbpctrl"
 	"github.com/FloatTech/zbputils/control"
 	"github.com/FloatTech/zbputils/ctxext"
 	"github.com/FloatTech/zbputils/img/text"
-	"github.com/go-ego/gse"
-	"github.com/golang/freetype"
-	"github.com/sirupsen/logrus"
-	"github.com/tidwall/gjson"
-	"github.com/wcharczuk/go-chart/v2"
+
 	zero "github.com/wdvxdr1123/ZeroBot"
 	"github.com/wdvxdr1123/ZeroBot/message"
+	"github.com/wdvxdr1123/ZeroBot/utils/helper"
 )
 
 var (
 	re        = regexp.MustCompile(`^[一-龥]+$`)
 	stopwords []string
+	seg       gse.Segmenter
 )
-var seg gse.Segmenter
 
 func init() {
 	engine := control.AutoRegister(&ctrl.Options[*zero.Ctx]{
@@ -42,7 +45,10 @@ func init() {
 	})
 	cachePath := engine.DataFolder() + "cache/"
 	// 读取gse内置中文词典
-	_ = seg.LoadDictEmbed("zh_s")
+	err := seg.LoadDictEmbed()
+	if err != nil {
+		panic(err)
+	}
 	_ = os.RemoveAll(cachePath)
 	_ = os.MkdirAll(cachePath, 0755)
 	engine.OnRegex(`^热词\s?(\d*)\s?(\d*)$`, zero.OnlyGroup, fcext.DoOnceOnSuccess(func(ctx *zero.Ctx) bool {
@@ -124,7 +130,7 @@ func init() {
 						if tex == "" {
 							continue
 						}
-						segments := seg.Segment([]byte(tex))
+						segments := seg.Segment(helper.StringToBytes(tex))
 						words := gse.ToSlice(segments, true)
 						for _, word := range words {
 							word = strings.TrimSpace(word)