Function initUserHabits

plugin/handou/habits.go:33–89

初始化用户习惯

()

// 初始化用户习惯
func initUserHabits() error {
	userHabits = &UserHabits{
		habits:  make(map[string]int),
		bigrams: make(map[string]int),
		idioms:  make(map[string]int),
	}

	if file.IsNotExist(userHabitsFile) {
		f, err := os.Create(userHabitsFile)
		if err != nil {
			return errors.New("创建用户习惯库时发生错误: " + err.Error())
		}
		_ = f.Close()
		return saveHabits()
	}

	// 读取现有习惯数据
	habitsFile, err := os.ReadFile(userHabitsFile)
	if err != nil {
		return errors.New("读取用户习惯库时发生错误: " + err.Error())
	}

	var savedData struct {
		Habits      map[string]int `json:"habits"`
		Bigrams     map[string]int `json:"bigrams"`
		Idioms      map[string]int `json:"idioms"`
		TotalWords  int            `json:"total_words"`
		TotalIdioms int            `json:"total_idioms"`
		LastUpdate  time.Time      `json:"last_update"`
	}

	if err := json.Unmarshal(habitsFile, &savedData); err != nil {
		// 如果是旧格式，尝试兼容
		var oldHabits map[string]int
		if err := json.Unmarshal(habitsFile, &oldHabits); err == nil {
			savedData.Habits = oldHabits
			// 从旧数据重新计算统计信息
			for _, count := range oldHabits {
				savedData.TotalWords += count
			}
		} else {
			return errors.New("解析用户习惯库时发生错误: " + err.Error())
		}
	}

	userHabits.mu.Lock()
	defer userHabits.mu.Unlock()

	userHabits.habits = savedData.Habits
	userHabits.bigrams = savedData.Bigrams
	userHabits.idioms = savedData.Idioms
	userHabits.totalWords = savedData.TotalWords
	userHabits.totalIdioms = savedData.TotalIdioms
	userHabits.lastUpdate = savedData.LastUpdate

	return nil
}