基于jeecgboot的OA日程安排开发（二）

package org.jeecg.modules.message.job;
import java.util.List;
import org.jeecg.common.util.DateUtils;
import org.jeecg.modules.message.entity.SysMessage;
import org.jeecg.modules.message.handle.ISendMsgHandle;
import org.jeecg.modules.message.handle.enums.SendMsgStatusEnum;
import org.jeecg.modules.message.handle.enums.SendMsgTypeEnum;
import org.jeecg.modules.message.service.ISysMessageService;
import org.quartz.Job;
import org.quartz.JobExecutionContext;
import org.quartz.JobExecutionException;
import org.springframework.beans.factory.annotation.Autowired;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import lombok.extern.slf4j.Slf4j;
/**
 * @Description: 发送websocket消息job
 * @author: nbacheng
 * @date: 2023-05-08
 * @version: V1.0
 */
@Slf4j
public class SendWebMsgJob implements Job {
  @Autowired
  private ISysMessageService sysMessageService;
  
  @Override
  public void execute(JobExecutionContext context) throws JobExecutionException {
    log.info(String.format(" Nbcio-Boot 发送消息任务 SendWebMsgJob !  时间:" + DateUtils.getTimestamp()));
    // 1.读取消息中心数据，只查询未发送的和发送失败不超过次数的
    QueryWrapper<SysMessage> queryWrapper = new QueryWrapper<SysMessage>();
    queryWrapper.eq("es_send_status", SendMsgStatusEnum.WAIT.getCode())
        .or(i -> i.eq("es_send_status", SendMsgStatusEnum.FAIL.getCode()).lt("es_send_num", 6));
    List<SysMessage> sysMessages = sysMessageService.list(queryWrapper);
    System.out.println(sysMessages);
    // 2.根据不同的类型走不通的发送实现类
    for (SysMessage sysMessage : sysMessages) {
      ISendMsgHandle sendMsgHandle = null;
      try {
        if (sysMessage.getEsType().equals(SendMsgTypeEnum.WEBMSG.getType())) {//目前只发送websocket消息
          sendMsgHandle = (ISendMsgHandle) Class.forName(SendMsgTypeEnum.WEBMSG.getImplClass()).newInstance();
        } 
      } catch (Exception e) {
        log.error(e.getMessage(),e);
      }
      Integer sendNum = sysMessage.getEsSendNum();
      try {
        sendMsgHandle.SendMsg(sysMessage.getEsReceiver(), sysMessage.getEsTitle(),
            sysMessage.getEsContent().toString(),sysMessage.getCreateBy());
        // 发送消息成功
        sysMessage.setEsSendStatus(SendMsgStatusEnum.SUCCESS.getCode());
      } catch (Exception e) {
        e.printStackTrace();
        // 发送消息出现异常
        sysMessage.setEsSendStatus(SendMsgStatusEnum.FAIL.getCode());
      }
      sysMessage.setEsSendNum(++sendNum);
      // 发送结果回写到数据库
      sysMessageService.updateById(sysMessage);
    }
  }
}

package org.jeecg.modules.message.handle.impl;
import lombok.extern.slf4j.Slf4j;
import org.jeecg.common.api.dto.message.MessageDTO;
import org.jeecg.common.system.api.ISysBaseAPI;
import org.jeecg.common.util.SpringContextUtils;
import org.jeecg.modules.message.handle.ISendMsgHandle;
@Slf4j
public class WebMsgSendMsgHandle implements ISendMsgHandle {
  @Override
  public void SendMsg(String es_receiver, String es_title, String es_content, String es_fromuser) {
    log.info("WebMsgSendMsgHandle 发websocket消息");
    ISysBaseAPI sysBaseAPI = (ISysBaseAPI) SpringContextUtils.getBean(ISysBaseAPI.class);
    MessageDTO messagedto = new MessageDTO();
    messagedto.setContent(es_content);
    messagedto.setCategory("1"); //放到通知通告了
    messagedto.setFromUser(es_fromuser);
    messagedto.setTitle(es_title);
    messagedto.setToUser(es_receiver);
    sysBaseAPI.sendSysAnnouncement(messagedto);
    
  }
}