98 lines
2.7 KiB
Go
98 lines
2.7 KiB
Go
|
package dao
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type User struct {
|
|||
|
|
ID int32 `gorm:"primarykey;type:int"`
|
|||
|
|
CreatedAt time.Time `gorm:"type:datetime(0)"`
|
|||
|
|
UpdatedAt time.Time `gorm:"type:datetime(0)"`
|
|||
|
|
DeletedAt gorm.DeletedAt `gorm:"index;type:datetime(0)"`
|
|||
|
|
Username string `gorm:"type:varchar(64);default null;comment:'用户名称'"`
|
|||
|
|
OpenId string `gorm:"type:varchar(255);default null;comment:'用户唯一标识'"`
|
|||
|
|
AvatarId string `gorm:"type:varchar(32);default null;comment:'头像图片uuid'"`
|
|||
|
|
Tags string `gorm:"type:json;default null;comment:'用户标签,json格式'"`
|
|||
|
|
Persona string `gorm:"type:json;default null;comment:'用户个性化数据,json格式'"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (User) TableName() string { return "user" }
|
|||
|
|
|
|||
|
|
func User_WX_Insert(data *User) (int32, error) {
|
|||
|
|
if err := s.db.Create(data).Error; err != nil {
|
|||
|
|
return 0, err
|
|||
|
|
}
|
|||
|
|
return data.ID, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func User_WX_Query(openId string) (*User, error) {
|
|||
|
|
var data User
|
|||
|
|
if err := s.db.Where("open_id = ?", openId).First(&data).Error; err != nil {
|
|||
|
|
if err == gorm.ErrRecordNotFound {
|
|||
|
|
return nil, nil
|
|||
|
|
}
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
return &data, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func User_Query(userId int32) (*User, error) {
|
|||
|
|
var data User
|
|||
|
|
if err := s.db.Where("id = ?", userId).First(&data).Error; err != nil {
|
|||
|
|
if err == gorm.ErrRecordNotFound {
|
|||
|
|
return nil, nil
|
|||
|
|
}
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
return &data, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 修改用户名称
|
|||
|
|
func User_Change_Username(userId int32, username string) error {
|
|||
|
|
user := &User{ID: userId}
|
|||
|
|
return s.db.Model(&user).Update("username", username).Error
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 修改用户头像
|
|||
|
|
func User_Change_AvatarId(userId int32, avatarId string) error {
|
|||
|
|
user := &User{ID: userId}
|
|||
|
|
return s.db.Model(&user).Update("avatar_id", avatarId).Error
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 查询用户信息列表，如果不存在，则设置为nil
|
|||
|
|
func User_Query_List(userIds []int32) (map[int32]*User, error) {
|
|||
|
|
var list []*User
|
|||
|
|
if err := s.db.Where("id in (?)", userIds).Find(&list).Error; err != nil {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
reply := make(map[int32]*User)
|
|||
|
|
for _, data := range list {
|
|||
|
|
reply[data.ID] = data
|
|||
|
|
}
|
|||
|
|
return reply, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type UserBaseInfo struct {
|
|||
|
|
ID int32 `gorm:"column:id"`
|
|||
|
|
Username string `gorm:"column:username"`
|
|||
|
|
AvatarId string `gorm:"column:avatar_id"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 批量查询用户的名称和头像信息
|
|||
|
|
func User_Query_List_BaseInfo(userIds []int32) (map[int32]*UserBaseInfo, error) {
|
|||
|
|
var list []*UserBaseInfo
|
|||
|
|
if err := s.db.Model(&User{}).Where("id in (?)", userIds).Select("id, username, avatar_id").Find(&list).Error; err != nil && err != gorm.ErrRecordNotFound {
|
|||
|
|
return nil, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
m := make(map[int32]*UserBaseInfo)
|
|||
|
|
for _, info := range list {
|
|||
|
|
m[info.ID] = info
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return m, nil
|
|||
|
|
}
|