调整MessageAction代码

core/src/main/java/com/wecloud/im/action/MessageAction.java

 package com.wecloud.im.action;
 
+import io.geekidea.springbootplus.framework.core.pagination.Paging;
+import io.swagger.annotations.Api;
+import io.swagger.annotations.ApiOperation;
+import lombok.extern.slf4j.Slf4j;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
 import com.alibaba.fastjson.JSON;
 import com.wecloud.dispatch.annotation.ActionMapping;
 import com.wecloud.dispatch.extend.ActionRequest;
@@ -13,13 +24,6 @@ import com.wecloud.im.vo.OfflineMsgDto;
 import com.wecloud.im.vo.ReaderList;
 import com.wecloud.im.ws.model.WsResponse;
 import com.wecloud.im.ws.sender.ChannelSender;
-import io.geekidea.springbootplus.framework.core.pagination.Paging;
-import io.netty.channel.socket.nio.NioSocketChannel;
-import io.swagger.annotations.Api;
-import io.swagger.annotations.ApiOperation;
-import lombok.extern.slf4j.Slf4j;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.stereotype.Component;
 
 /**
  * @Author wenzhida
@@ -46,11 +50,12 @@ public class MessageAction {
      */
     @ActionMapping("/withdraw")
     @ApiOperation("消息撤回")
-    public void withdraw(ActionRequest request, ImMsgRecall msgRecall, String reqId) {
+    public WsResponse<Map<String, Boolean>> withdraw(ActionRequest request, ImMsgRecall msgRecall, String reqId) {
         log.info("撤回消息内容：{}", JSON.toJSONString(msgRecall));
-        WsResponse<Boolean> wsResponse = new WsResponse<>();
-        wsResponse.setData(imMessageService.withdrawMsg(request, msgRecall));
-        channelSender.sendMsgLocal((NioSocketChannel)request.getSenderChannel(), wsResponse);
+        boolean r = imMessageService.withdrawMsg(request, msgRecall);
+        Map<String, Boolean> result = new HashMap<>();
+        result.put("result", r);
+        return WsResponse.ok(result);
     }
 
     /**
@@ -61,11 +66,12 @@ public class MessageAction {
      */
     @ActionMapping("/delete")
     @ApiOperation("删除消息内容")
-    public void deleteMsg(ActionRequest request, MsgDeleteParam msgDeleteParam, String reqId) {
+    public WsResponse<Map<String, Boolean>> deleteMsg(ActionRequest request, MsgDeleteParam msgDeleteParam, String reqId) {
         log.info("删除消息内容：{}", JSON.toJSONString(msgDeleteParam));
-        WsResponse<Boolean> wsResponse = new WsResponse<>();
-        wsResponse.setData(imMessageService.deleteMsg(request, msgDeleteParam));
-        channelSender.sendMsgLocal((NioSocketChannel)request.getSenderChannel(), wsResponse);
+        boolean r = imMessageService.deleteMsg(request, msgDeleteParam);
+        Map<String, Boolean> result = new HashMap<>();
+        result.put("result", r);
+        return WsResponse.ok(result);
     }
 
     /**
@@ -76,11 +82,12 @@ public class MessageAction {
      */
     @ActionMapping("/updateMsgById")
     @ApiOperation("修改消息体")
-    public void updateMsgById(ActionRequest request, ImMsgUpdate imMsgUpdate, String reqId) {
+    public WsResponse<Map<String, Boolean>> updateMsgById(ActionRequest request, ImMsgUpdate imMsgUpdate, String reqId) {
         log.info("修改消息内容：{}", JSON.toJSONString(imMsgUpdate));
-        WsResponse<Boolean> wsResponse = new WsResponse<>();
-        wsResponse.setData(imMessageService.updateMsgById(imMsgUpdate));
-        channelSender.sendMsgLocal((NioSocketChannel)request.getSenderChannel(), wsResponse);
+        boolean r = imMessageService.updateMsgById(imMsgUpdate);
+        Map<String, Boolean> result = new HashMap<>();
+        result.put("result", r);
+        return WsResponse.ok(result);
     }
 
     /**
@@ -88,14 +95,13 @@ public class MessageAction {
      * @param request
      * @param param
      * @param reqId
+     * @return
      */
     @ActionMapping("/getHistoryMsg")
     @ApiOperation("查询某个会话历史消息分页列表")
-    public void getHistoryMsg(ActionRequest request, ImHistoryMessagePageParam param, String reqId) {
+    public WsResponse<Paging<OfflineMsgDto>> getHistoryMsg(ActionRequest request, ImHistoryMessagePageParam param, String reqId) {
         log.info("查询某个会话历史消息分页列表：{}", JSON.toJSONString(param));
-        WsResponse<Paging<OfflineMsgDto>> wsResponse = new WsResponse<>();
-        wsResponse.setData(imMessageService.getHistoryMsgConversationId(param));
-        channelSender.sendMsgLocal((NioSocketChannel)request.getSenderChannel(), wsResponse);
+        return WsResponse.ok(imMessageService.getHistoryMsgConversationId(param));
     }
 
     /**
@@ -103,14 +109,13 @@ public class MessageAction {
      * @param request
      * @param param
      * @param reqId
+     * @return
      */
     @ActionMapping("/getReaders")
     @ApiOperation("查询某个消息已读client列表和未读client")
-    public void getReaders(ActionRequest request, GetReadersParam param, String reqId) {
+    public WsResponse<ReaderList> getReaders(ActionRequest request, GetReadersParam param, String reqId) {
         log.info("查询某个消息已读client列表和未读client：{}", JSON.toJSONString(param));
-        WsResponse<ReaderList> wsResponse = new WsResponse<>();
-        wsResponse.setData(imMessageService.getReaders(request, param));
-        channelSender.sendMsgLocal((NioSocketChannel)request.getSenderChannel(), wsResponse);
+        return WsResponse.ok(imMessageService.getReaders(request, param));
     }
 
 }