package cn.exrick.xboot.core.serviceimpl;
|
|
import cn.exrick.xboot.core.common.constant.CommonConstant;
|
import cn.exrick.xboot.core.common.exception.XbootException;
|
import cn.exrick.xboot.core.dao.MessageDao;
|
import cn.exrick.xboot.core.dao.MessageSendDao;
|
import cn.exrick.xboot.core.entity.Message;
|
import cn.exrick.xboot.core.entity.MessageSend;
|
import cn.exrick.xboot.core.service.MessageSendService;
|
import cn.hutool.core.util.StrUtil;
|
import com.google.gson.Gson;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.data.domain.Page;
|
import org.springframework.data.domain.Pageable;
|
import org.springframework.data.jpa.domain.Specification;
|
import org.springframework.lang.Nullable;
|
import org.springframework.messaging.simp.SimpMessagingTemplate;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import javax.persistence.criteria.*;
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* 消息发送接口实现
|
* @author Exrick
|
*/
|
@Slf4j
|
@Service
|
@Transactional
|
public class MessageSendServiceImpl implements MessageSendService {
|
|
@Autowired
|
private MessageDao messageDao;
|
|
@Autowired
|
private MessageSendDao messageSendDao;
|
|
@Autowired
|
private SimpMessagingTemplate messagingTemplate;
|
|
@Override
|
public MessageSendDao getRepository() {
|
return messageSendDao;
|
}
|
|
@Override
|
public MessageSend send(MessageSend messageSend) {
|
|
MessageSend ms = messageSendDao.save(messageSend);
|
messagingTemplate.convertAndSendToUser(messageSend.getUserId(), "/queue/subscribe", "您收到了新的消息");
|
return ms;
|
}
|
|
@Override
|
public void deleteByMessageId(String messageId) {
|
|
messageSendDao.deleteByMessageId(messageId);
|
}
|
|
@Override
|
public Page<MessageSend> findByCondition(MessageSend messageSend, Pageable pageable) {
|
|
return messageSendDao.findAll(new Specification<MessageSend>() {
|
@Nullable
|
@Override
|
public Predicate toPredicate(Root<MessageSend> root, CriteriaQuery<?> cq, CriteriaBuilder cb) {
|
|
Path<String> messageIdField = root.get("messageId");
|
Path<String> userIdField = root.get("userId");
|
Path<Integer> statusField = root.get("status");
|
|
List<Predicate> list = new ArrayList<>();
|
|
if (StrUtil.isNotBlank(messageSend.getMessageId())) {
|
list.add(cb.equal(messageIdField, messageSend.getMessageId()));
|
}
|
if (StrUtil.isNotBlank(messageSend.getUserId())) {
|
list.add(cb.equal(userIdField, messageSend.getUserId()));
|
}
|
if (messageSend.getStatus() != null) {
|
list.add(cb.equal(statusField, messageSend.getStatus()));
|
}
|
|
Predicate[] arr = new Predicate[list.size()];
|
cq.where(list.toArray(arr));
|
return null;
|
}
|
}, pageable);
|
}
|
|
@Override
|
public void updateStatusByUserId(String userId, Integer status) {
|
|
messageSendDao.updateStatusByUserId(userId, status);
|
}
|
|
@Override
|
public void deleteByUserId(String userId) {
|
|
messageSendDao.deleteByUserId(userId, CommonConstant.MESSAGE_STATUS_READ);
|
}
|
|
@Override
|
public Message getTemplateMessage(String messageId, Map<String, String> params) {
|
|
Message message = messageDao.findById(messageId).orElse(null);
|
if (message == null) {
|
throw new XbootException("消息ID：" + messageId + "不存在");
|
}
|
// 放入变量
|
Message newMessage = new Message();
|
newMessage.setTitle(changeParams(message.getTitle(), params));
|
newMessage.setContent(changeParams(message.getContent(), params));
|
return newMessage;
|
}
|
|
@Override
|
public void sendTemplateMessage(List<String> userIds, String messageId, Map<String, String> params) {
|
|
Message message = messageDao.findById(messageId).orElse(null);
|
if (message == null) {
|
throw new XbootException("消息ID：" + messageId + "不存在");
|
}
|
List<MessageSend> messageSends = new ArrayList<>();
|
for (String userId : userIds) {
|
MessageSend ms = new MessageSend();
|
ms.setUserId(userId).setMessageId(messageId);
|
// 放入变量
|
ms.setTitle(changeParams(message.getTitle(), params));
|
ms.setContent(changeParams(message.getContent(), params));
|
ms.setParams(new Gson().toJson(params));
|
messageSends.add(ms);
|
messagingTemplate.convertAndSendToUser(userId, "/queue/subscribe", "您收到了新的消息");
|
}
|
messageSendDao.saveAll(messageSends);
|
}
|
|
@Override
|
public void sendTemplateMessage(String userId, String messageId, Map<String, String> params) {
|
|
List<String> users = new ArrayList<>();
|
users.add(userId);
|
this.sendTemplateMessage(users, messageId, params);
|
}
|
|
public String changeParams(String v, Map<String, String> params) {
|
|
if (v == null || params == null) {
|
return "";
|
}
|
for (Map.Entry entry : params.entrySet()) {
|
String key = entry.getKey().toString();
|
String value = entry.getValue().toString();
|
v = v.replaceAll("\\$\\{" + key + "\\}", value);
|
}
|
return v;
|
}
|
}
|
