shanshanzhong
0f38b3fcd3
Some checks failed
Build docker and publish / build (20.15.1) (push) Failing after 6m39s
feat(database): 添加用户算法和盐字段的迁移脚本 fix(subscribe): 修复服务器用户列表缓存问题，临时禁用缓存 style(model): 清理用户模型注释，简化代码结构 chore: 删除无用脚本和测试文件 docs: 添加用户绑定流程文档 perf(login): 优化设备登录性能，添加设备缓存键 fix(unbind): 修复设备解绑时的缓存清理逻辑 refactor(verify): 简化邮箱验证逻辑，移除冗余代码 build(docker): 更新Dockerfile配置，使用scratch基础镜像
94 lines
2.9 KiB
Go
94 lines
2.9 KiB
Go
package user
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/perfect-panel/server/pkg/logger"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func (m *defaultUserModel) FindUserAuthMethods(ctx context.Context, userId int64) ([]*AuthMethods, error) {
|
|
var data []*AuthMethods
|
|
err := m.QueryNoCacheCtx(ctx, &data, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&AuthMethods{}).Where("user_id = ?", userId).Find(&data).Error
|
|
})
|
|
return data, err
|
|
}
|
|
|
|
func (m *defaultUserModel) FindUserAuthMethodByOpenID(ctx context.Context, method, openID string) (*AuthMethods, error) {
|
|
var data AuthMethods
|
|
err := m.QueryNoCacheCtx(ctx, &data, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&AuthMethods{}).Where("auth_type = ? AND auth_identifier = ?", method, openID).First(&data).Error
|
|
})
|
|
return &data, err
|
|
}
|
|
|
|
func (m *defaultUserModel) FindUserAuthMethodByPlatform(ctx context.Context, userId int64, platform string) (*AuthMethods, error) {
|
|
var data AuthMethods
|
|
err := m.QueryNoCacheCtx(ctx, &data, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&AuthMethods{}).Where("user_id = ? AND auth_type = ?", userId, platform).First(&data).Error
|
|
})
|
|
return &data, err
|
|
}
|
|
|
|
func (m *defaultUserModel) InsertUserAuthMethods(ctx context.Context, data *AuthMethods, tx ...*gorm.DB) error {
|
|
u, err := m.FindOne(ctx, data.UserId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
|
if len(tx) > 0 {
|
|
conn = tx[0]
|
|
}
|
|
if err = conn.Model(&AuthMethods{}).Create(data).Error; err != nil {
|
|
return err
|
|
}
|
|
return m.ClearUserCache(ctx, u)
|
|
})
|
|
}
|
|
|
|
func (m *defaultUserModel) UpdateUserAuthMethods(ctx context.Context, data *AuthMethods, tx ...*gorm.DB) error {
|
|
u, err := m.FindOne(ctx, data.UserId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
|
if len(tx) > 0 {
|
|
conn = tx[0]
|
|
}
|
|
err = conn.Model(&AuthMethods{}).Where("user_id = ? AND auth_type = ?", data.UserId, data.AuthType).Save(data).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return m.ClearUserCache(ctx, u)
|
|
})
|
|
}
|
|
|
|
func (m *defaultUserModel) DeleteUserAuthMethods(ctx context.Context, userId int64, platform string, tx ...*gorm.DB) error {
|
|
u, err := m.FindOne(ctx, userId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer func() {
|
|
if err = m.ClearUserCache(context.Background(), u); err != nil {
|
|
logger.Errorf("[UserModel] clear user cache failed: %v", err.Error())
|
|
}
|
|
}()
|
|
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
|
if len(tx) > 0 {
|
|
conn = tx[0]
|
|
}
|
|
return conn.Model(&AuthMethods{}).Where("user_id = ? AND auth_type = ?", userId, platform).Delete(&AuthMethods{}).Error
|
|
})
|
|
}
|
|
|
|
func (m *defaultUserModel) FindUserAuthMethodByUserId(ctx context.Context, method string, userId int64) (*AuthMethods, error) {
|
|
var data AuthMethods
|
|
err := m.QueryNoCacheCtx(ctx, &data, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&AuthMethods{}).Where("auth_type = ? AND user_id = ?", method, userId).First(&data).Error
|
|
})
|
|
return &data, err
|
|
}
|