文档库--发送消息添添加websocket

jero-boot/jero-boot-module-system/src/main/java/com/jero/modules/system/controller/SysAnnouncementSendController.java

@@ -8,8 +8,10 @@ import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.jero.common.api.vo.Result;
 import com.jero.common.constant.CommonConstant;
+import com.jero.common.constant.WebsocketConst;
 import com.jero.common.system.vo.LoginUser;
 import com.jero.common.util.oConvertUtils;
+import com.jero.modules.message.websocket.WebSocket;
 import com.jero.modules.system.entity.SysAnnouncementSend;
 import com.jero.modules.system.model.AnnouncementSendModel;
 import com.jero.modules.system.service.ISysAnnouncementSendService;
@@ -26,6 +28,7 @@ import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
 
+import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
 import java.util.ArrayList;
 import java.util.Arrays;
@@ -47,6 +50,8 @@ import java.util.List;
 public class SysAnnouncementSendController {
 	@Autowired
 	private ISysAnnouncementSendService sysAnnouncementSendService;
+	@Resource
+	private WebSocket webSocket;
 
 	/**
 	  * 分页列表查询
@@ -257,6 +262,12 @@ public class SysAnnouncementSendController {
 			list.add(sysAnnouncementSend);
 		}
 		sysAnnouncementSendService.updateBatchById(list);
+		//
+		JSONObject obj = new JSONObject();
+		obj.put(WebsocketConst.MSG_CMD, WebsocketConst.CMD_TOPIC);
+		obj.put(WebsocketConst.MSG_ID, ids);
+		obj.put(WebsocketConst.MSG_TXT, ids);
+		webSocket.sendMessage(obj.toJSONString());
 		result.setSuccess(true);
 		result.setMessage("标记已读");
 		return result;