From e4e5608c8c6c78afded9400672e9ed4b8d3b3d4c Mon Sep 17 00:00:00 2001
From: wenxu12345 <44203734@qq.com>
Date: Tue, 7 Dec 2021 14:34:41 +0800
Subject: [PATCH] ReceiveMessageOpt

---
 internal/rpc/user/conversation.go | 59 +++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 internal/rpc/user/conversation.go

diff --git a/internal/rpc/user/conversation.go b/internal/rpc/user/conversation.go
new file mode 100644
index 0000000..1f87c1c
--- /dev/null
+++ b/internal/rpc/user/conversation.go
@@ -0,0 +1,59 @@
+package user
+
+import (
+	"Open_IM/pkg/common/constant"
+	"Open_IM/pkg/common/db"
+	"Open_IM/pkg/common/log"
+	pbUser "Open_IM/pkg/proto/user"
+	"context"
+)
+
+func (s *userServer) SetReceiveMessageOpt(ctx context.Context, req *pbUser.SetReceiveMessageOptReq) (*pbUser.SetReceiveMessageOptResp, error) {
+	m := make(map[string]int, len(req.ConversationId))
+	for _, v := range req.ConversationId {
+		m[v] = int(req.Opt)
+	}
+	err := db.DB.SetMultiConversationMsgOpt(req.UId, m)
+	if err != nil {
+		log.NewError(req.OperationID, "SetMultiConversationMsgOpt failed ", err.Error(), req)
+		return &pbUser.SetReceiveMessageOptResp{ErrCode: constant.DatabaseError, ErrMsg: err.Error()}, nil
+	}
+	var resp pbUser.SetReceiveMessageOptResp
+	resp.ErrCode = 0
+
+	for _, v := range req.ConversationId {
+		resp.OptResult = append(resp.OptResult, &pbUser.OptResult{ConversationId: v, Result: 0})
+	}
+	log.NewInfo(req.OperationID, "SetReceiveMessageOpt req, resp ", req, resp)
+	return &resp, nil
+}
+
+func (s *userServer) GetReceiveMessageOpt(ctx context.Context, req *pbUser.GetReceiveMessageOptReq) (*pbUser.GetReceiveMessageOptResp, error) {
+	m, err := db.DB.GetMultiConversationMsgOpt(req.UId, req.ConversationId)
+	if err != nil {
+		log.NewError(req.OperationID, "GetMultiConversationMsgOpt failed ", err.Error(), req)
+		return &pbUser.GetReceiveMessageOptResp{ErrCode: constant.DatabaseError, ErrMsg: err.Error()}, nil
+	}
+	var resp pbUser.GetReceiveMessageOptResp
+	resp.ErrCode = 0
+	for k, v := range m {
+		resp.ConversationOptResult = append(resp.ConversationOptResult, &pbUser.OptResult{ConversationId: k, Result: int32(v)})
+	}
+	log.NewInfo(req.OperationID, "GetReceiveMessageOpt, req, resp", req, resp)
+	return &resp, nil
+}
+
+func (s *userServer) GetAllConversationMsgOpt(ctx context.Context, req *pbUser.GetAllConversationMsgOptReq) (*pbUser.GetAllConversationMsgOptResp, error) {
+	m, err := db.DB.GetAllConversationMsgOpt(req.UId)
+	if err != nil {
+		log.NewError(req.OperationID, "GetAllConversationMsgOpt failed ", err.Error(), req)
+		return &pbUser.GetAllConversationMsgOptResp{ErrCode: constant.DatabaseError, ErrMsg: err.Error()}, nil
+	}
+	var resp pbUser.GetAllConversationMsgOptResp
+	resp.ErrCode = 0
+	for k, v := range m {
+		resp.ConversationOptResult = append(resp.ConversationOptResult, &pbUser.OptResult{ConversationId: k, Result: int32(v)})
+	}
+	log.NewInfo(req.OperationID, "GetAllConversationMsgOpt, req, resp", req, resp)
+	return &resp, nil
+}
-- 
GitLab