/**
|
* Copyright (c) 2019 人人开源 All rights reserved.
|
*
|
* https://www.renren.io
|
*
|
* 版权所有，侵权必究！
|
*/
|
package com.zt.modules.message.service;
|
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
import java.util.stream.Collectors;
|
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import com.zt.common.db.query.QueryFilter;
|
import com.zt.common.service.BaseService;
|
import com.zt.core.context.UserContext;
|
import com.zt.core.sys.service.ISysUserService;
|
import com.zt.modules.message.dao.SysNoticeDao;
|
import com.zt.modules.message.enums.NoticeReadStatus;
|
import com.zt.modules.message.enums.NoticeStatus;
|
import com.zt.modules.message.enums.ReceiverType;
|
import com.zt.modules.message.model.SysNotice;
|
import com.zt.modules.message.model.SysNoticeUser;
|
import com.zt.modules.message.websocket.WebSocketServer;
|
import com.zt.modules.message.websocket.data.MessageData;
|
|
/**
|
* 通知管理
|
*
|
* @author hehz
|
*/
|
@Service
|
public class SysNoticeService extends BaseService<SysNoticeDao, SysNotice> {
|
@Autowired
|
private SysNoticeUserService sysNoticeUserService;
|
@Autowired
|
private ISysUserService sysUserService;
|
@Autowired
|
private WebSocketServer webSocketServer;
|
|
public List<SysNotice> page(QueryFilter queryFilter) {
|
return super.query(queryFilter);
|
}
|
|
/**
|
* 获取我的通知列表
|
*/
|
public List<SysNotice> getMyNoticePage(QueryFilter queryFilter) {
|
// 查询
|
List<SysNotice> list = baseDao.getMyNoticeList(UserContext.getUserId());
|
|
return queryFilter.getPageList(list);
|
}
|
|
@Transactional(rollbackFor = Exception.class)
|
public void insert(SysNotice entity) {
|
// 更新发送者信息
|
if (entity.getStatus() == NoticeStatus.SEND.value()) {
|
entity.setSenderId(UserContext.getUser().getId());
|
entity.setSendDate(new Date());
|
}
|
entity.setReceiverIds(StringUtils.join(entity.getReceiverList(), ","));
|
baseDao.insert(entity);
|
|
// 发送通知
|
sendNotice(entity);
|
}
|
|
@Transactional(rollbackFor = Exception.class)
|
public void update(SysNotice entity) {
|
// 更新发送者信息
|
if (entity.getStatus() == NoticeStatus.SEND.value()) {
|
entity.setSenderId(UserContext.getUser().getId());
|
entity.setSendDate(new Date());
|
}
|
entity.setReceiverIds(StringUtils.join(entity.getReceiverList(), ","));
|
baseDao.updateById(entity);
|
|
// 发送通知
|
sendNotice(entity);
|
}
|
|
/**
|
* 发送通知
|
*/
|
public void sendNotice(SysNotice notice) {
|
// 如果是草稿，在不发送通知
|
if (notice.getStatus() == NoticeStatus.DRAFT.value()) {
|
return;
|
}
|
|
List<Long> userIdList = new ArrayList<>();
|
|
if (ReceiverType.DEPT.getValue().equals(notice.getReceiverType())) {// 部门
|
userIdList.addAll(sysUserService.getByDeptIds(notice.getReceiverList()).stream().map(user -> user.getId())
|
.collect(Collectors.toList()));
|
} else { // 选中用户
|
userIdList.addAll(notice.getReceiverList());
|
}
|
if (userIdList.size() > 0) {
|
// 发送给用户
|
userIdList.forEach(userId -> {
|
SysNoticeUser noticeUser = new SysNoticeUser().setNoticeId(notice.getId()).setReceiverId(userId)
|
.setStatus(NoticeReadStatus.UNREAD.value());
|
|
sysNoticeUserService.insert(noticeUser);
|
});
|
|
// 通过WebSocket，提示选中用户，有新通知
|
MessageData<String> message = new MessageData<String>().msg(notice.getTitle());
|
webSocketServer.sendMessage(userIdList, message);
|
}
|
}
|
}
|
