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.
92 lines
2.3 KiB
92 lines
2.3 KiB
package model
|
|
|
|
import (
|
|
"Open_IM/pkg/common/db/cache"
|
|
"Open_IM/pkg/common/db/mongo"
|
|
"Open_IM/pkg/common/db/mysql"
|
|
"Open_IM/pkg/common/trace_log"
|
|
"Open_IM/pkg/utils"
|
|
"context"
|
|
"encoding/json"
|
|
//"github.com/dtm-labs/rockscache"
|
|
"gorm.io/gorm"
|
|
//"time"
|
|
)
|
|
|
|
type GroupModel struct {
|
|
db *mysql.Group
|
|
cache *cache.GroupCache
|
|
mongo *mongo.Client
|
|
}
|
|
|
|
|
|
func NewGroupModel() {
|
|
var groupModel GroupModel
|
|
redisClient := cache.InitRedis()
|
|
rdb := cache.NewRedisClient(redisClient)
|
|
groupModel.db = mysql.NewGroupDB()
|
|
//mgo := mongo.In()
|
|
}
|
|
|
|
func (g *GroupModel) Find(ctx context.Context, groupIDs []string) (groups []*mysql.Group, err error) {
|
|
g.cache.Client.
|
|
for _, groupID := range groupIDs {
|
|
group, err := g.getGroupInfoFromCache(ctx, groupID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
groups = append(groups, group)
|
|
}
|
|
return groups, nil
|
|
}
|
|
|
|
func (g *GroupModel) Create(ctx context.Context, groups []*mysql.Group) error {
|
|
return g.db.Create(ctx, groups)
|
|
}
|
|
|
|
func (g *GroupModel) Delete(ctx context.Context, groupIDs []string) error {
|
|
err := g.db.DB.Transaction(func(tx *gorm.DB) error {
|
|
if err := g.db.Delete(ctx, groupIDs, tx); err != nil {
|
|
return err
|
|
}
|
|
if err := g.deleteGroupsInCache(ctx, groupIDs); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
})
|
|
return err
|
|
}
|
|
|
|
func (g *GroupModel) deleteGroupsInCache(ctx context.Context, groupIDs []string) error {
|
|
for _, groupID := range groupIDs {
|
|
if err := g.weakRc.Cache.TagAsDeleted(g.getGroupCacheKey(groupID)); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (g *GroupModel) getGroupInfoFromCache(ctx context.Context, groupID string) (groupInfo *mysql.Group, err error) {
|
|
getGroupInfo := func() (string, error) {
|
|
groupInfo, err := mysql.GetGroupInfoByGroupID(groupID)
|
|
if err != nil {
|
|
return "", utils.Wrap(err, "")
|
|
}
|
|
bytes, err := json.Marshal(groupInfo)
|
|
if err != nil {
|
|
return "", utils.Wrap(err, "")
|
|
}
|
|
return string(bytes), nil
|
|
}
|
|
groupInfo = &mysql.Group{}
|
|
defer func() {
|
|
trace_log.SetCtxDebug(ctx, utils.GetFuncName(1), err, "groupID", groupID, "groupInfo", groupInfo)
|
|
}()
|
|
groupInfoStr, err := g.weakRc.Cache.Fetch(groupInfoCache+groupID, GroupExpireTime, getGroupInfo)
|
|
if err != nil {
|
|
return nil, utils.Wrap(err, "")
|
|
}
|
|
err = json.Unmarshal([]byte(groupInfoStr), groupInfo)
|
|
return groupInfo, utils.Wrap(err, "")
|
|
}
|