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

package msg

import (
	"context"
	"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/msg"
	"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/sdkws"
)

func (m *msgServer) SetMessageReactionExtensions(ctx context.Context, req *msg.SetMessageReactionExtensionsReq) (resp *msg.SetMessageReactionExtensionsResp, err error) {
	//resp = &msg.SetMessageReactionExtensionsResp{}
	////resp.ClientMsgID = req.ClientMsgID
	////resp.MsgFirstModifyTime = req.MsgFirstModifyTime
	//
	//if err := CallbackSetMessageReactionExtensions(ctx, req); err != nil {
	//	return nil, err
	//}
	////if ExternalExtension
	//if req.IsExternalExtensions {
	//	resp.MsgFirstModifyTime = req.MsgFirstModifyTime
	//	notification.ExtendMessageUpdatedNotification(req.OperationID, req.OpUserID, req.SourceID, req.SessionType, req, &resp, !req.IsReact, false)
	//	return resp, nil
	//}
	//isExists, err := m.MsgDatabase.JudgeMessageReactionExist(ctx, req.ClientMsgID, req.SessionType)
	//if err != nil {
	//	return nil, err
	//}
	//
	//if !isExists {
	//	if !req.IsReact {
	//		resp.MsgFirstModifyTime = utils.GetCurrentTimestampByMill()
	//		for k, v := range req.ReactionExtensions {
	//			err := m.MessageLocker.LockMessageTypeKey(ctx, req.ClientMsgID, k)
	//			if err != nil {
	//				return nil, err
	//			}
	//			v.LatestUpdateTime = utils.GetCurrentTimestampByMill()
	//			if err := m.MsgDatabase.SetMessageTypeKeyValue(ctx, req.ClientMsgID, req.SessionType, k, utils.StructToJsonString(v)); err != nil {
	//				return nil, err
	//			}
	//		}
	//		resp.IsReact = true
	//		_, err := m.MsgDatabase.SetMessageReactionExpire(ctx, req.ClientMsgID, req.SessionType, time.Duration(24*3)*time.Hour)
	//		if err != nil {
	//			return nil, err
	//		}
	//	} else {
	//		err := m.MessageLocker.LockGlobalMessage(ctx, req.ClientMsgID)
	//		if err != nil {
	//			return nil, err
	//		}
	//		mongoValue, err := m.MsgDatabase.GetExtendMsg(ctx, req.SourceID, req.SessionType, req.ClientMsgID, req.MsgFirstModifyTime)
	//		if err != nil {
	//			return nil, err
	//		}
	//		setValue := make(map[string]*sdkws.KeyValue)
	//		for k, v := range req.ReactionExtensions {
	//
	//			temp := new(sdkws.KeyValue)
	//			if vv, ok := mongoValue.ReactionExtensions[k]; ok {
	//				utils.CopyStructFields(temp, &vv)
	//				if v.LatestUpdateTime != vv.LatestUpdateTime {
	//					setKeyResultInfo(&resp, 300, "message have update", req.ClientMsgID, k, temp)
	//					continue
	//				}
	//			}
	//			temp.TypeKey = k
	//			temp.Value = v.Value
	//			temp.LatestUpdateTime = utils.GetCurrentTimestampByMill()
	//			setValue[k] = temp
	//		}
	//		err = db.DB.InsertOrUpdateReactionExtendMsgSet(req.SourceID, req.SessionType, req.ClientMsgID, req.MsgFirstModifyTime, setValue)
	//		if err != nil {
	//			for _, value := range setValue {
	//				temp := new(msg.KeyValueResp)
	//				temp.KeyValue = value
	//				temp.ErrMsg = err.Error()
	//				temp.ErrCode = 100
	//				resp.Result = append(resp.Result, temp)
	//			}
	//		} else {
	//			for _, value := range setValue {
	//				temp := new(msg.KeyValueResp)
	//				temp.KeyValue = value
	//				resp.Result = append(resp.Result, temp)
	//			}
	//		}
	//		lockErr := m.dMessageLocker.UnLockGlobalMessage(req.ClientMsgID)
	//		if lockErr != nil {
	//			log.Error(req.OperationID, "UnLockGlobalMessage err:", lockErr.Error())
	//		}
	//	}
	//
	//} else {
	//	log.Debug(req.OperationID, "redis handle secondly", req.String())
	//
	//	for k, v := range req.Pb2Model {
	//		err := m.dMessageLocker.LockMessageTypeKey(req.ClientMsgID, k)
	//		if err != nil {
	//			setKeyResultInfo(&resp, 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(&resp, 200, err.Error(), req.ClientMsgID, k, v)
	//			continue
	//		}
	//		temp := new(sdkws.KeyValue)
	//		utils.JsonStringToStruct(redisValue, temp)
	//		if v.LatestUpdateTime != temp.LatestUpdateTime {
	//			setKeyResultInfo(&resp, 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(&resp, 201, newerr.Error(), req.ClientMsgID, k, temp)
	//				continue
	//			}
	//			setKeyResultInfo(&resp, 0, "", req.ClientMsgID, k, v)
	//		}
	//
	//	}
	//}
	//if !isExists {
	//	if !req.IsReact {
	//		notification.ExtendMessageUpdatedNotification(req.OperationID, req.OpUserID, req.SourceID, req.SessionType, req, &resp, true, true)
	//	} else {
	//		notification.ExtendMessageUpdatedNotification(req.OperationID, req.OpUserID, req.SourceID, req.SessionType, req, &resp, false, false)
	//	}
	//} else {
	//	notification.ExtendMessageUpdatedNotification(req.OperationID, req.OpUserID, req.SourceID, req.SessionType, req, &resp, false, true)
	//}
	//log.Debug(req.OperationID, utils.GetSelfFuncName(), "m return is:", resp.String())
	return resp, nil

}
func (m *msgServer) setKeyResultInfo(ctx context.Context, r *msg.SetMessageReactionExtensionsResp, errCode int32, errMsg, clientMsgID, typeKey string, keyValue *sdkws.KeyValue) {
	temp := new(msg.KeyValueResp)
	temp.KeyValue = keyValue
	temp.ErrCode = errCode
	temp.ErrMsg = errMsg
	r.Result = append(r.Result, temp)
	_ = m.MessageLocker.UnLockMessageTypeKey(ctx, clientMsgID, typeKey)
}
func (m *msgServer) setDeleteKeyResultInfo(ctx context.Context, r *msg.DeleteMessagesReactionExtensionsResp, errCode int32, errMsg, clientMsgID, typeKey string, keyValue *sdkws.KeyValue) {
	temp := new(msg.KeyValueResp)
	temp.KeyValue = keyValue
	temp.ErrCode = errCode
	temp.ErrMsg = errMsg
	r.Result = append(r.Result, temp)
	_ = m.MessageLocker.UnLockMessageTypeKey(ctx, clientMsgID, typeKey)
}

func (m *msgServer) GetMessagesReactionExtensions(ctx context.Context, req *msg.GetMessagesReactionExtensionsReq) (resp *msg.GetMessagesReactionExtensionsResp, err error) {
	//log.Debug(req.OperationID, utils.GetSelfFuncName(), "m args is:", req.String())
	//var rResp msg.GetMessageListReactionExtensionsResp
	//for _, messageValue := range req.MessageReactionKeyList {
	//	var oneMessage msg.SingleMessageExtensionResult
	//	oneMessage.ClientMsgID = messageValue.ClientMsgID
	//
	//	isExists, err := db.DB.JudgeMessageReactionExist(messageValue.ClientMsgID, req.SessionType)
	//	if err != nil {
	//		rResp.ErrCode = 100
	//		rResp.ErrMsg = err.Error()
	//		return &rResp, nil
	//	}
	//	if isExists {
	//		redisValue, err := db.DB.GetOneMessageAllReactionList(messageValue.ClientMsgID, req.SessionType)
	//		if err != nil {
	//			oneMessage.ErrCode = 100
	//			oneMessage.ErrMsg = err.Error()
	//			rResp.SingleMessageResult = append(rResp.SingleMessageResult, &oneMessage)
	//			continue
	//		}
	//		keyMap := make(map[string]*sdkws.KeyValue)
	//
	//		for k, v := range redisValue {
	//			temp := new(sdkws.KeyValue)
	//			utils.JsonStringToStruct(v, temp)
	//			keyMap[k] = temp
	//		}
	//		oneMessage.Pb2Model = keyMap
	//
	//	} else {
	//		mongoValue, err := db.DB.GetExtendMsg(req.SourceID, req.SessionType, messageValue.ClientMsgID, messageValue.MsgFirstModifyTime)
	//		if err != nil {
	//			oneMessage.ErrCode = 100
	//			oneMessage.ErrMsg = err.Error()
	//			rResp.SingleMessageResult = append(rResp.SingleMessageResult, &oneMessage)
	//			continue
	//		}
	//		keyMap := make(map[string]*sdkws.KeyValue)
	//
	//		for k, v := range mongoValue.Pb2Model {
	//			temp := new(sdkws.KeyValue)
	//			temp.TypeKey = v.TypeKey
	//			temp.Value = v.Value
	//			temp.LatestUpdateTime = v.LatestUpdateTime
	//			keyMap[k] = temp
	//		}
	//		oneMessage.Pb2Model = keyMap
	//	}
	//	rResp.SingleMessageResult = append(rResp.SingleMessageResult, &oneMessage)
	//}
	//log.Debug(req.OperationID, utils.GetSelfFuncName(), "m return is:", rResp.String())
	return resp, nil

}

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

func (m *msgServer) DeleteMessageReactionExtensions(ctx context.Context, req *msg.DeleteMessagesReactionExtensionsReq) (resp *msg.DeleteMessagesReactionExtensionsResp, err error) {
	//log.Debug(req.OperationID, utils.GetSelfFuncName(), "m args is:", req.String())
	//var rResp msg.DeleteMessagesReactionExtensionsResp
	//callbackResp := notification.callbackDeleteMessageReactionExtensions(req)
	//if callbackResp.ActionCode != constant.ActionAllow || callbackResp.ErrCode != 0 {
	//	rResp.ErrCode = int32(callbackResp.ErrCode)
	//	rResp.ErrMsg = callbackResp.ErrMsg
	//	for _, value := range req.Pb2Model {
	//		temp := new(msg.KeyValueResp)
	//		temp.KeyValue = value
	//		temp.ErrMsg = callbackResp.ErrMsg
	//		temp.ErrCode = 100
	//		rResp.Result = append(rResp.Result, temp)
	//	}
	//	return &rResp, nil
	//}
	////if ExternalExtension
	//if req.IsExternalExtensions {
	//	rResp.Result = callbackResp.ResultReactionExtensionList
	//	notification.ExtendMessageDeleteNotification(req.OperationID, req.OpUserID, req.SourceID, req.SessionType, req, &rResp, false, false)
	//	return &rResp, nil
	//
	//}
	//for _, v := range callbackResp.ResultReactionExtensions {
	//	if v.ErrCode != 0 {
	//		func(req *[]*sdkws.KeyValue, typeKey string) {
	//			for i := 0; i < len(*req); i++ {
	//				if (*req)[i].TypeKey == typeKey {
	//					*req = append((*req)[:i], (*req)[i+1:]...)
	//				}
	//			}
	//		}(&req.Pb2Model, v.KeyValue.TypeKey)
	//		rResp.Result = append(rResp.Result, v)
	//	}
	//}
	//isExists, err := db.DB.JudgeMessageReactionExist(req.ClientMsgID, req.SessionType)
	//if err != nil {
	//	rResp.ErrCode = 100
	//	rResp.ErrMsg = err.Error()
	//	for _, value := range req.Pb2Model {
	//		temp := new(msg.KeyValueResp)
	//		temp.KeyValue = value
	//		temp.ErrMsg = err.Error()
	//		temp.ErrCode = 100
	//		rResp.Result = append(rResp.Result, temp)
	//	}
	//	return &rResp, nil
	//}
	//
	//if isExists {
	//	log.Debug(req.OperationID, "redis handle this delete", req.String())
	//	for _, v := range req.Pb2Model {
	//		err := m.dMessageLocker.LockMessageTypeKey(req.ClientMsgID, v.TypeKey)
	//		if err != nil {
	//			setDeleteKeyResultInfo(&rResp, 100, err.Error(), req.ClientMsgID, v.TypeKey, v)
	//			continue
	//		}
	//
	//		redisValue, err := db.DB.GetMessageTypeKeyValue(req.ClientMsgID, req.SessionType, v.TypeKey)
	//		if err != nil && err != go_redis.Nil {
	//			setDeleteKeyResultInfo(&rResp, 200, err.Error(), req.ClientMsgID, v.TypeKey, v)
	//			continue
	//		}
	//		temp := new(sdkws.KeyValue)
	//		utils.JsonStringToStruct(redisValue, temp)
	//		if v.LatestUpdateTime != temp.LatestUpdateTime {
	//			setDeleteKeyResultInfo(&rResp, 300, "message have update", req.ClientMsgID, v.TypeKey, temp)
	//			continue
	//		} else {
	//			newErr := db.DB.DeleteOneMessageKey(req.ClientMsgID, req.SessionType, v.TypeKey)
	//			if newErr != nil {
	//				setDeleteKeyResultInfo(&rResp, 201, newErr.Error(), req.ClientMsgID, v.TypeKey, temp)
	//				continue
	//			}
	//			setDeleteKeyResultInfo(&rResp, 0, "", req.ClientMsgID, v.TypeKey, v)
	//		}
	//	}
	//} else {
	//	err := m.dMessageLocker.LockGlobalMessage(req.ClientMsgID)
	//	if err != nil {
	//		rResp.ErrCode = 100
	//		rResp.ErrMsg = err.Error()
	//		for _, value := range req.Pb2Model {
	//			temp := new(msg.KeyValueResp)
	//			temp.KeyValue = value
	//			temp.ErrMsg = err.Error()
	//			temp.ErrCode = 100
	//			rResp.Result = append(rResp.Result, temp)
	//		}
	//		return &rResp, nil
	//	}
	//	mongoValue, err := db.DB.GetExtendMsg(req.SourceID, req.SessionType, req.ClientMsgID, req.MsgFirstModifyTime)
	//	if err != nil {
	//		rResp.ErrCode = 200
	//		rResp.ErrMsg = err.Error()
	//		for _, value := range req.Pb2Model {
	//			temp := new(msg.KeyValueResp)
	//			temp.KeyValue = value
	//			temp.ErrMsg = err.Error()
	//			temp.ErrCode = 100
	//			rResp.Result = append(rResp.Result, temp)
	//		}
	//		return &rResp, nil
	//	}
	//	setValue := make(map[string]*sdkws.KeyValue)
	//	for _, v := range req.Pb2Model {
	//
	//		temp := new(sdkws.KeyValue)
	//		if vv, ok := mongoValue.Pb2Model[v.TypeKey]; ok {
	//			utils.CopyStructFields(temp, &vv)
	//			if v.LatestUpdateTime != vv.LatestUpdateTime {
	//				setDeleteKeyResultInfo(&rResp, 300, "message have update", req.ClientMsgID, v.TypeKey, temp)
	//				continue
	//			}
	//		} else {
	//			setDeleteKeyResultInfo(&rResp, 400, "key not in", req.ClientMsgID, v.TypeKey, v)
	//			continue
	//		}
	//		temp.TypeKey = v.TypeKey
	//		setValue[v.TypeKey] = temp
	//	}
	//	err = db.DB.DeleteReactionExtendMsgSet(req.SourceID, req.SessionType, req.ClientMsgID, req.MsgFirstModifyTime, setValue)
	//	if err != nil {
	//		for _, value := range setValue {
	//			temp := new(msg.KeyValueResp)
	//			temp.KeyValue = value
	//			temp.ErrMsg = err.Error()
	//			temp.ErrCode = 100
	//			rResp.Result = append(rResp.Result, temp)
	//		}
	//	} else {
	//		for _, value := range setValue {
	//			temp := new(msg.KeyValueResp)
	//			temp.KeyValue = value
	//			rResp.Result = append(rResp.Result, temp)
	//		}
	//	}
	//	lockErr := m.dMessageLocker.UnLockGlobalMessage(req.ClientMsgID)
	//	if lockErr != nil {
	//		log.Error(req.OperationID, "UnLockGlobalMessage err:", lockErr.Error())
	//	}
	//
	//}
	//notification.ExtendMessageDeleteNotification(req.OperationID, req.OpUserID, req.SourceID, req.SessionType, req, &rResp, false, isExists)
	//log.Debug(req.OperationID, utils.GetSelfFuncName(), "m return is:", rResp.String())
	return resp, nil
}