Open-IM-Server/internal/rpc/msg/extend_msg.go

package msg

import (
	"Open_IM/pkg/common/db"
	"Open_IM/pkg/common/log"
	"Open_IM/pkg/proto/msg"
	"Open_IM/pkg/proto/sdk_ws"
	"Open_IM/pkg/utils"
	"context"
	go_redis "github.com/go-redis/redis/v8"

	"time"
)

func (rpc *rpcChat) SetMessageReactionExtensions(ctx context.Context, req *msg.ModifyMessageReactionExtensionsReq) (resp *msg.ModifyMessageReactionExtensionsResp, err error) {
	var rResp msg.ModifyMessageReactionExtensionsResp
	var extendMsgResp msg.ExtendMsgResp
	var failedExtendMsgResp msg.ExtendMsgResp
	var oneExtendMsg msg.ExtendMsg
	var failedExtendMsg msg.ExtendMsg
	oneExtendMsg.ClientMsgID = req.ClientMsgID
	oneExtendMsg.MsgFirstModifyTime = req.MsgFirstModifyTime
	oneFailedReactionExtensionList := make(map[string]*msg.KeyValueResp)
	oneSuccessReactionExtensionList := make(map[string]*msg.KeyValueResp)
	isExists, err := db.DB.JudgeMessageReactionEXISTS(req.ClientMsgID, req.SessionType)
	if err != nil {
		extendMsgResp.ErrCode = 100
		extendMsgResp.ErrMsg = err.Error()
		for k, value := range req.ReactionExtensionList {
			temp := new(msg.KeyValueResp)
			temp.KeyValue = value
			temp.ErrMsg = err.Error()
			temp.ErrCode = 100
			oneFailedReactionExtensionList[k] = temp
		}
		oneExtendMsg.ReactionExtensionList = oneFailedReactionExtensionList
		extendMsgResp.ExtendMsg = &oneExtendMsg
		rResp.FailedList = append(rResp.FailedList, &extendMsgResp)
		return &rResp, nil
	}

	if !isExists {
		if !req.IsReact {
			log.Debug(req.OperationID, "redis handle firstly", req.String())
			oneExtendMsg.MsgFirstModifyTime = utils.GetCurrentTimestampByMill()
			//redis处理
			for k, v := range req.ReactionExtensionList {
				//抢占分布式锁
				err := lockMessageTypeKey(req.ClientMsgID, k)
				if err != nil {
					setKeyResultInfo(oneFailedReactionExtensionList, 100, err.Error(), req.ClientMsgID, k, v)
					continue
				}
				redisValue, err := db.DB.GetMessageTypeKeyValue(req.ClientMsgID, req.SessionType, k)
				if err != nil && err != go_redis.Nil {
					setKeyResultInfo(oneFailedReactionExtensionList, 200, err.Error(), req.ClientMsgID, k, v)
					continue
				}
				temp := new(server_api_params.KeyValue)
				utils.JsonStringToStruct(redisValue, temp)
				if v.LatestUpdateTime != temp.LatestUpdateTime {
					setKeyResultInfo(oneFailedReactionExtensionList, 300, "message have update", req.ClientMsgID, k, temp)
					continue
				} else {
					v.LatestUpdateTime = utils.GetCurrentTimestampByMill()
					newerr := db.DB.SetMessageTypeKeyValue(req.ClientMsgID, req.SessionType, k, utils.StructToJsonString(v))
					if newerr != nil {
						setKeyResultInfo(oneFailedReactionExtensionList, 201, newerr.Error(), req.ClientMsgID, k, temp)
						continue
					}
					setKeyResultInfo(oneSuccessReactionExtensionList, 0, "", req.ClientMsgID, k, v)
				}

			}

		} else {
			//mongo处理
		}

	} else {
		log.Debug(req.OperationID, "redis handle secondly", req.String())

		for k, v := range req.ReactionExtensionList {
			//抢占分布式锁
			err := lockMessageTypeKey(req.ClientMsgID, k)
			if err != nil {
				setKeyResultInfo(oneFailedReactionExtensionList, 100, err.Error(), req.ClientMsgID, k, v)
				continue
			}
			redisValue, err := db.DB.GetMessageTypeKeyValue(req.ClientMsgID, req.SessionType, k)
			if err != nil && err != go_redis.Nil {
				setKeyResultInfo(oneFailedReactionExtensionList, 200, err.Error(), req.ClientMsgID, k, v)
				continue
			}
			temp := new(server_api_params.KeyValue)
			utils.JsonStringToStruct(redisValue, temp)
			if v.LatestUpdateTime != temp.LatestUpdateTime {
				setKeyResultInfo(oneFailedReactionExtensionList, 300, "message have update", req.ClientMsgID, k, temp)
				continue
			} else {
				v.LatestUpdateTime = utils.GetCurrentTimestampByMill()
				newerr := db.DB.SetMessageTypeKeyValue(req.ClientMsgID, req.SessionType, k, utils.StructToJsonString(v))
				if newerr != nil {
					setKeyResultInfo(oneFailedReactionExtensionList, 201, newerr.Error(), req.ClientMsgID, k, temp)
					continue
				}
				setKeyResultInfo(oneSuccessReactionExtensionList, 0, "", req.ClientMsgID, k, v)
			}

		}
	}
	oneExtendMsg.ReactionExtensionList = oneSuccessReactionExtensionList
	extendMsgResp.ExtendMsg = &oneExtendMsg
	failedExtendMsg.ReactionExtensionList = oneFailedReactionExtensionList
	failedExtendMsgResp.ExtendMsg = &failedExtendMsg
	rResp.FailedList = append(rResp.FailedList, &failedExtendMsgResp)
	rResp.SuccessList = append(rResp.FailedList, &extendMsgResp)
	return &rResp, nil

}
func setKeyResultInfo(m map[string]*msg.KeyValueResp, errCode int32, errMsg, clientMsgID, typeKey string, keyValue *server_api_params.KeyValue) {
	temp := new(msg.KeyValueResp)
	temp.KeyValue = keyValue
	temp.ErrCode = errCode
	temp.ErrMsg = errMsg
	m[typeKey] = temp
	_ = db.DB.UnLockMessageTypeKey(clientMsgID, typeKey)
}

func (rpc *rpcChat) GetMessageListReactionExtensions(ctx context.Context, req *msg.OperateMessageListReactionExtensionsReq) (resp *msg.OperateMessageListReactionExtensionsResp, err error) {
	//for _, messageValue := range req.MessageReactionKeyList {
	//	isExists, err := db.DB.JudgeMessageReactionEXISTS(messageValue.ClientMsgID,req.SessionType)
	//	if err != nil {
	//
	//	}
	//	var failedList []*msg.ExtendMsgResp
	//	var successList []*msg.ExtendMsgResp
	//	var oneExtendMsg   msg.ExtendMsg
	//	oneExtendMsg.ClientMsgID = req.ClientMsgID
	//	oneFailedReactionExtensionList:=make(map[string]*msg.KeyValueResp)
	//	oneSuccessReactionExtensionList:=make(map[string]*msg.KeyValueResp)
	//	if !isExists {
	//		if  !req.IsReact {
	//			oneExtendMsg.MsgFirstModifyTime = utils.GetCurrentTimestampByMill()
	//			//redis处理
	//			for k, v := range req.ReactionExtensionList {
	//				//抢占分布式锁
	//				err:=lockMessageTypeKey(req.ClientMsgID,k)
	//				if err != nil {
	//					setKeyResultInfo(oneFailedReactionExtensionList,100,err.Error(),req.ClientMsgID,k,v)
	//					continue
	//				}
	//				redisValue,err:=db.DB.GetMessageTypeKeyValue(req.ClientMsgID,req.SessionType,k)
	//				if err != nil&&err!=go_redis.Nil {
	//					setKeyResultInfo(oneFailedReactionExtensionList,200,err.Error(),req.ClientMsgID,k,v)
	//					continue
	//				}
	//				temp:=new(server_api_params.KeyValue)
	//				utils.JsonStringToStruct(redisValue,temp)
	//				if v.LatestUpdateTime != temp.LatestUpdateTime {
	//					setKeyResultInfo(oneFailedReactionExtensionList,300,"message have update",req.ClientMsgID,k,temp)
	//					continue
	//				}else{
	//					v.LatestUpdateTime = utils.GetCurrentTimestampByMill()
	//					newerr:=db.DB.SetMessageTypeKeyValue(req.ClientMsgID,req.SessionType,k,utils.StructToJsonString(v))
	//					if newerr != nil {
	//						setKeyResultInfo(oneFailedReactionExtensionList,201,newerr.Error(),req.ClientMsgID,k,temp)
	//						continue
	//					}
	//					setKeyResultInfo(oneSuccessReactionExtensionList,0,"",req.ClientMsgID,k,v)
	//				}
	//
	//			}
	//
	//		}else{
	//			//mongo处理
	//		}
	//
	//	}else{
	//
	//	}
	//	return
	//}
	return

}

func (rpc *rpcChat) AddMessageReactionExtensions(ctx context.Context, req *msg.ModifyMessageReactionExtensionsReq) (resp *msg.ModifyMessageReactionExtensionsResp, err error) {
	return
}

func (rpc *rpcChat) DeleteMessageReactionExtensions(ctx context.Context, req *msg.OperateMessageListReactionExtensionsReq) (resp *msg.OperateMessageListReactionExtensionsResp, err error) {
	return
}
func lockMessageTypeKey(clientMsgID, typeKey string) (err error) {
	for i := 0; i < 3; i++ {
		err = db.DB.LockMessageTypeKey(clientMsgID, typeKey)
		if err != nil {
			time.Sleep(time.Millisecond * 100)
			continue
		} else {
			break
		}
	}
	return err

}