349 lines
9.6 KiB
Go
349 lines
9.6 KiB
Go
package services
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"time"
|
||
|
||
"gorm.io/gorm"
|
||
|
||
"github.com/Nanqipro/YunQue-Tech-Projects/ai_english_learning/serve/internal/common"
|
||
"github.com/Nanqipro/YunQue-Tech-Projects/ai_english_learning/serve/internal/models"
|
||
"github.com/Nanqipro/YunQue-Tech-Projects/ai_english_learning/serve/internal/utils"
|
||
)
|
||
|
||
// UserService 用户服务
|
||
type UserService struct {
|
||
db *gorm.DB
|
||
}
|
||
|
||
// NewUserService 创建用户服务实例
|
||
func NewUserService(db *gorm.DB) *UserService {
|
||
return &UserService{db: db}
|
||
}
|
||
|
||
// CreateUser 创建用户
|
||
func (s *UserService) CreateUser(username, email, password string) (*models.User, error) {
|
||
// 检查用户名是否已存在
|
||
var existingUser models.User
|
||
if err := s.db.Where("username = ? OR email = ?", username, email).First(&existingUser).Error; err == nil {
|
||
if existingUser.Username == username {
|
||
return nil, common.ErrUsernameExists
|
||
}
|
||
if existingUser.Email == email {
|
||
return nil, common.ErrEmailExists
|
||
}
|
||
} else if !errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, err
|
||
}
|
||
|
||
// 生成密码哈希
|
||
passwordHash, err := utils.HashPassword(password)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 创建用户
|
||
user := &models.User{
|
||
Username: username,
|
||
Email: email,
|
||
PasswordHash: passwordHash,
|
||
Status: "active",
|
||
Timezone: "Asia/Shanghai",
|
||
Language: "zh-CN",
|
||
CreatedAt: time.Now(),
|
||
UpdatedAt: time.Now(),
|
||
}
|
||
|
||
if err := s.db.Create(user).Error; err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 创建用户偏好设置
|
||
preference := &models.UserPreference{
|
||
UserID: user.ID,
|
||
DailyGoal: 50,
|
||
WeeklyGoal: 350,
|
||
ReminderEnabled: true,
|
||
DifficultyLevel: "beginner",
|
||
LearningMode: "casual",
|
||
CreatedAt: time.Now(),
|
||
UpdatedAt: time.Now(),
|
||
}
|
||
|
||
if err := s.db.Create(preference).Error; err != nil {
|
||
// 如果创建偏好设置失败，记录日志但不影响用户创建
|
||
// 可以在这里添加日志记录
|
||
}
|
||
|
||
return user, nil
|
||
}
|
||
|
||
// GetUserByID 根据ID获取用户
|
||
func (s *UserService) GetUserByID(userID int64) (*models.User, error) {
|
||
var user models.User
|
||
if err := s.db.Preload("Preferences").Preload("SocialLinks").Where("id = ?", userID).First(&user).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, common.ErrUserNotFound
|
||
}
|
||
return nil, err
|
||
}
|
||
return &user, nil
|
||
}
|
||
|
||
// GetUserByEmail 根据邮箱获取用户
|
||
func (s *UserService) GetUserByEmail(email string) (*models.User, error) {
|
||
var user models.User
|
||
if err := s.db.Where("email = ?", email).First(&user).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, common.ErrUserNotFound
|
||
}
|
||
return nil, err
|
||
}
|
||
return &user, nil
|
||
}
|
||
|
||
// GetUserByUsername 根据用户名获取用户
|
||
func (s *UserService) GetUserByUsername(username string) (*models.User, error) {
|
||
var user models.User
|
||
if err := s.db.Where("username = ?", username).First(&user).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, common.ErrUserNotFound
|
||
}
|
||
return nil, err
|
||
}
|
||
return &user, nil
|
||
}
|
||
|
||
// UpdateUser 更新用户信息
|
||
func (s *UserService) UpdateUser(userID int64, updates map[string]interface{}) (*models.User, error) {
|
||
// 检查用户是否存在
|
||
var user models.User
|
||
if err := s.db.Where("id = ?", userID).First(&user).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, common.ErrUserNotFound
|
||
}
|
||
return nil, err
|
||
}
|
||
|
||
// 如果更新邮箱，检查邮箱是否已被其他用户使用
|
||
if email, ok := updates["email"]; ok {
|
||
var existingUser models.User
|
||
if err := s.db.Where("email = ? AND id != ?", email, userID).First(&existingUser).Error; err == nil {
|
||
return nil, common.ErrEmailExists
|
||
} else if !errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
// 如果更新用户名，检查用户名是否已被其他用户使用
|
||
if username, ok := updates["username"]; ok {
|
||
var existingUser models.User
|
||
if err := s.db.Where("username = ? AND id != ?", username, userID).First(&existingUser).Error; err == nil {
|
||
return nil, common.ErrUsernameExists
|
||
} else if !errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
// 更新时间戳
|
||
updates["updated_at"] = time.Now()
|
||
|
||
// 执行更新
|
||
if err := s.db.Model(&user).Updates(updates).Error; err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 重新获取更新后的用户信息
|
||
return s.GetUserByID(userID)
|
||
}
|
||
|
||
// UpdatePassword 更新用户密码
|
||
func (s *UserService) UpdatePassword(userID int64, oldPassword, newPassword string) error {
|
||
// 获取用户
|
||
var user models.User
|
||
if err := s.db.Where("id = ?", userID).First(&user).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return common.ErrUserNotFound
|
||
}
|
||
return err
|
||
}
|
||
|
||
// 验证旧密码
|
||
if !utils.CheckPasswordHash(oldPassword, user.PasswordHash) {
|
||
return common.ErrInvalidPassword
|
||
}
|
||
|
||
// 生成新密码哈希
|
||
newPasswordHash, err := utils.HashPassword(newPassword)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 更新密码
|
||
return s.db.Model(&user).Updates(map[string]interface{}{
|
||
"password_hash": newPasswordHash,
|
||
"updated_at": time.Now(),
|
||
}).Error
|
||
}
|
||
|
||
// UpdateLoginInfo 更新登录信息
|
||
func (s *UserService) UpdateLoginInfo(userID int64, loginIP string) error {
|
||
now := time.Now()
|
||
return s.db.Model(&models.User{}).Where("id = ?", userID).Updates(map[string]interface{}{
|
||
"last_login_at": &now,
|
||
"last_login_ip": loginIP,
|
||
"login_count": gorm.Expr("login_count + 1"),
|
||
"updated_at": now,
|
||
}).Error
|
||
}
|
||
|
||
// VerifyPassword 验证密码
|
||
func (s *UserService) VerifyPassword(userID int64, password string) error {
|
||
var user models.User
|
||
if err := s.db.Where("id = ?", userID).First(&user).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return common.ErrUserNotFound
|
||
}
|
||
return err
|
||
}
|
||
|
||
if !utils.CheckPasswordHash(password, user.PasswordHash) {
|
||
return common.ErrInvalidPassword
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// DeleteUser 删除用户
|
||
func (s *UserService) DeleteUser(userID int64) error {
|
||
// 检查用户是否存在
|
||
var user models.User
|
||
if err := s.db.Where("id = ?", userID).First(&user).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return common.ErrUserNotFound
|
||
}
|
||
return err
|
||
}
|
||
|
||
// 软删除用户
|
||
return s.db.Delete(&user).Error
|
||
}
|
||
|
||
// GetUserPreferences 获取用户偏好设置
|
||
func (s *UserService) GetUserPreferences(userID int64) (*models.UserPreference, error) {
|
||
var preference models.UserPreference
|
||
if err := s.db.Where("user_id = ?", userID).First(&preference).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, common.ErrUserNotFound
|
||
}
|
||
return nil, err
|
||
}
|
||
return &preference, nil
|
||
}
|
||
|
||
// UpdateUserPreferences 更新用户偏好设置
|
||
func (s *UserService) UpdateUserPreferences(userID int64, updates map[string]interface{}) (*models.UserPreference, error) {
|
||
// 检查偏好设置是否存在
|
||
var preference models.UserPreference
|
||
if err := s.db.Where("user_id = ?", userID).First(&preference).Error; err != nil {
|
||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||
return nil, common.ErrUserNotFound
|
||
}
|
||
return nil, err
|
||
}
|
||
|
||
// 更新时间戳
|
||
updates["updated_at"] = time.Now()
|
||
|
||
// 执行更新
|
||
if err := s.db.Model(&preference).Updates(updates).Error; err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 重新获取更新后的偏好设置
|
||
return s.GetUserPreferences(userID)
|
||
}
|
||
|
||
// GetUserLearningProgress 获取用户学习进度
|
||
func (s *UserService) GetUserLearningProgress(userID string, page, limit int) ([]map[string]interface{}, int64, error) {
|
||
// 初始化为空切片而不是nil，避免JSON序列化为null
|
||
progressList := []map[string]interface{}{}
|
||
var total int64
|
||
|
||
// 查询用户在各个词汇书的学习进度
|
||
query := `
|
||
SELECT
|
||
vb.id,
|
||
vb.name as title,
|
||
vb.description as category,
|
||
vb.level,
|
||
COUNT(DISTINCT vbw.vocabulary_id) as total_words,
|
||
COUNT(DISTINCT CASE WHEN uwp.status IN ('learning', 'reviewing', 'mastered') THEN uwp.vocabulary_id END) as learned_words,
|
||
MAX(uwp.last_studied_at) as last_study_date
|
||
FROM ai_vocabulary_books vb
|
||
LEFT JOIN ai_vocabulary_book_words vbw ON vbw.book_id = vb.id
|
||
LEFT JOIN ai_user_word_progress uwp ON CAST(uwp.vocabulary_id AS UNSIGNED) = CAST(vbw.vocabulary_id AS UNSIGNED) AND uwp.user_id = ?
|
||
WHERE vb.is_system = 1
|
||
GROUP BY vb.id, vb.name, vb.description, vb.level
|
||
HAVING total_words > 0
|
||
ORDER BY last_study_date DESC
|
||
`
|
||
|
||
// 获取总数
|
||
countQuery := `
|
||
SELECT COUNT(*) FROM (
|
||
SELECT vb.id
|
||
FROM ai_vocabulary_books vb
|
||
LEFT JOIN ai_vocabulary_book_words vbw ON vbw.book_id = vb.id
|
||
WHERE vb.is_system = 1
|
||
GROUP BY vb.id
|
||
HAVING COUNT(DISTINCT vbw.vocabulary_id) > 0
|
||
) as subquery
|
||
`
|
||
if err := s.db.Raw(countQuery).Scan(&total).Error; err != nil {
|
||
return nil, 0, err
|
||
}
|
||
|
||
// 分页查询
|
||
offset := (page - 1) * limit
|
||
rows, err := s.db.Raw(query+" LIMIT ? OFFSET ?", userID, limit, offset).Rows()
|
||
if err != nil {
|
||
return nil, 0, err
|
||
}
|
||
defer rows.Close()
|
||
|
||
for rows.Next() {
|
||
var (
|
||
id int64
|
||
title string
|
||
category string
|
||
level string
|
||
totalWords int
|
||
learnedWords int
|
||
lastStudyDate *time.Time
|
||
)
|
||
|
||
if err := rows.Scan(&id, &title, &category, &level, &totalWords, &learnedWords, &lastStudyDate); err != nil {
|
||
continue
|
||
}
|
||
|
||
progress := float64(0)
|
||
if totalWords > 0 {
|
||
progress = float64(learnedWords) / float64(totalWords)
|
||
}
|
||
|
||
progressList = append(progressList, map[string]interface{}{
|
||
"id": fmt.Sprintf("%d", id),
|
||
"title": title,
|
||
"category": category,
|
||
"level": level,
|
||
"total_words": totalWords,
|
||
"learned_words": learnedWords,
|
||
"progress": progress,
|
||
"last_study_date": lastStudyDate,
|
||
})
|
||
}
|
||
|
||
return progressList, total, nil
|
||
} |