mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-04-06 04:15:46 +08:00
1680b96b65
# Conflicts: # cmd/api/main.go # internal/api/a2r/api2rpc.go # internal/apiresp/resp.go # internal/msgtransfer/online_history_msg_handler.go # internal/push/consumer_init.go # pkg/common/db/controller/conversation.go # pkg/common/mw/gin.go # pkg/statistics/statistics.go
107 lines
3.8 KiB
Go
107 lines
3.8 KiB
Go
package controller
|
||
|
||
import (
|
||
"context"
|
||
"github.com/OpenIMSDK/Open-IM-Server/pkg/common/db/table/relation"
|
||
"github.com/OpenIMSDK/Open-IM-Server/pkg/errs"
|
||
"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
|
||
)
|
||
|
||
type UserDatabase interface {
|
||
//获取指定用户的信息 如有userID未找到 也返回错误
|
||
FindWithError(ctx context.Context, userIDs []string) (users []*relation.UserModel, err error)
|
||
//获取指定用户的信息 如有userID未找到 不返回错误
|
||
Find(ctx context.Context, userIDs []string) (users []*relation.UserModel, err error)
|
||
//插入多条 外部保证userID 不重复 且在db中不存在
|
||
Create(ctx context.Context, users []*relation.UserModel) (err error)
|
||
//更新（非零值） 外部保证userID存在
|
||
Update(ctx context.Context, users []*relation.UserModel) (err error)
|
||
//更新（零值） 外部保证userID存在
|
||
UpdateByMap(ctx context.Context, userID string, args map[string]interface{}) (err error)
|
||
//如果没找到，不返回错误
|
||
Page(ctx context.Context, pageNumber, showNumber int32) (users []*relation.UserModel, count int64, err error)
|
||
//只要有一个存在就为true
|
||
IsExist(ctx context.Context, userIDs []string) (exist bool, err error)
|
||
//获取所有用户ID
|
||
GetAllUserID(ctx context.Context) ([]string, error)
|
||
//函数内部先查询db中是否存在，存在则什么都不做；不存在则插入
|
||
InitOnce(ctx context.Context, users []*relation.UserModel) (err error)
|
||
}
|
||
|
||
type userDatabase struct {
|
||
userDB relation.UserModelInterface
|
||
}
|
||
|
||
func NewUserDatabase(userDB relation.UserModelInterface) UserDatabase {
|
||
return &userDatabase{userDB: userDB}
|
||
}
|
||
|
||
func (u *userDatabase) InitOnce(ctx context.Context, users []*relation.UserModel) (err error) {
|
||
userIDs := utils.Slice(users, func(e *relation.UserModel) string {
|
||
return e.UserID
|
||
})
|
||
result, err := u.userDB.Find(ctx, userIDs)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
miss := utils.SliceAnySub(users, result, func(e *relation.UserModel) string { return e.UserID })
|
||
if len(miss) > 0 {
|
||
_ = u.userDB.Create(ctx, miss)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 获取指定用户的信息 如有userID未找到 也返回错误
|
||
func (u *userDatabase) FindWithError(ctx context.Context, userIDs []string) (users []*relation.UserModel, err error) {
|
||
users, err = u.userDB.Find(ctx, userIDs)
|
||
if err != nil {
|
||
return
|
||
}
|
||
if len(users) != len(userIDs) {
|
||
err = errs.ErrRecordNotFound.Wrap()
|
||
}
|
||
return
|
||
}
|
||
|
||
// 获取指定用户的信息 如有userID未找到 不返回错误
|
||
func (u *userDatabase) Find(ctx context.Context, userIDs []string) (users []*relation.UserModel, err error) {
|
||
users, err = u.userDB.Find(ctx, userIDs)
|
||
return
|
||
}
|
||
|
||
// 插入多条 外部保证userID 不重复 且在db中不存在
|
||
func (u *userDatabase) Create(ctx context.Context, users []*relation.UserModel) (err error) {
|
||
return u.userDB.Create(ctx, users)
|
||
}
|
||
|
||
// 更新（非零值） 外部保证userID存在
|
||
func (u *userDatabase) Update(ctx context.Context, users []*relation.UserModel) (err error) {
|
||
return u.userDB.Update(ctx, users)
|
||
}
|
||
|
||
// 更新（零值） 外部保证userID存在
|
||
func (u *userDatabase) UpdateByMap(ctx context.Context, userID string, args map[string]interface{}) (err error) {
|
||
return u.userDB.UpdateByMap(ctx, userID, args)
|
||
}
|
||
|
||
// 获取，如果没找到，不返回错误
|
||
func (u *userDatabase) Page(ctx context.Context, pageNumber, showNumber int32) (users []*relation.UserModel, count int64, err error) {
|
||
return u.userDB.Page(ctx, pageNumber, showNumber)
|
||
}
|
||
|
||
// userIDs是否存在 只要有一个存在就为true
|
||
func (u *userDatabase) IsExist(ctx context.Context, userIDs []string) (exist bool, err error) {
|
||
users, err := u.userDB.Find(ctx, userIDs)
|
||
if err != nil {
|
||
return false, err
|
||
}
|
||
if len(users) > 0 {
|
||
return true, nil
|
||
}
|
||
return false, nil
|
||
}
|
||
|
||
func (u *userDatabase) GetAllUserID(ctx context.Context) (userIDs []string, err error) {
|
||
return u.userDB.GetAllUserID(ctx)
|
||
}
|