feat: 添加种子数据初始化功能，重构多个处理程序以简化错误响应和用户验证

internal/repository/token_repository.go

@@ -2,48 +2,38 @@ package repository
 
 import (
 	"carrotskin/internal/model"
-	"carrotskin/pkg/database"
 )
 
 func CreateToken(token *model.Token) error {
-	db := database.MustGetDB()
-	return db.Create(token).Error
+	return getDB().Create(token).Error
 }
 
 func GetTokensByUserId(userId int64) ([]*model.Token, error) {
-	db := database.MustGetDB()
-	tokens := make([]*model.Token, 0)
-	err := db.Where("user_id = ?", userId).Find(&tokens).Error
-	if err != nil {
-		return nil, err
-	}
-	return tokens, nil
+	var tokens []*model.Token
+	err := getDB().Where("user_id = ?", userId).Find(&tokens).Error
+	return tokens, err
 }
 
 func BatchDeleteTokens(tokensToDelete []string) (int64, error) {
-	db := database.MustGetDB()
 	if len(tokensToDelete) == 0 {
-		return 0, nil // 无需要删除的令牌，直接返回
+		return 0, nil
 	}
-	result := db.Where("access_token IN ?", tokensToDelete).Delete(&model.Token{})
-
+	result := getDB().Where("access_token IN ?", tokensToDelete).Delete(&model.Token{})
 	return result.RowsAffected, result.Error
 }
 
 func FindTokenByID(accessToken string) (*model.Token, error) {
-	db := database.MustGetDB()
-	var tokens []*model.Token
-	err := db.Where("_id = ?", accessToken).Find(&tokens).Error
+	var token model.Token
+	err := getDB().Where("access_token = ?", accessToken).First(&token).Error
 	if err != nil {
 		return nil, err
 	}
-	return tokens[0], nil
+	return &token, nil
 }
 
 func GetUUIDByAccessToken(accessToken string) (string, error) {
-	db := database.MustGetDB()
 	var token model.Token
-	err := db.Where("access_token = ?", accessToken).First(&token).Error
+	err := getDB().Where("access_token = ?", accessToken).First(&token).Error
 	if err != nil {
 		return "", err
 	}
@@ -51,9 +41,8 @@ func GetUUIDByAccessToken(accessToken string) (string, error) {
 }
 
 func GetUserIDByAccessToken(accessToken string) (int64, error) {
-	db := database.MustGetDB()
 	var token model.Token
-	err := db.Where("access_token = ?", accessToken).First(&token).Error
+	err := getDB().Where("access_token = ?", accessToken).First(&token).Error
 	if err != nil {
 		return 0, err
 	}
@@ -61,9 +50,8 @@ func GetUserIDByAccessToken(accessToken string) (int64, error) {
 }
 
 func GetTokenByAccessToken(accessToken string) (*model.Token, error) {
-	db := database.MustGetDB()
 	var token model.Token
-	err := db.Where("access_token = ?", accessToken).First(&token).Error
+	err := getDB().Where("access_token = ?", accessToken).First(&token).Error
 	if err != nil {
 		return nil, err
 	}
@@ -71,19 +59,9 @@ func GetTokenByAccessToken(accessToken string) (*model.Token, error) {
 }
 
 func DeleteTokenByAccessToken(accessToken string) error {
-	db := database.MustGetDB()
-	err := db.Where("access_token = ?", accessToken).Delete(&model.Token{}).Error
-	if err != nil {
-		return err
-	}
-	return nil
+	return getDB().Where("access_token = ?", accessToken).Delete(&model.Token{}).Error
 }
 
 func DeleteTokenByUserId(userId int64) error {
-	db := database.MustGetDB()
-	err := db.Where("user_id = ?", userId).Delete(&model.Token{}).Error
-	if err != nil {
-		return err
-	}
-	return nil
+	return getDB().Where("user_id = ?", userId).Delete(&model.Token{}).Error
 }