lcj/xinnengyuan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

消息优化

Browse Source
This commit is contained in:
zt
2025-08-26 14:58:08 +08:00
parent 508d81bedf
commit 971101d5e8
19 changed files with 249 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/land/controller/listener/GenericExcelListener.java → xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/common/utils/listener/excel/GenericExcelListener.java
View File

@ -1,11 +1,10 @@
package org.dromara.land.controller.listener;
package org.dromara.common.utils.listener.excel;
import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.alibaba.excel.exception.ExcelDataConvertException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.interceptor.TransactionAspectSupport;
import java.util.ArrayList;

54
xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/common/utils/listener/redis/SseMessageDbListener.java Normal file
View File

@ -0,0 +1,54 @@
package org.dromara.common.utils.listener.redis;
import cn.hutool.core.collection.CollUtil;
import lombok.extern.slf4j.Slf4j;
import org.dromara.common.sse.core.SseEmitterManager;
import org.dromara.message.domain.MsgNotice;
import org.dromara.message.service.IMsgNoticeService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.core.Ordered;
import org.springframework.stereotype.Component;
import java.util.ArrayList;
/**
* SSE 消息数据库存储监听器
*/
@Slf4j
@Component
public class SseMessageDbListener implements ApplicationRunner, Ordered {
@Autowired
private SseEmitterManager sseEmitterManager;
@Autowired
private IMsgNoticeService noticeService; // 注入你的数据库服务
@Override
public void run(ApplicationArguments args) throws Exception {
sseEmitterManager.subscribeMessage((message) -> {
log.info("SSE主题订阅收到消息session keys={} message={}", message.getUserIds(), message.getMessage());
if (CollUtil.isNotEmpty(message.getUserIds())) {
ArrayList<MsgNotice> noticeList = new ArrayList<>();
for (Long key : message.getUserIds()) {
MsgNotice notice = new MsgNotice();
notice.setContent(message.getMessage());
notice.setProjectId(message.getProjectId());
notice.setRecipientId(key);
noticeList.add(notice);
}
noticeService.saveBatch(noticeList);
}
log.info("SSE消息已存储到数据库: {}", message.getMessage());
});
log.info("初始化SSE消息数据库存储监听器成功");
}
@Override
public int getOrder() {
return 0; // 可以调整顺序
}
}

66
xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/design/service/impl/DesPrelimSchemeServiceImpl.java
View File

@ -98,7 +98,7 @@ public class DesPrelimSchemeServiceImpl extends ServiceImpl<DesPrelimSchemeMappe
private LambdaQueryWrapper<DesPrelimScheme> buildQueryWrapper(DesPrelimSchemeBo bo) {
Map<String, Object> params = bo.getParams();
LambdaQueryWrapper<DesPrelimScheme> lqw = Wrappers.lambdaQuery();
lqw.orderByAsc(DesPrelimScheme::getOrderNum);
lqw.orderByDesc(DesPrelimScheme::getCreateTime);
lqw.orderByDesc(DesPrelimScheme::getId);
lqw.eq(bo.getProjectId() != null, DesPrelimScheme::getProjectId, bo.getProjectId());
lqw.eq(bo.getOssId()!= null, DesPrelimScheme::getOssId, bo.getOssId());
@ -207,38 +207,38 @@ public class DesPrelimSchemeServiceImpl extends ServiceImpl<DesPrelimSchemeMappe
}
this.updateById(prelimScheme);
//状态为已完成发送消息
if (BusinessStatusEnum.FINISH.getStatus().equals(processEvent.getStatus())) {
//发送消息
MsgConfig msgConfig = msgConfigService.getOne(
new LambdaQueryWrapper<MsgConfig>().
eq(MsgConfig::getProjectId, prelimScheme.getProjectId()).
eq(MsgConfig::getMsgKey, "prelimScheme")
);
if (msgConfig != null) {
String userId = msgConfig.getUserId();
List<Long> userIdList = Arrays.stream(userId.split(","))
.map(String::trim)
.map(s -> {
try {
return Long.valueOf(s);
} catch (NumberFormatException e) {
// 处理转换失败的情况
return null; // 或者根据需求处理
}
})
.filter(Objects::nonNull) // 过滤掉转换失败的null值
.collect(Collectors.toList());
//发送消息
SendMsgDto sendMsgDto = new SendMsgDto();
sendMsgDto.setProjectId(prelimScheme.getProjectId());
sendMsgDto.setRecipientIds(userIdList);
sendMsgDto.setSenderId(0L);
sendMsgDto.setConfigId(msgConfig.getId());
sendMsgDto.setDetailId(prelimScheme.getId().toString());
sendMsgDto.setContent(msgConfig.getMsgContent());
msgNoticeService.sendMsg(sendMsgDto);
}
}
// if (BusinessStatusEnum.FINISH.getStatus().equals(processEvent.getStatus())) {
// //发送消息
// MsgConfig msgConfig = msgConfigService.getOne(
// new LambdaQueryWrapper<MsgConfig>().
// eq(MsgConfig::getProjectId, prelimScheme.getProjectId()).
// eq(MsgConfig::getMsgKey, "prelimScheme")
// );
// if (msgConfig != null) {
// String userId = msgConfig.getUserId();
// List<Long> userIdList = Arrays.stream(userId.split(","))
// .map(String::trim)
// .map(s -> {
// try {
// return Long.valueOf(s);
// } catch (NumberFormatException e) {
// // 处理转换失败的情况
// return null; // 或者根据需求处理
// }
// })
// .filter(Objects::nonNull) // 过滤掉转换失败的null值
// .collect(Collectors.toList());
// //发送消息
// SendMsgDto sendMsgDto = new SendMsgDto();
// sendMsgDto.setProjectId(prelimScheme.getProjectId());
// sendMsgDto.setRecipientIds(userIdList);
// sendMsgDto.setSenderId(0L);
// sendMsgDto.setConfigId(msgConfig.getId());
// sendMsgDto.setDetailId(prelimScheme.getId().toString());
// sendMsgDto.setContent(msgConfig.getMsgContent());
// msgNoticeService.sendMsg(sendMsgDto);
// }
// }
}
/**

2
xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/design/service/impl/DesSchemeServiceImpl.java
View File

@ -93,7 +93,7 @@ public class DesSchemeServiceImpl extends ServiceImpl<DesSchemeMapper, DesScheme
private LambdaQueryWrapper<DesScheme> buildQueryWrapper(DesSchemeBo bo) {
Map<String, Object> params = bo.getParams();
LambdaQueryWrapper<DesScheme> lqw = Wrappers.lambdaQuery();
lqw.orderByAsc(DesScheme::getOrderNum);
lqw.orderByDesc(DesScheme::getCreateTime);
lqw.orderByDesc(DesScheme::getId);
lqw.eq(bo.getProjectId() != null, DesScheme::getProjectId, bo.getProjectId());
lqw.eq(bo.getOssId()!= null, DesScheme::getOssId, bo.getOssId());

18
xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/facility/service/impl/FacPhotovoltaicPanelPartsServiceImpl.java
View File

@ -12,6 +12,8 @@ import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.mybatis.core.page.PageQuery;
import org.dromara.common.mybatis.core.page.TableDataInfo;
import org.dromara.common.satoken.utils.LoginHelper;
import org.dromara.common.sse.config.SseProperties;
import org.dromara.common.sse.dto.SeeMessageContentDto;
import org.dromara.common.sse.dto.SseMessageDto;
import org.dromara.common.sse.utils.SseMessageUtils;
import org.dromara.common.utils.JSTUtil;
@ -46,6 +48,7 @@ import java.util.concurrent.*;
import java.util.function.Function;
import java.util.stream.Collectors;
/**
* @author lilemy
* @date 2025/4/21 15:53
@ -85,6 +88,9 @@ public class FacPhotovoltaicPanelPartsServiceImpl implements IFacPhotovoltaicPan
@Resource
private FacPhotovoltaicPanelPartsServiceImpl self; // 注入自己
@Resource
private SseProperties sseProperties;
/**
* 查询光伏板（桩点、立柱、支架）列表
*
@ -153,7 +159,10 @@ public class FacPhotovoltaicPanelPartsServiceImpl implements IFacPhotovoltaicPan
log.info("已接收第 {} 批数据，共 {} 批", batchNum, totalBatch);
Long userId = LoginHelper.getUserId();
SseMessageDto messageDto = new SseMessageDto();
SeeMessageContentDto contentDto = new SeeMessageContentDto();
contentDto.setType(sseProperties.getProject());
messageDto.setUserIds(List.of(userId));
messageDto.setProjectId(projectId);
// 如果是最后一批，开始合并
if (batchNum == totalBatch) {
List<FacFeatureByPoint> allData = new ArrayList<>();
@ -169,14 +178,16 @@ public class FacPhotovoltaicPanelPartsServiceImpl implements IFacPhotovoltaicPan
// 设置 redis key，防止多次操作
String operationPartsRedisKey = FacRedisKeyConstant.getPartsInOperationByProjectRedisKey(projectId);
redisTemplate.opsForValue().set(operationPartsRedisKey, true);
messageDto.setMessage("桩点、立柱、支架数据上传完毕，正在处理中");
contentDto.setContent("桩点、立柱、支架数据上传完毕，正在处理中");
messageDto.setMessage(JSONUtil.toJsonStr(contentDto));
SseMessageUtils.publishMessage(messageDto);
scheduledExecutorService.execute(() -> {
try {
// 合并后的数据处理，如入库
this.saveBatch(projectId, allData, userId);
} catch (Exception e) {
messageDto.setMessage("桩点、立柱、支架数据处理失败，请联系管理员处理");
contentDto.setContent("桩点、立柱、支架数据处理失败，请联系管理员处理");
messageDto.setMessage(JSONUtil.toJsonStr(contentDto));
SseMessageUtils.publishMessage(messageDto);
log.error("桩点、立柱、支架数据处理失败", e);
throw new ServiceException("桩点、立柱、支架数据处理失败", HttpStatus.ERROR);
@ -188,7 +199,8 @@ public class FacPhotovoltaicPanelPartsServiceImpl implements IFacPhotovoltaicPan
}
redisTemplate.delete(operationPartsRedisKey);
}
messageDto.setMessage("桩点、立柱、支架数据处理完毕");
contentDto.setContent("桩点、立柱、支架数据处理完毕");
messageDto.setMessage(JSONUtil.toJsonStr(contentDto));
SseMessageUtils.publishMessage(messageDto);
});
return true;

17
xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/facility/service/impl/FacPhotovoltaicPanelServiceImpl.java
View File

@ -14,6 +14,8 @@ import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.mybatis.core.page.PageQuery;
import org.dromara.common.mybatis.core.page.TableDataInfo;
import org.dromara.common.satoken.utils.LoginHelper;
import org.dromara.common.sse.config.SseProperties;
import org.dromara.common.sse.dto.SeeMessageContentDto;
import org.dromara.common.sse.dto.SseMessageDto;
import org.dromara.common.sse.utils.SseMessageUtils;
import org.dromara.common.utils.JSTUtil;
@ -83,6 +85,9 @@ public class FacPhotovoltaicPanelServiceImpl extends ServiceImpl<FacPhotovoltaic
@Resource
private FacPhotovoltaicPanelServiceImpl self; // 注入自己
@Resource
private SseProperties sseProperties;
/**
* 查询设施-光伏板
*
@ -214,7 +219,10 @@ public class FacPhotovoltaicPanelServiceImpl extends ServiceImpl<FacPhotovoltaic
log.info("已接收第 {} 批数据，共 {} 批", batchNum, totalBatch);
Long userId = LoginHelper.getUserId();
SseMessageDto messageDto = new SseMessageDto();
SeeMessageContentDto contentDto = new SeeMessageContentDto();
contentDto.setType(sseProperties.getProject());
messageDto.setUserIds(List.of(userId));
messageDto.setProjectId(projectId);
// 如果是最后一批，开始合并
if (batchNum == totalBatch) {
List<FacFeatureByPlane> locationFeatures = new ArrayList<>();
@ -238,7 +246,8 @@ public class FacPhotovoltaicPanelServiceImpl extends ServiceImpl<FacPhotovoltaic
// 设置 redis key，防止多次操作
String operationPanelRedisKey = FacRedisKeyConstant.getPanelInOperationByProjectRedisKey(projectId);
redisTemplate.opsForValue().set(operationPanelRedisKey, true);
messageDto.setMessage("光伏板数据上传完毕，正在处理中");
contentDto.setContent("光伏板数据上传完毕，正在处理中");
messageDto.setMessage(JSONUtil.toJsonStr(contentDto));
SseMessageUtils.publishMessage(messageDto);
scheduledExecutorService.execute(() -> {
try {
@ -388,10 +397,12 @@ public class FacPhotovoltaicPanelServiceImpl extends ServiceImpl<FacPhotovoltaic
if (!result) {
throw new ServiceException("更新进度类别失败，数据库异常", HttpStatus.ERROR);
}
messageDto.setMessage("光伏板数据处理完毕");
contentDto.setContent("光伏板数据处理完毕");
messageDto.setMessage(JSONUtil.toJsonStr(contentDto));
SseMessageUtils.publishMessage(messageDto);
} catch (Exception e) {
messageDto.setMessage("光伏板数据处理失败，请联系管理员处理");
contentDto.setContent("光伏板数据处理失败，请联系管理员处理");
messageDto.setMessage(JSONUtil.toJsonStr(contentDto));
SseMessageUtils.publishMessage(messageDto);
log.error("光伏板数据处理失败", e);
throw new ServiceException("光伏板数据处理失败", HttpStatus.ERROR);

5
xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/land/controller/BusEnterRoadController.java
View File

@ -19,12 +19,10 @@ import org.dromara.common.log.enums.BusinessType;
import org.dromara.common.mybatis.core.page.PageQuery;
import org.dromara.common.mybatis.core.page.TableDataInfo;
import org.dromara.common.web.core.BaseController;
import org.dromara.land.controller.listener.GenericExcelListener;
import org.dromara.common.utils.listener.excel.GenericExcelListener;
import org.dromara.land.domain.BusEnterRoad;
import org.dromara.land.domain.BusLandBlock;
import org.dromara.land.domain.bo.BusEnterRoadBo;
import org.dromara.land.domain.bo.BusEnterRoadImportBo;
import org.dromara.land.domain.bo.BusLandBlockImportBo;
import org.dromara.land.domain.vo.BusEnterRoadVo;
import org.dromara.land.service.IBusEnterRoadService;
import org.springframework.validation.annotation.Validated;
@ -41,7 +39,6 @@ import org.springframework.web.multipart.MultipartFile;
import java.io.IOException;
import java.util.List;
import java.util.UUID;
import java.util.concurrent.ThreadLocalRandom;
import java.util.stream.Collectors;
/**

3
xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/land/controller/BusLandBlockController.java
View File

@ -19,7 +19,7 @@ import org.dromara.common.log.enums.BusinessType;
import org.dromara.common.mybatis.core.page.PageQuery;
import org.dromara.common.mybatis.core.page.TableDataInfo;
import org.dromara.common.web.core.BaseController;
import org.dromara.land.controller.listener.GenericExcelListener;
import org.dromara.common.utils.listener.excel.GenericExcelListener;
import org.dromara.land.domain.BusLandBlock;
import org.dromara.land.domain.bo.BusLandBlockBo;
import org.dromara.land.domain.bo.BusLandBlockImportBo;
@ -32,7 +32,6 @@ import org.springframework.web.multipart.MultipartFile;
import java.io.IOException;
import java.util.List;
import java.util.UUID;
import java.util.concurrent.ThreadLocalRandom;
import java.util.stream.Collectors;
/**

3
xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/message/service/impl/MsgNoticeServiceImpl.java
View File

@ -200,7 +200,6 @@ public class MsgNoticeServiceImpl extends ServiceImpl<MsgNoticeMapper, MsgNotice
messageDto2.setUserIds(recipientIds);
messageDto2.setMessage(bo.getContent());
messageDto2.setRoute(byId.getRoute());
messageDto2.setDetailId(bo.getDetailId());
Gson gson = new Gson();
String messageDtoJson = gson.toJson(messageDto2);
@ -214,7 +213,7 @@ public class MsgNoticeServiceImpl extends ServiceImpl<MsgNoticeMapper, MsgNotice
}
/**

17
xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/project/service/impl/BusProjectFileServiceImpl.java
View File

@ -19,6 +19,8 @@ import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.mybatis.core.page.PageQuery;
import org.dromara.common.mybatis.core.page.TableDataInfo;
import org.dromara.common.satoken.utils.LoginHelper;
import org.dromara.common.sse.config.SseProperties;
import org.dromara.common.sse.dto.SeeMessageContentDto;
import org.dromara.common.sse.dto.SseMessageDto;
import org.dromara.common.sse.utils.SseMessageUtils;
import org.dromara.common.utils.Dxf2JsonUtil;
@ -71,6 +73,9 @@ public class BusProjectFileServiceImpl extends ServiceImpl<BusProjectFileMapper,
@Value("${dxf2GeoJson.file-name}")
private String dxf2GeoJsonFileName;
@Resource
private SseProperties sseProperties;
/**
* 查询项目文件存储
*
@ -292,8 +297,12 @@ public class BusProjectFileServiceImpl extends ServiceImpl<BusProjectFileMapper,
projectFileId = null;
}
SseMessageDto messageDto = new SseMessageDto();
messageDto.setMessage("DXF 文件上传成功，正在转换中......");
SeeMessageContentDto contentDto = new SeeMessageContentDto();
contentDto.setType(sseProperties.getProject());
contentDto.setContent("DXF 文件上传成功，正在转换中......");
messageDto.setMessage(JSONUtil.toJsonStr(contentDto));
messageDto.setUserIds(List.of(userId));
messageDto.setProjectId(projectId);
SseMessageUtils.publishMessage(messageDto);
scheduledExecutorService.execute(() -> {
try {
@ -311,11 +320,13 @@ public class BusProjectFileServiceImpl extends ServiceImpl<BusProjectFileMapper,
projectFile.setRemark(dxfFilePath);
boolean update = this.saveOrUpdate(projectFile);
if (!update) throw new ServiceException("数据库修改异常");
messageDto.setMessage("DXF 文件转换成 GeoJSON 成功");
contentDto.setContent("DXF 文件转换成 GeoJSON 成功");
messageDto.setMessage(JSONUtil.toJsonStr(contentDto));
SseMessageUtils.publishMessage(messageDto);
} catch (Exception e) {
log.error("DXF 转换失败", e);
messageDto.setMessage("DXF 文件转换失败，请联系管理员处理");
contentDto.setContent("DXF 文件转换失败，请联系管理员处理");
messageDto.setMessage(JSONUtil.toJsonStr(contentDto));
SseMessageUtils.publishMessage(messageDto);
} finally {
// 无论成功或失败都释放状态

12
xinnengyuan/ruoyi-modules/ruoyi-system/src/main/java/org/dromara/safety/service/impl/HseViolationRecordServiceImpl.java
View File

@ -1,6 +1,7 @@
package org.dromara.safety.service.impl;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.json.JSONUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
@ -13,6 +14,8 @@ import org.dromara.common.core.utils.ObjectUtils;
import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.mybatis.core.page.PageQuery;
import org.dromara.common.mybatis.core.page.TableDataInfo;
import org.dromara.common.sse.config.SseProperties;
import org.dromara.common.sse.dto.SeeMessageContentDto;
import org.dromara.common.sse.dto.SseMessageDto;
import org.dromara.common.sse.utils.SseMessageUtils;
import org.dromara.safety.domain.HseRecognizeRecord;
@ -65,6 +68,9 @@ public class HseViolationRecordServiceImpl extends ServiceImpl<HseViolationRecor
@Resource
private IHseViolationLevelPostService violationLevelPostService;
@Resource
private SseProperties sseProperties;
/**
* 查询违规记录
*
@ -193,7 +199,11 @@ public class HseViolationRecordServiceImpl extends ServiceImpl<HseViolationRecor
SseMessageDto messageDto = new SseMessageDto();
messageDto.setUserIds(new ArrayList<>(userIds));
for (HseViolationRecord violationRecord : violationRecordList) {
messageDto.setMessage("您有一份重要的违章需要处理！工单号为：" + violationRecord.getId());
SeeMessageContentDto seeMessageContentDto = new SeeMessageContentDto();
seeMessageContentDto.setType(sseProperties.getViolationRecord());
seeMessageContentDto.setContent("您有一份重要的违章需要处理！工单号为：" + violationRecord.getId());
messageDto.setMessage(JSONUtil.toJsonStr(seeMessageContentDto));
messageDto.setProjectId(violationRecord.getProjectId());
SseMessageUtils.publishMessage(messageDto);
}
}