errcode

pkg/common/db/unrelation/mongo.go

@@ -61,20 +61,8 @@ func (m *Mongo) GetClient() *mongo.Client {
 	return m.db
 }
 
-func (m *Mongo) CreateTagIndex() {
-	if err := m.createMongoIndex(unrelation.CSendLog, false, "send_id", "-send_time"); err != nil {
-		panic(err.Error() + " index create failed " + unrelation.CSendLog + " send_id, -send_time")
-	}
-	if err := m.createMongoIndex(unrelation.CTag, false, "user_id", "-create_time"); err != nil {
-		panic(err.Error() + "index create failed " + unrelation.CTag + " user_id, -create_time")
-	}
-	if err := m.createMongoIndex(unrelation.CTag, true, "tag_id"); err != nil {
-		panic(err.Error() + "index create failed " + unrelation.CTag + " tag_id")
-	}
-}
-
 func (m *Mongo) CreateMsgIndex() {
-	if err := m.createMongoIndex(unrelation.CChat, false, "uid"); err != nil {
+	if err := m.createMongoIndex(unrelation, false, "uid"); err != nil {
 		fmt.Println(err.Error() + " index create failed " + unrelation.CChat + " uid, please create index by yourself in field uid")
 	}
 }
@@ -88,21 +76,9 @@ func (m *Mongo) CreateSuperGroupIndex() {
 	}
 }
 
-func (m *Mongo) CreateWorkMomentIndex() {
-	if err := m.createMongoIndex(unrelation.CWorkMoment, true, "-create_time", "work_moment_id"); err != nil {
-		panic(err.Error() + "index create failed " + unrelation.CWorkMoment + " -create_time, work_moment_id")
-	}
-	if err := m.createMongoIndex(unrelation.CWorkMoment, true, "work_moment_id"); err != nil {
-		panic(err.Error() + "index create failed " + unrelation.CWorkMoment + " work_moment_id ")
-	}
-	if err := m.createMongoIndex(unrelation.CWorkMoment, false, "user_id", "-create_time"); err != nil {
-		panic(err.Error() + "index create failed " + unrelation.CWorkMoment + "user_id, -create_time")
-	}
-}
-
 func (m *Mongo) CreateExtendMsgSetIndex() {
 	if err := m.createMongoIndex(unrelation.CExtendMsgSet, true, "-create_time", "work_moment_id"); err != nil {
-		panic(err.Error() + "index create failed " + unrelation.CWorkMoment + " -create_time, work_moment_id")
+		panic(err.Error() + "index create failed " + unrelation.CExtendMsgSet + " -create_time, work_moment_id")
 	}
 }
 

pkg/common/db/unrelation/mongo_model.go

@@ -69,7 +69,7 @@ package unrelation
 //	return nil
 //}
 
-//func (d *db.DataBases) ReplaceMsgByIndex(suffixUserID string, msg *open_im_sdk.MsgData, operationID string, seqIndex int) error {
+//func (d *db.DataBases) ReplaceMsgByIndex(suffixUserID string, msg *sdkws.MsgData, operationID string, seqIndex int) error {
 //	log.NewInfo(operationID, utils.GetSelfFuncName(), suffixUserID, *msg)
 //	ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second)
 //	c := d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cChat)
@@ -90,7 +90,7 @@ package unrelation
 //	return nil
 //}
 
-//func (d *db.DataBases) ReplaceMsgBySeq(uid string, msg *open_im_sdk.MsgData, operationID string) error {
+//func (d *db.DataBases) ReplaceMsgBySeq(uid string, msg *sdkws.MsgData, operationID string) error {
 //	log.NewInfo(operationID, utils.GetSelfFuncName(), uid, *msg)
 //	ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second)
 //	c := d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cChat)
@@ -122,7 +122,7 @@ package unrelation
 //	return err
 //}
 //
-//func (d *db.DataBases) GetMsgBySeqList(uid string, seqList []uint32, operationID string) (seqMsg []*open_im_sdk.MsgData, err error) {
+//func (d *db.DataBases) GetMsgBySeqList(uid string, seqList []uint32, operationID string) (seqMsg []*sdkws.MsgData, err error) {
 //	log.NewInfo(operationID, utils.GetSelfFuncName(), uid, seqList)
 //	var hasSeqList []uint32
 //	singleCount := 0
@@ -153,7 +153,7 @@ package unrelation
 //		}
 //		singleCount = 0
 //		for i := 0; i < len(sChat.Msg); i++ {
-//			msg := new(open_im_sdk.MsgData)
+//			msg := new(sdkws.MsgData)
 //			if err = proto.Unmarshal(sChat.Msg[i].Msg, msg); err != nil {
 //				log.NewError(operationID, "Unmarshal err", seqUid, value, uid, seqList, err.Error())
 //				return nil, err
@@ -217,11 +217,11 @@ package unrelation
 //	}
 //	for i, msg := range userChat.Msg {
 //		if i <= index {
-//			msgPb := &open_im_sdk.MsgData{}
+//			msgPb := &sdkws.MsgData{}
 //			if err = proto.Unmarshal(msg.Msg, msgPb); err != nil {
 //				continue
 //			}
-//			newMsgPb := &open_im_sdk.MsgData{Seq: msgPb.Seq}
+//			newMsgPb := &sdkws.MsgData{Seq: msgPb.Seq}
 //			bytes, err := proto.Marshal(newMsgPb)
 //			if err != nil {
 //				continue
@@ -235,7 +235,7 @@ package unrelation
 //	return replaceMaxSeq, err
 //}
 //
-//func (d *db.DataBases) GetNewestMsg(ID string) (msg *open_im_sdk.MsgData, err error) {
+//func (d *db.DataBases) GetNewestMsg(ID string) (msg *sdkws.MsgData, err error) {
 //	ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second)
 //	c := d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cChat)
 //	regex := fmt.Sprintf("^%s", ID)
@@ -251,7 +251,7 @@ package unrelation
 //	}
 //	if len(userChats) > 0 {
 //		if len(userChats[0].Msg) > 0 {
-//			msgPb := &open_im_sdk.MsgData{}
+//			msgPb := &sdkws.MsgData{}
 //			err = proto.Unmarshal(userChats[0].Msg[len(userChats[0].Msg)-1].Msg, msgPb)
 //			if err != nil {
 //				return nil, utils.Wrap(err, "")
@@ -263,7 +263,7 @@ package unrelation
 //	return nil, nil
 //}
 //
-//func (d *db.DataBases) GetOldestMsg(ID string) (msg *open_im_sdk.MsgData, err error) {
+//func (d *db.DataBases) GetOldestMsg(ID string) (msg *sdkws.MsgData, err error) {
 //	ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second)
 //	c := d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cChat)
 //	regex := fmt.Sprintf("^%s", ID)
@@ -288,7 +288,7 @@ package unrelation
 //		if len(oldestMsg) == 0 {
 //			oldestMsg = userChats[0].Msg[len(userChats[0].Msg)-1].Msg
 //		}
-//		msgPb := &open_im_sdk.MsgData{}
+//		msgPb := &sdkws.MsgData{}
 //		err = proto.Unmarshal(oldestMsg, msgPb)
 //		if err != nil {
 //			return nil, utils.Wrap(err, "")
@@ -298,7 +298,7 @@ package unrelation
 //	return nil, nil
 //}
 //
-//func (d *db.DataBases) GetMsgBySeqListMongo2(uid string, seqList []uint32, operationID string) (seqMsg []*open_im_sdk.MsgData, err error) {
+//func (d *db.DataBases) GetMsgBySeqListMongo2(uid string, seqList []uint32, operationID string) (seqMsg []*sdkws.MsgData, err error) {
 //	var hasSeqList []uint32
 //	singleCount := 0
 //	ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second)
@@ -325,7 +325,7 @@ package unrelation
 //		}
 //		singleCount = 0
 //		for i := 0; i < len(sChat.Msg); i++ {
-//			msg := new(open_im_sdk.MsgData)
+//			msg := new(sdkws.MsgData)
 //			if err = proto.Unmarshal(sChat.Msg[i].Msg, msg); err != nil {
 //				log.NewError(operationID, "Unmarshal err", seqUid, value, uid, seqList, err.Error())
 //				return nil, err
@@ -349,7 +349,7 @@ package unrelation
 //	}
 //	return seqMsg, nil
 //}
-//func (d *db.DataBases) GetSuperGroupMsgBySeqListMongo(groupID string, seqList []uint32, operationID string) (seqMsg []*open_im_sdk.MsgData, err error) {
+//func (d *db.DataBases) GetSuperGroupMsgBySeqListMongo(groupID string, seqList []uint32, operationID string) (seqMsg []*sdkws.MsgData, err error) {
 //	var hasSeqList []uint32
 //	singleCount := 0
 //	ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second)
@@ -376,7 +376,7 @@ package unrelation
 //		}
 //		singleCount = 0
 //		for i := 0; i < len(sChat.Msg); i++ {
-//			msg := new(open_im_sdk.MsgData)
+//			msg := new(sdkws.MsgData)
 //			if err = proto.Unmarshal(sChat.Msg[i].Msg, msg); err != nil {
 //				log.NewError(operationID, "Unmarshal err", seqUid, value, groupID, seqList, err.Error())
 //				return nil, err
@@ -401,7 +401,7 @@ package unrelation
 //	return seqMsg, nil
 //}
 //
-//func (d *db.DataBases) GetMsgAndIndexBySeqListInOneMongo2(suffixUserID string, seqList []uint32, operationID string) (seqMsg []*open_im_sdk.MsgData, indexList []int, unexistSeqList []uint32, err error) {
+//func (d *db.DataBases) GetMsgAndIndexBySeqListInOneMongo2(suffixUserID string, seqList []uint32, operationID string) (seqMsg []*sdkws.MsgData, indexList []int, unexistSeqList []uint32, err error) {
 //	ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second)
 //	c := d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cChat)
 //	sChat := UserChat{}
@@ -412,7 +412,7 @@ package unrelation
 //	singleCount := 0
 //	var hasSeqList []uint32
 //	for i := 0; i < len(sChat.Msg); i++ {
-//		msg := new(open_im_sdk.MsgData)
+//		msg := new(sdkws.MsgData)
 //		if err = proto.Unmarshal(sChat.Msg[i].Msg, msg); err != nil {
 //			log.NewError(operationID, "Unmarshal err", msg.String(), err.Error())
 //			return nil, nil, nil, err
@@ -436,18 +436,18 @@ package unrelation
 //	return seqMsg, indexList, unexistSeqList, nil
 //}
 //
-//func genExceptionMessageBySeqList(seqList []uint32) (exceptionMsg []*open_im_sdk.MsgData) {
+//func genExceptionMessageBySeqList(seqList []uint32) (exceptionMsg []*sdkws.MsgData) {
 //	for _, v := range seqList {
-//		msg := new(open_im_sdk.MsgData)
+//		msg := new(sdkws.MsgData)
 //		msg.Seq = v
 //		exceptionMsg = append(exceptionMsg, msg)
 //	}
 //	return exceptionMsg
 //}
 //
-//func genExceptionSuperGroupMessageBySeqList(seqList []uint32, groupID string) (exceptionMsg []*open_im_sdk.MsgData) {
+//func genExceptionSuperGroupMessageBySeqList(seqList []uint32, groupID string) (exceptionMsg []*sdkws.MsgData) {
 //	for _, v := range seqList {
-//		msg := new(open_im_sdk.MsgData)
+//		msg := new(sdkws.MsgData)
 //		msg.Seq = v
 //		msg.GroupID = groupID
 //		msg.SessionType = constant.SuperGroupChatType
@@ -598,13 +598,7 @@ package unrelation
 //	return utils.Md5(tagName + userID + strconv.Itoa(rand.Int()) + time.Now().String())
 //}
 
-//func generateWorkMomentID(userID string) string {
-//	return utils.Md5(userID + strconv.Itoa(rand.Int()) + time.Now().String())
-//}
 
-//func generateWorkMomentCommentID(workMomentID string) string {
-//	return utils.Md5(workMomentID + strconv.Itoa(rand.Int()) + time.Now().String())
-//}
 
 //func getCurrentTimestampByMill() int64 {
 //	return time.Now().UnixNano() / 1e6

pkg/common/db/unrelation/msg.go

@@ -121,7 +121,7 @@ func (m *MsgMongoDriver) ReplaceMsgByIndex(ctx context.Context, suffixUserID str
 	return nil
 }
 
-func (d *db.DataBases) ReplaceMsgBySeq(uid string, msg *open_im_sdk.MsgData, operationID string) error {
+func (d *db.DataBases) ReplaceMsgBySeq(uid string, msg *sdkws.MsgData, operationID string) error {
 	log.NewInfo(operationID, utils.GetSelfFuncName(), uid, *msg)
 	ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second)
 	c := d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cChat)
@@ -153,7 +153,7 @@ func (d *db.DataBases) UpdateOneMsgList(msg *UserChat) error {
 	return err
 }
 
-func (d *db.DataBases) GetMsgBySeqList(uid string, seqList []uint32, operationID string) (seqMsg []*open_im_sdk.MsgData, err error) {
+func (d *db.DataBases) GetMsgBySeqList(uid string, seqList []uint32, operationID string) (seqMsg []*sdkws.MsgData, err error) {
 	log.NewInfo(operationID, utils.GetSelfFuncName(), uid, seqList)
 	var hasSeqList []uint32
 	singleCount := 0
@@ -184,7 +184,7 @@ func (d *db.DataBases) GetMsgBySeqList(uid string, seqList []uint32, operationID
 		}
 		singleCount = 0
 		for i := 0; i < len(sChat.Msg); i++ {
-			msg := new(open_im_sdk.MsgData)
+			msg := new(sdkws.MsgData)
 			if err = proto.Unmarshal(sChat.Msg[i].Msg, msg); err != nil {
 				log.NewError(operationID, "Unmarshal err", seqUid, value, uid, seqList, err.Error())
 				return nil, err
@@ -251,11 +251,11 @@ func (d *db.DataBases) ReplaceMsgToBlankByIndex(suffixID string, index int) (rep
 	}
 	for i, msg := range userChat.Msg {
 		if i <= index {
-			msgPb := &open_im_sdk.MsgData{}
+			msgPb := &sdkws.MsgData{}
 			if err = proto.Unmarshal(msg.Msg, msgPb); err != nil {
 				continue
 			}
-			newMsgPb := &open_im_sdk.MsgData{Seq: msgPb.Seq}
+			newMsgPb := &sdkws.MsgData{Seq: msgPb.Seq}
 			bytes, err := proto.Marshal(newMsgPb)
 			if err != nil {
 				continue
@@ -269,7 +269,7 @@ func (d *db.DataBases) ReplaceMsgToBlankByIndex(suffixID string, index int) (rep
 	return replaceMaxSeq, err
 }
 
-func (d *db.DataBases) GetNewestMsg(ID string) (msg *open_im_sdk.MsgData, err error) {
+func (d *db.DataBases) GetNewestMsg(ID string) (msg *sdkws.MsgData, err error) {
 	ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second)
 	c := d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cChat)
 	regex := fmt.Sprintf("^%s", ID)
@@ -285,7 +285,7 @@ func (d *db.DataBases) GetNewestMsg(ID string) (msg *open_im_sdk.MsgData, err er
 	}
 	if len(userChats) > 0 {
 		if len(userChats[0].Msg) > 0 {
-			msgPb := &open_im_sdk.MsgData{}
+			msgPb := &sdkws.MsgData{}
 			err = proto.Unmarshal(userChats[0].Msg[len(userChats[0].Msg)-1].Msg, msgPb)
 			if err != nil {
 				return nil, utils.Wrap(err, "")
@@ -297,7 +297,7 @@ func (d *db.DataBases) GetNewestMsg(ID string) (msg *open_im_sdk.MsgData, err er
 	return nil, nil
 }
 
-func (d *db.DataBases) GetOldestMsg(ID string) (msg *open_im_sdk.MsgData, err error) {
+func (d *db.DataBases) GetOldestMsg(ID string) (msg *sdkws.MsgData, err error) {
 	ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second)
 	c := d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cChat)
 	regex := fmt.Sprintf("^%s", ID)
@@ -322,7 +322,7 @@ func (d *db.DataBases) GetOldestMsg(ID string) (msg *open_im_sdk.MsgData, err er
 		if len(oldestMsg) == 0 {
 			oldestMsg = userChats[0].Msg[len(userChats[0].Msg)-1].Msg
 		}
-		msgPb := &open_im_sdk.MsgData{}
+		msgPb := &sdkws.MsgData{}
 		err = proto.Unmarshal(oldestMsg, msgPb)
 		if err != nil {
 			return nil, utils.Wrap(err, "")
@@ -332,7 +332,7 @@ func (d *db.DataBases) GetOldestMsg(ID string) (msg *open_im_sdk.MsgData, err er
 	return nil, nil
 }
 
-func (d *db.DataBases) GetMsgBySeqListMongo2(uid string, seqList []uint32, operationID string) (seqMsg []*open_im_sdk.MsgData, err error) {
+func (d *db.DataBases) GetMsgBySeqListMongo2(uid string, seqList []uint32, operationID string) (seqMsg []*sdkws.MsgData, err error) {
 	var hasSeqList []uint32
 	singleCount := 0
 	ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second)
@@ -359,7 +359,7 @@ func (d *db.DataBases) GetMsgBySeqListMongo2(uid string, seqList []uint32, opera
 		}
 		singleCount = 0
 		for i := 0; i < len(sChat.Msg); i++ {
-			msg := new(open_im_sdk.MsgData)
+			msg := new(sdkws.MsgData)
 			if err = proto.Unmarshal(sChat.Msg[i].Msg, msg); err != nil {
 				log.NewError(operationID, "Unmarshal err", seqUid, value, uid, seqList, err.Error())
 				return nil, err
@@ -383,7 +383,7 @@ func (d *db.DataBases) GetMsgBySeqListMongo2(uid string, seqList []uint32, opera
 	}
 	return seqMsg, nil
 }
-func (d *db.DataBases) GetSuperGroupMsgBySeqListMongo(groupID string, seqList []uint32, operationID string) (seqMsg []*open_im_sdk.MsgData, err error) {
+func (d *db.DataBases) GetSuperGroupMsgBySeqListMongo(groupID string, seqList []uint32, operationID string) (seqMsg []*sdkws.MsgData, err error) {
 	var hasSeqList []uint32
 	singleCount := 0
 	ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second)
@@ -410,7 +410,7 @@ func (d *db.DataBases) GetSuperGroupMsgBySeqListMongo(groupID string, seqList []
 		}
 		singleCount = 0
 		for i := 0; i < len(sChat.Msg); i++ {
-			msg := new(open_im_sdk.MsgData)
+			msg := new(sdkws.MsgData)
 			if err = proto.Unmarshal(sChat.Msg[i].Msg, msg); err != nil {
 				log.NewError(operationID, "Unmarshal err", seqUid, value, groupID, seqList, err.Error())
 				return nil, err
@@ -435,7 +435,7 @@ func (d *db.DataBases) GetSuperGroupMsgBySeqListMongo(groupID string, seqList []
 	return seqMsg, nil
 }
 
-func (d *db.DataBases) GetMsgAndIndexBySeqListInOneMongo2(suffixUserID string, seqList []uint32, operationID string) (seqMsg []*open_im_sdk.MsgData, indexList []int, unexistSeqList []uint32, err error) {
+func (d *db.DataBases) GetMsgAndIndexBySeqListInOneMongo2(suffixUserID string, seqList []uint32, operationID string) (seqMsg []*sdkws.MsgData, indexList []int, unexistSeqList []uint32, err error) {
 	ctx, _ := context.WithTimeout(context.Background(), time.Duration(config.Config.Mongo.DBTimeout)*time.Second)
 	c := d.mongoClient.Database(config.Config.Mongo.DBDatabase).Collection(cChat)
 	sChat := UserChat{}
@@ -446,7 +446,7 @@ func (d *db.DataBases) GetMsgAndIndexBySeqListInOneMongo2(suffixUserID string, s
 	singleCount := 0
 	var hasSeqList []uint32
 	for i := 0; i < len(sChat.Msg); i++ {
-		msg := new(open_im_sdk.MsgData)
+		msg := new(sdkws.MsgData)
 		if err = proto.Unmarshal(sChat.Msg[i].Msg, msg); err != nil {
 			log.NewError(operationID, "Unmarshal err", msg.String(), err.Error())
 			return nil, nil, nil, err
@@ -470,18 +470,18 @@ func (d *db.DataBases) GetMsgAndIndexBySeqListInOneMongo2(suffixUserID string, s
 	return seqMsg, indexList, unexistSeqList, nil
 }
 
-func genExceptionMessageBySeqList(seqList []uint32) (exceptionMsg []*open_im_sdk.MsgData) {
+func genExceptionMessageBySeqList(seqList []uint32) (exceptionMsg []*sdkws.MsgData) {
 	for _, v := range seqList {
-		msg := new(open_im_sdk.MsgData)
+		msg := new(sdkws.MsgData)
 		msg.Seq = v
 		exceptionMsg = append(exceptionMsg, msg)
 	}
 	return exceptionMsg
 }
 
-func genExceptionSuperGroupMessageBySeqList(seqList []uint32, groupID string) (exceptionMsg []*open_im_sdk.MsgData) {
+func genExceptionSuperGroupMessageBySeqList(seqList []uint32, groupID string) (exceptionMsg []*sdkws.MsgData) {
 	for _, v := range seqList {
-		msg := new(open_im_sdk.MsgData)
+		msg := new(sdkws.MsgData)
 		msg.Seq = v
 		msg.GroupID = groupID
 		msg.SessionType = constant.SuperGroupChatType

pkg/common/db/unrelation/tag.go

@@ -1,117 +0,0 @@
-package unrelation
-
-import (
-	"Open_IM/pkg/common/db/table/unrelation"
-	"Open_IM/pkg/utils"
-	"context"
-	"go.mongodb.org/mongo-driver/bson"
-	"go.mongodb.org/mongo-driver/mongo"
-	"go.mongodb.org/mongo-driver/mongo/options"
-	"math/rand"
-	"strconv"
-	"time"
-)
-
-type TagMongoDriver struct {
-	mgoDB                *mongo.Database
-	TagCollection        *mongo.Collection
-	TagSendLogCollection *mongo.Collection
-}
-
-func NewTagMongoDriver(mgoDB *mongo.Database) *TagMongoDriver {
-	return &TagMongoDriver{mgoDB: mgoDB, TagCollection: mgoDB.Collection(unrelation.CTag), TagSendLogCollection: mgoDB.Collection(unrelation.CSendLog)}
-}
-
-func (db *TagMongoDriver) generateTagID(tagName, userID string) string {
-	return utils.Md5(tagName + userID + strconv.Itoa(rand.Int()) + time.Now().String())
-}
-
-func (db *TagMongoDriver) GetUserTags(ctx context.Context, userID string) ([]unrelation.TagModel, error) {
-	var tags []unrelation.TagModel
-	cursor, err := db.TagCollection.Find(ctx, bson.M{"user_id": userID})
-	if err != nil {
-		return tags, err
-	}
-	if err = cursor.All(ctx, &tags); err != nil {
-		return tags, err
-	}
-	return tags, nil
-}
-
-func (db *TagMongoDriver) CreateTag(ctx context.Context, userID, tagName string, userList []string) error {
-	tagID := generateTagID(tagName, userID)
-	tag := unrelation.TagModel{
-		UserID:   userID,
-		TagID:    tagID,
-		TagName:  tagName,
-		UserList: userList,
-	}
-	_, err := db.TagCollection.InsertOne(ctx, tag)
-	return err
-}
-
-func (db *TagMongoDriver) GetTagByID(ctx context.Context, userID, tagID string) (unrelation.TagModel, error) {
-	var tag unrelation.TagModel
-	err := db.TagCollection.FindOne(ctx, bson.M{"user_id": userID, "tag_id": tagID}).Decode(&tag)
-	return tag, err
-}
-
-func (db *TagMongoDriver) DeleteTag(ctx context.Context, userID, tagID string) error {
-	_, err := db.TagCollection.DeleteOne(ctx, bson.M{"user_id": userID, "tag_id": tagID})
-	return err
-}
-
-func (db *TagMongoDriver) SetTag(ctx context.Context, userID, tagID, newName string, increaseUserIDList []string, reduceUserIDList []string) error {
-	var tag unrelation.TagModel
-	if err := db.TagCollection.FindOne(ctx, bson.M{"tag_id": tagID, "user_id": userID}).Decode(&tag); err != nil {
-		return err
-	}
-	if newName != "" {
-		_, err := db.TagCollection.UpdateOne(ctx, bson.M{"user_id": userID, "tag_id": tagID}, bson.M{"$set": bson.M{"tag_name": newName}})
-		if err != nil {
-			return err
-		}
-	}
-	tag.UserList = append(tag.UserList, increaseUserIDList...)
-	tag.UserList = utils.RemoveRepeatedStringInList(tag.UserList)
-	for _, v := range reduceUserIDList {
-		for i2, v2 := range tag.UserList {
-			if v == v2 {
-				tag.UserList[i2] = ""
-			}
-		}
-	}
-	var newUserList []string
-	for _, v := range tag.UserList {
-		if v != "" {
-			newUserList = append(newUserList, v)
-		}
-	}
-	_, err := db.TagCollection.UpdateOne(ctx, bson.M{"user_id": userID, "tag_id": tagID}, bson.M{"$set": bson.M{"user_list": newUserList}})
-	if err != nil {
-		return err
-	}
-	return nil
-}
-
-func (db *TagMongoDriver) GetUserIDListByTagID(ctx context.Context, userID, tagID string) ([]string, error) {
-	var tag unrelation.TagModel
-	err := db.TagCollection.FindOne(ctx, bson.M{"user_id": userID, "tag_id": tagID}).Decode(&tag)
-	return tag.UserList, err
-}
-
-func (db *TagMongoDriver) SaveTagSendLog(ctx context.Context, tagSendLog *unrelation.TagSendLogModel) error {
-	_, err := db.TagSendLogCollection.InsertOne(ctx, tagSendLog)
-	return err
-}
-
-func (db *TagMongoDriver) GetTagSendLogs(ctx context.Context, userID string, showNumber, pageNumber int32) ([]unrelation.TagSendLogModel, error) {
-	var tagSendLogs []unrelation.TagSendLogModel
-	findOpts := options.Find().SetLimit(int64(showNumber)).SetSkip(int64(showNumber) * (int64(pageNumber) - 1)).SetSort(bson.M{"send_time": -1})
-	cursor, err := db.TagSendLogCollection.Find(ctx, bson.M{"send_id": userID}, findOpts)
-	if err != nil {
-		return tagSendLogs, err
-	}
-	err = cursor.All(ctx, &tagSendLogs)
-	return tagSendLogs, err
-}

pkg/common/db/unrelation/work_moment.go

@@ -1,151 +0,0 @@
-package unrelation
-
-import (
-	"Open_IM/pkg/common/config"
-	"Open_IM/pkg/common/constant"
-	"Open_IM/pkg/common/db/table/unrelation"
-	"Open_IM/pkg/utils"
-	"context"
-	"go.mongodb.org/mongo-driver/bson"
-	"go.mongodb.org/mongo-driver/mongo"
-	"go.mongodb.org/mongo-driver/mongo/options"
-	"math/rand"
-	"strconv"
-	"time"
-)
-
-type WorkMomentMongoDriver struct {
-	mgoDB                *mongo.Database
-	WorkMomentCollection *mongo.Collection
-}
-
-func NewWorkMomentMongoDriver(mgoDB *mongo.Database) *WorkMomentMongoDriver {
-	return &WorkMomentMongoDriver{mgoDB: mgoDB, WorkMomentCollection: mgoDB.Collection(unrelation.CWorkMoment)}
-}
-
-func (db *WorkMomentMongoDriver) generateWorkMomentID(userID string) string {
-	return utils.Md5(userID + strconv.Itoa(rand.Int()) + time.Now().String())
-}
-
-func (db *WorkMomentMongoDriver) generateWorkMomentCommentID(workMomentID string) string {
-	return utils.Md5(workMomentID + strconv.Itoa(rand.Int()) + time.Now().String())
-}
-
-func (db *WorkMomentMongoDriver) CreateOneWorkMoment(ctx context.Context, workMoment *unrelation.WorkMoment) error {
-	workMomentID := db.generateWorkMomentID(workMoment.UserID)
-	workMoment.WorkMomentID = workMomentID
-	workMoment.CreateTime = int32(time.Now().Unix())
-	_, err := db.WorkMomentCollection.InsertOne(ctx, workMoment)
-	return err
-}
-
-func (db *WorkMomentMongoDriver) DeleteOneWorkMoment(ctx context.Context, workMomentID string) error {
-	_, err := db.WorkMomentCollection.DeleteOne(ctx, bson.M{"work_moment_id": workMomentID})
-	return err
-}
-
-func (db *WorkMomentMongoDriver) DeleteComment(ctx context.Context, workMomentID, contentID, opUserID string) error {
-	_, err := db.WorkMomentCollection.UpdateOne(ctx, bson.D{{"work_moment_id", workMomentID},
-		{"$or", bson.A{
-			bson.D{{"user_id", opUserID}},
-			bson.D{{"comments", bson.M{"$elemMatch": bson.M{"user_id": opUserID}}}},
-		},
-		}}, bson.M{"$pull": bson.M{"comments": bson.M{"content_id": contentID}}})
-	return err
-}
-
-func (db *WorkMomentMongoDriver) GetWorkMomentByID(ctx context.Context, workMomentID string) (*unrelation.WorkMoment, error) {
-	workMoment := &unrelation.WorkMoment{}
-	err := db.WorkMomentCollection.FindOne(ctx, bson.M{"work_moment_id": workMomentID}).Decode(workMoment)
-	return workMoment, err
-}
-
-func (db *WorkMomentMongoDriver) LikeOneWorkMoment(ctx context.Context, likeUserID, userName, workMomentID string) (*unrelation.WorkMoment, bool, error) {
-	workMoment, err := db.GetWorkMomentByID(ctx, workMomentID)
-	if err != nil {
-		return nil, false, err
-	}
-	var isAlreadyLike bool
-	for i, user := range workMoment.LikeUserList {
-		if likeUserID == user.UserID {
-			isAlreadyLike = true
-			workMoment.LikeUserList = append(workMoment.LikeUserList[0:i], workMoment.LikeUserList[i+1:]...)
-		}
-	}
-	if !isAlreadyLike {
-		workMoment.LikeUserList = append(workMoment.LikeUserList, &unrelation.CommonUserModel{UserID: likeUserID, UserName: userName})
-	}
-	_, err = db.WorkMomentCollection.UpdateOne(ctx, bson.M{"work_moment_id": workMomentID}, bson.M{"$set": bson.M{"like_user_list": workMoment.LikeUserList}})
-	return workMoment, !isAlreadyLike, err
-}
-
-func (db *WorkMomentMongoDriver) CommentOneWorkMoment(ctx context.Context, comment *unrelation.Comment, workMomentID string) (unrelation.WorkMoment, error) {
-	comment.ContentID = generateWorkMomentCommentID(workMomentID)
-	var workMoment unrelation.WorkMoment
-	err := db.WorkMomentCollection.FindOneAndUpdate(ctx, bson.M{"work_moment_id": workMomentID}, bson.M{"$push": bson.M{"comments": comment}}).Decode(&workMoment)
-	return workMoment, err
-}
-
-func (db *WorkMomentMongoDriver) GetUserSelfWorkMoments(ctx context.Context, userID string, showNumber, pageNumber int32) ([]unrelation.WorkMoment, error) {
-	var workMomentList []unrelation.WorkMoment
-	findOpts := options.Find().SetLimit(int64(showNumber)).SetSkip(int64(showNumber) * (int64(pageNumber) - 1)).SetSort(bson.M{"create_time": -1})
-	result, err := db.WorkMomentCollection.Find(ctx, bson.M{"user_id": userID}, findOpts)
-	if err != nil {
-		return workMomentList, nil
-	}
-	err = result.All(ctx, &workMomentList)
-	return workMomentList, err
-}
-
-func (db *WorkMomentMongoDriver) GetUserWorkMoments(ctx context.Context, opUserID, userID string, showNumber, pageNumber int32, friendIDList []string) ([]unrelation.WorkMoment, error) {
-	var workMomentList []unrelation.WorkMoment
-	findOpts := options.Find().SetLimit(int64(showNumber)).SetSkip(int64(showNumber) * (int64(pageNumber) - 1)).SetSort(bson.M{"create_time": -1})
-	result, err := db.WorkMomentCollection.Find(ctx, bson.D{ // 等价条件: select * from
-		{"user_id", userID},
-		{"$or", bson.A{
-			bson.D{{"permission", constant.WorkMomentPermissionCantSee}, {"permission_user_id_list", bson.D{{"$nin", bson.A{opUserID}}}}},
-			bson.D{{"permission", constant.WorkMomentPermissionCanSee}, {"permission_user_id_list", bson.D{{"$in", bson.A{opUserID}}}}},
-			bson.D{{"permission", constant.WorkMomentPublic}},
-		}},
-	}, findOpts)
-	if err != nil {
-		return workMomentList, nil
-	}
-	err = result.All(ctx, &workMomentList)
-	return workMomentList, err
-}
-
-func (db *WorkMomentMongoDriver) GetUserFriendWorkMoments(ctx context.Context, showNumber, pageNumber int32, userID string, friendIDList []string) ([]unrelation.WorkMoment, error) {
-	var workMomentList []unrelation.WorkMoment
-	findOpts := options.Find().SetLimit(int64(showNumber)).SetSkip(int64(showNumber) * (int64(pageNumber) - 1)).SetSort(bson.M{"create_time": -1})
-	var filter bson.D
-	permissionFilter := bson.D{
-		{"$or", bson.A{
-			bson.D{{"permission", constant.WorkMomentPermissionCantSee}, {"permission_user_id_list", bson.D{{"$nin", bson.A{userID}}}}},
-			bson.D{{"permission", constant.WorkMomentPermissionCanSee}, {"permission_user_id_list", bson.D{{"$in", bson.A{userID}}}}},
-			bson.D{{"permission", constant.WorkMomentPublic}},
-		}}}
-	if config.Config.WorkMoment.OnlyFriendCanSee {
-		filter = bson.D{
-			{"$or", bson.A{
-				bson.D{{"user_id", userID}}, //self
-				bson.D{{"$and", bson.A{permissionFilter, bson.D{{"user_id", bson.D{{"$in", friendIDList}}}}}}},
-			},
-			},
-		}
-	} else {
-		filter = bson.D{
-			{"$or", bson.A{
-				bson.D{{"user_id", userID}}, //self
-				permissionFilter,
-			},
-			},
-		}
-	}
-	result, err := db.WorkMomentCollection.Find(ctx, filter, findOpts)
-	if err != nil {
-		return workMomentList, err
-	}
-	err = result.All(ctx, &workMomentList)
-	return workMomentList, err
-}