first commit
This commit is contained in:
@@ -0,0 +1,61 @@
|
||||
// Package user 提供用户数据访问。所有查询强制按 mini_id 过滤,
|
||||
// 从数据层保证各小程序用户体系相互独立。
|
||||
package user
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"gorm.io/gorm"
|
||||
|
||||
"portal-mini-go/internal/model"
|
||||
"portal-mini-go/internal/pkg/errors"
|
||||
)
|
||||
|
||||
type Repository struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
func New(db *gorm.DB) *Repository { return &Repository{db: db} }
|
||||
|
||||
// UpsertByOpenID 按 mini_id + openid 创建或返回已有用户。
|
||||
func (r *Repository) UpsertByOpenID(ctx context.Context, miniID uint64, openID, unionID, nickname, avatar string) (*model.User, error) {
|
||||
var u model.User
|
||||
err := r.db.WithContext(ctx).
|
||||
Where("mini_id = ? AND openid = ?", miniID, openID).
|
||||
Attrs(model.User{MiniID: miniID, OpenID: openID, UnionID: unionID, Nickname: nickname, Avatar: avatar}).
|
||||
FirstOrCreate(&u).Error
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(50000, "upsert user", err)
|
||||
}
|
||||
return &u, nil
|
||||
}
|
||||
|
||||
// GetByID 按 mini_id + id 查询用户,跨小程序查询直接返回未找到。
|
||||
func (r *Repository) GetByID(ctx context.Context, miniID uint64, id uint64) (*model.User, error) {
|
||||
var u model.User
|
||||
err := r.db.WithContext(ctx).
|
||||
Where("mini_id = ? AND id = ?", miniID, id).
|
||||
First(&u).Error
|
||||
if err != nil {
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, errors.ErrNotFound
|
||||
}
|
||||
return nil, errors.Wrap(50000, "query user", err)
|
||||
}
|
||||
return &u, nil
|
||||
}
|
||||
|
||||
// UpdateProfile 更新用户资料,限定在 mini_id 范围内。
|
||||
func (r *Repository) UpdateProfile(ctx context.Context, miniID uint64, id uint64, nickname, avatar string) error {
|
||||
res := r.db.WithContext(ctx).
|
||||
Model(&model.User{}).
|
||||
Where("mini_id = ? AND id = ?", miniID, id).
|
||||
Updates(map[string]interface{}{"nickname": nickname, "avatar": avatar})
|
||||
if res.Error != nil {
|
||||
return errors.Wrap(50000, "update user", res.Error)
|
||||
}
|
||||
if res.RowsAffected == 0 {
|
||||
return errors.ErrNotFound
|
||||
}
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user