admin
e6956aa001
- React frontend with route-level code splitting - Backend rebranded from Baihu to TaskPool - DB brand migration script and local compatibility
91 lines
1.7 KiB
Go
91 lines
1.7 KiB
Go
package constant
|
|
|
|
import (
|
|
"bytes"
|
|
_ "embed"
|
|
"encoding/json"
|
|
"math/rand"
|
|
"sync"
|
|
)
|
|
|
|
//go:embed sentence1-10000.json
|
|
var sentenceData []byte
|
|
|
|
type Sentence struct {
|
|
Name string `json:"name"`
|
|
From string `json:"from"`
|
|
}
|
|
|
|
var (
|
|
lineCount int
|
|
lineOffsets []int64
|
|
sentenceOnce sync.Once
|
|
)
|
|
|
|
// initSentences 初始化：统计行数和记录每行偏移
|
|
func initSentences() {
|
|
sentenceOnce.Do(func() {
|
|
data := sentenceData
|
|
var offset int64 = 0
|
|
for {
|
|
// 记录当前行的起始位置
|
|
lineOffsets = append(lineOffsets, offset)
|
|
lineCount++
|
|
|
|
// 寻找下一个换行符
|
|
idx := bytes.IndexByte(data[offset:], '\n')
|
|
if idx == -1 {
|
|
// 最后一行没有换行符
|
|
break
|
|
}
|
|
|
|
// 移动到下一行的起始位置
|
|
offset += int64(idx) + 1
|
|
if offset >= int64(len(data)) {
|
|
break
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
// GetRandomSentence 随机获取一条古诗词
|
|
func GetRandomSentence() string {
|
|
initSentences()
|
|
if lineCount <= 0 {
|
|
return "欢迎使用任务池"
|
|
}
|
|
|
|
targetIndex := rand.Intn(lineCount)
|
|
start := lineOffsets[targetIndex]
|
|
|
|
// 确定当前行的结束位置
|
|
var end int64
|
|
if targetIndex < lineCount-1 {
|
|
end = lineOffsets[targetIndex+1]
|
|
} else {
|
|
end = int64(len(sentenceData))
|
|
}
|
|
|
|
// 提取行并清理两端的空白字符（包括 \r, \n）
|
|
line := bytes.TrimSpace(sentenceData[start:end])
|
|
if len(line) == 0 {
|
|
return "欢迎使用任务池"
|
|
}
|
|
|
|
var sData []string
|
|
if err := json.Unmarshal(line, &sData); err == nil && len(sData) >= 1 {
|
|
name := sData[0]
|
|
from := ""
|
|
if len(sData) >= 2 {
|
|
from = sData[1]
|
|
}
|
|
|
|
if from != "" {
|
|
return "\"" + name + "\"—— " + from
|
|
}
|
|
return name
|
|
}
|
|
|
|
return "欢迎使用任务池"
|
|
}
|