You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
136 lines
5.5 KiB
136 lines
5.5 KiB
package controller
|
|
|
|
import (
|
|
"Open_IM/pkg/common/constant"
|
|
"Open_IM/pkg/common/db/relation"
|
|
relation2 "Open_IM/pkg/common/db/table/relation"
|
|
"context"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type UserInterface interface {
|
|
//获取指定用户的信息 如果有记录未找到 也返回错误
|
|
Find(ctx context.Context, userIDs []string) (users []*relation2.UserModel, err error)
|
|
//插入
|
|
Create(ctx context.Context, users []*relation2.UserModel) error
|
|
//更新
|
|
Update(ctx context.Context, users []*relation2.UserModel) (err error)
|
|
//更新带零值的
|
|
UpdateByMap(ctx context.Context, userID string, args map[string]interface{}) (err error)
|
|
//通过名字搜索
|
|
GetByName(ctx context.Context, userName string, showNumber, pageNumber int32) (users []*relation2.UserModel, count int64, err error)
|
|
//通过名字和id搜索
|
|
GetByNameAndID(ctx context.Context, content string, showNumber, pageNumber int32) (users []*relation2.UserModel, count int64, err error)
|
|
//获取，如果没找到，不不返回错误
|
|
Get(ctx context.Context, showNumber, pageNumber int32) (users []*relation2.UserModel, count int64, err error)
|
|
//userIDs是否存在 只要有一个存在就为true
|
|
IsExist(ctx context.Context, userIDs []string) (exist bool, err error)
|
|
}
|
|
|
|
type UserController struct {
|
|
database UserDatabaseInterface
|
|
}
|
|
|
|
func (u *UserController) Find(ctx context.Context, userIDs []string) (users []*relation2.UserModel, err error) {
|
|
return u.database.Find(ctx, userIDs)
|
|
}
|
|
func (u *UserController) Create(ctx context.Context, users []*relation2.UserModel) error {
|
|
return u.database.Create(ctx, users)
|
|
}
|
|
|
|
func (u *UserController) Update(ctx context.Context, users []*relation2.UserModel) (err error) {
|
|
return u.database.Update(ctx, users)
|
|
}
|
|
func (u *UserController) UpdateByMap(ctx context.Context, userID string, args map[string]interface{}) (err error) {
|
|
return u.database.UpdateByMap(ctx, userID, args)
|
|
}
|
|
|
|
func (u *UserController) GetByName(ctx context.Context, userName string, showNumber, pageNumber int32) (users []*relation2.UserModel, count int64, err error) {
|
|
return u.database.GetByName(ctx, userName, showNumber, pageNumber)
|
|
}
|
|
|
|
func (u *UserController) GetByNameAndID(ctx context.Context, content string, showNumber, pageNumber int32) (users []*relation2.UserModel, count int64, err error) {
|
|
return u.database.GetByNameAndID(ctx, content, showNumber, pageNumber)
|
|
}
|
|
|
|
func (u *UserController) Get(ctx context.Context, showNumber, pageNumber int32) (users []*relation2.UserModel, count int64, err error) {
|
|
return u.database.Get(ctx, showNumber, pageNumber)
|
|
}
|
|
|
|
func (u *UserController) IsExist(ctx context.Context, userIDs []string) (exist bool, err error) {
|
|
return u.IsExist(ctx, userIDs)
|
|
}
|
|
func NewUserController(db *gorm.DB) *UserController {
|
|
controller := &UserController{database: newUserDatabase(db)}
|
|
return controller
|
|
}
|
|
|
|
type UserDatabaseInterface interface {
|
|
Find(ctx context.Context, userIDs []string) (users []*relation2.UserModel, err error)
|
|
Create(ctx context.Context, users []*relation2.UserModel) error
|
|
Update(ctx context.Context, users []*relation2.UserModel) (err error)
|
|
UpdateByMap(ctx context.Context, userID string, args map[string]interface{}) (err error)
|
|
GetByName(ctx context.Context, userName string, showNumber, pageNumber int32) (users []*relation2.UserModel, count int64, err error)
|
|
GetByNameAndID(ctx context.Context, content string, showNumber, pageNumber int32) (users []*relation2.UserModel, count int64, err error)
|
|
Get(ctx context.Context, showNumber, pageNumber int32) (users []*relation2.UserModel, count int64, err error)
|
|
IsExist(ctx context.Context, userIDs []string) (exist bool, err error)
|
|
}
|
|
|
|
type UserDatabase struct {
|
|
sqlDB *relation.UserGorm
|
|
}
|
|
|
|
func newUserDatabase(db *gorm.DB) *UserDatabase {
|
|
sqlDB := relation.NewUserGorm(db)
|
|
database := &UserDatabase{
|
|
sqlDB: sqlDB,
|
|
}
|
|
return database
|
|
}
|
|
|
|
// 获取指定用户的信息 如果有记录未找到 也返回错误
|
|
func (u *UserDatabase) Find(ctx context.Context, userIDs []string) (users []*relation2.UserModel, err error) {
|
|
users, err = u.sqlDB.Find(ctx, userIDs)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if len(users) != len(userIDs) {
|
|
err = constant.ErrRecordNotFound.Wrap()
|
|
}
|
|
return
|
|
}
|
|
|
|
func (u *UserDatabase) Create(ctx context.Context, users []*relation2.UserModel) (err error) {
|
|
return u.sqlDB.Create(ctx, users)
|
|
}
|
|
|
|
func (u *UserDatabase) Update(ctx context.Context, users []*relation2.UserModel) (err error) {
|
|
return u.sqlDB.Update(ctx, users)
|
|
}
|
|
func (u *UserDatabase) UpdateByMap(ctx context.Context, userID string, args map[string]interface{}) (err error) {
|
|
return u.sqlDB.UpdateByMap(ctx, userID, args)
|
|
}
|
|
func (u *UserDatabase) GetByName(ctx context.Context, userName string, showNumber, pageNumber int32) (users []*relation2.UserModel, count int64, err error) {
|
|
return u.sqlDB.GetByName(ctx, userName, showNumber, pageNumber)
|
|
}
|
|
func (u *UserDatabase) GetByNameAndID(ctx context.Context, content string, showNumber, pageNumber int32) (users []*relation2.UserModel, count int64, err error) {
|
|
return u.sqlDB.GetByNameAndID(ctx, content, showNumber, pageNumber)
|
|
}
|
|
|
|
// 获取，如果没找到，不返回错误
|
|
func (u *UserDatabase) Get(ctx context.Context, showNumber, pageNumber int32) (users []*relation2.UserModel, count int64, err error) {
|
|
return u.sqlDB.Get(ctx, showNumber, pageNumber)
|
|
}
|
|
|
|
// userIDs是否存在 只要有一个存在就为true
|
|
func (u *UserDatabase) IsExist(ctx context.Context, userIDs []string) (exist bool, err error) {
|
|
users, err := u.sqlDB.Find(ctx, userIDs)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if len(users) > 0 {
|
|
return true, nil
|
|
}
|
|
return false, nil
|
|
}
|