blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-08-10 13:09:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

errCode

Browse Source
This commit is contained in:
wangchuxiao 2023-01-04 19:35:07 +08:00
parent b041cb6cbf
commit 7743801795
3 changed files with 31 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
internal/rpc/friend/friend.go
View File

@ -19,6 +19,7 @@ import (
"context" "context"
"errors" "errors"
"fmt" "fmt"
"gorm.io/gorm"
"net" "net"
"strconv" "strconv"
"strings" "strings"

39
pkg/common/db/mysql_model/im_mysql_model/group_model_k.go
View File

@ -2,7 +2,9 @@ package im_mysql_model
import ( import (
"Open_IM/pkg/common/db" "Open_IM/pkg/common/db"
"Open_IM/pkg/common/trace_log"
"Open_IM/pkg/utils" "Open_IM/pkg/utils"
"context"
"gorm.io/gorm" "gorm.io/gorm"
"time" "time"
) )
@ -26,27 +28,38 @@ type Group struct {
DB *gorm.DB `gorm:"-" json:"-"` DB *gorm.DB `gorm:"-" json:"-"`
} }
func (*Group) Create(groupList []*Group) error { func (*Group) Create(ctx context.Context, groupList []*Group) (err error) {
return utils.Wrap(db.DB.MysqlDB.DefaultGormDB().Create(&groupList).Error, "") defer trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), err, "groupList", groupList)
err = utils.Wrap(db.DB.MysqlDB.DefaultGormDB().Create(&groupList).Error, "")
return err
} }
func (*Group) Delete(groupIDList []string) error { func (*Group) Delete(ctx context.Context, groupIDList []string) (err error) {
return utils.Wrap(db.DB.MysqlDB.DefaultGormDB().Where("group_id in (?)", groupIDList).Delete(&Group{}).Error, "") defer trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), err, "groupIDList", groupIDList)
err = utils.Wrap(db.DB.MysqlDB.DefaultGormDB().Where("group_id in (?)", groupIDList).Delete(&Group{}).Error, "")
return err
} }
func (tb *Group) Get(groupIDs []string) ([]*Group, error) { func (*Group) UpdateByMap(ctx context.Context, groupID string, args map[string]interface{}) (err error) {
var ms []*Group defer trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), err, "groupID", groupID, "args", args)
return ms, utils.Wrap(db.DB.MysqlDB.DefaultGormDB().Where("group_id in (?)", groupIDs).Find(&ms).Error, "") err = utils.Wrap(db.DB.MysqlDB.DefaultGormDB().Where("group_id = ?", groupID).Updates(args).Error, "")
return err
} }
func (tb *Group) Update(groups []*Group) error { func (*Group) Update(ctx context.Context, groups []*Group) (err error) {
return utils.Wrap(utils.Wrap(db.DB.MysqlDB.DefaultGormDB().Updates(groups).Error, ""), "") defer trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), err, "groups", groups)
return utils.Wrap(db.DB.MysqlDB.DefaultGormDB().Updates(&groups).Error, "")
} }
func (*Group) Find(groupIDList []string) ([]*Group, error) { func (*Group) Find(ctx context.Context, groupIDList []string) (groupList []*Group, err error) {
return nil, nil defer trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), err, "groupIDList", groupIDList)
err = utils.Wrap(db.DB.MysqlDB.DefaultGormDB().Where("group_id in (?)", groupIDList).Find(&groupList).Error, "")
return groupList, err
} }
func (*Group) Take(groupID string) (*Group, error) { func (*Group) Take(ctx context.Context, groupID string) (group *Group, err error) {
return nil, nil defer trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), err, "groupID", groupID)
group = &Group{}
err = utils.Wrap(db.DB.MysqlDB.DefaultGormDB().Where("group_id = ?", groupID).Take(group).Error, "")
return group, err
} }

7
pkg/common/db/mysql_model/im_mysql_model/user_black_list_model.go
View File

@ -8,10 +8,11 @@ import (
"time" "time"
) )
func InsertInToUserBlackList(ctx context.Context, black db.Black) error { func InsertInToUserBlackList(ctx context.Context, black db.Black) (err error) {
defer trace_log.SetContextInfo(ctx) defer trace_log.SetContextInfo(ctx, utils.GetSelfFuncName(), err, "black", black)
black.CreateTime = time.Now() black.CreateTime = time.Now()
return db.DB.MysqlDB.DefaultGormDB().Table("blacks").Create(black).Error err = db.DB.MysqlDB.DefaultGormDB().Table("blacks").Create(black).Error
return err
} }
// type Black struct { // type Black struct {