0bcd9336c4
- Refactored multiple service and repository methods to accept context as a parameter, enhancing consistency and enabling better control over request lifecycles. - Updated handlers to utilize context in method calls, improving error handling and performance. - Cleaned up Dockerfile by removing unnecessary whitespace.
72 lines
2.2 KiB
Go
72 lines
2.2 KiB
Go
package repository
|
|
|
|
import (
|
|
"carrotskin/internal/model"
|
|
"context"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// tokenRepository TokenRepository的实现
|
|
type tokenRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewTokenRepository 创建TokenRepository实例
|
|
func NewTokenRepository(db *gorm.DB) TokenRepository {
|
|
return &tokenRepository{db: db}
|
|
}
|
|
|
|
func (r *tokenRepository) Create(ctx context.Context, token *model.Token) error {
|
|
return r.db.WithContext(ctx).Create(token).Error
|
|
}
|
|
|
|
func (r *tokenRepository) FindByAccessToken(ctx context.Context, accessToken string) (*model.Token, error) {
|
|
var token model.Token
|
|
err := r.db.WithContext(ctx).Where("access_token = ?", accessToken).First(&token).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &token, nil
|
|
}
|
|
|
|
func (r *tokenRepository) GetByUserID(ctx context.Context, userId int64) ([]*model.Token, error) {
|
|
var tokens []*model.Token
|
|
err := r.db.WithContext(ctx).Where("user_id = ?", userId).Find(&tokens).Error
|
|
return tokens, err
|
|
}
|
|
|
|
func (r *tokenRepository) GetUUIDByAccessToken(ctx context.Context, accessToken string) (string, error) {
|
|
var token model.Token
|
|
err := r.db.WithContext(ctx).Select("profile_id").Where("access_token = ?", accessToken).First(&token).Error
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return token.ProfileId, nil
|
|
}
|
|
|
|
func (r *tokenRepository) GetUserIDByAccessToken(ctx context.Context, accessToken string) (int64, error) {
|
|
var token model.Token
|
|
err := r.db.WithContext(ctx).Select("user_id").Where("access_token = ?", accessToken).First(&token).Error
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return token.UserID, nil
|
|
}
|
|
|
|
func (r *tokenRepository) DeleteByAccessToken(ctx context.Context, accessToken string) error {
|
|
return r.db.WithContext(ctx).Where("access_token = ?", accessToken).Delete(&model.Token{}).Error
|
|
}
|
|
|
|
func (r *tokenRepository) DeleteByUserID(ctx context.Context, userId int64) error {
|
|
return r.db.WithContext(ctx).Where("user_id = ?", userId).Delete(&model.Token{}).Error
|
|
}
|
|
|
|
func (r *tokenRepository) BatchDelete(ctx context.Context, accessTokens []string) (int64, error) {
|
|
if len(accessTokens) == 0 {
|
|
return 0, nil
|
|
}
|
|
result := r.db.WithContext(ctx).Where("access_token IN ?", accessTokens).Delete(&model.Token{})
|
|
return result.RowsAffected, result.Error
|
|
}
|