mirror of
https://github.com/openimsdk/open-im-server.git
synced 2026-04-28 14:29:19 +08:00
withchao
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)
|
|
}
|