package com.fzzy.igds.service;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.fzzy.igds.constant.OrderRespEnum;
|
import com.fzzy.igds.constant.RedisConst;
|
import com.fzzy.igds.data.GrainData;
|
import com.fzzy.igds.data.GrainParam;
|
import com.fzzy.igds.domain.Depot;
|
import com.fzzy.igds.domain.Grain;
|
import com.fzzy.igds.mapper.GrainMapper;
|
import com.fzzy.igds.utils.ContextUtil;
|
import com.fzzy.igds.utils.DateUtil;
|
import com.fzzy.igds.websocket.WebSocketPacket;
|
import com.fzzy.igds.websocket.WebSocketServer;
|
import com.ruoyi.common.core.redis.RedisCache;
|
import com.ruoyi.common.utils.StringUtils;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.time.DateFormatUtils;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.stereotype.Service;
|
import javax.annotation.Resource;
|
import java.util.*;
|
|
/**
|
* @Description
|
* @Author CZT
|
* @Date 2025/12/9 10:00
|
*/
|
@Slf4j
|
@Service
|
public class GrainService {
|
|
@Resource
|
private GrainMapper grainMapper;
|
@Resource
|
private RedisCache redisCache;
|
@Resource
|
private DepotService depotService;
|
|
/**
|
* 查询数据
|
* @param param
|
*/
|
public List<Grain> listData(GrainParam param) {
|
if(StringUtils.isBlank(param.getCompanyId())){
|
param.setCompanyId(ContextUtil.getCompanyId());
|
}
|
if(StringUtils.isBlank(param.getDeptId())){
|
param.setDeptId(ContextUtil.subDeptId(null));
|
}
|
|
QueryWrapper<Grain> queryWrapper = new QueryWrapper<>();
|
if (StringUtils.isNotBlank(param.getCompanyId())) {
|
queryWrapper.eq("company_id", param.getCompanyId());
|
}
|
if (StringUtils.isNotBlank(param.getDeptId())) {
|
queryWrapper.eq("dept_id", param.getDeptId());
|
}
|
if (StringUtils.isNotBlank(param.getDepotId())) {
|
queryWrapper.eq("depot_id", param.getDepotId());
|
}
|
if (StringUtils.isNotBlank(param.getBatchId())) {
|
queryWrapper.eq("batch_id", param.getBatchId());
|
}
|
if (null != param.getStart()) {
|
queryWrapper.ge("receive_date", DateUtil.getCurZero(param.getStart()));
|
}
|
if (null != param.getEnd()) {
|
queryWrapper.le("receive_date", DateUtil.getNextZero(param.getEnd()));
|
}
|
queryWrapper.orderByDesc("batch_id");
|
return grainMapper.selectList(queryWrapper);
|
}
|
|
/**
|
* 分页查询数据
|
* @param page
|
* @param param
|
*/
|
public void pageData(Page<Grain> page, Map<String, Object> param) {
|
|
QueryWrapper<Grain> queryWrapper = new QueryWrapper<>();
|
|
queryWrapper.eq("company_id", ContextUtil.getCompanyId());
|
queryWrapper.eq("dept_id", ContextUtil.subDeptId(null));
|
|
String ids = (String) param.get("ids");
|
if (StringUtils.isNotBlank(ids)) {
|
queryWrapper.in("depot_id", ids);
|
}
|
Date date = (Date) param.get(("start"));
|
if (null != date) {
|
queryWrapper.ge("receive_date", DateUtil.getCurZero(date));
|
}
|
date = (Date) param.get(("end"));
|
if (null != date) {
|
queryWrapper.le("receive_date", DateUtil.getNextZero(date));
|
}
|
queryWrapper.orderByDesc("batch_id");
|
grainMapper.selectPage(page, queryWrapper);
|
}
|
|
/**
|
* JPA - 更新数据
|
* @param data
|
*/
|
public void saveOrUpdateGrain(Grain data, String batchTag) {
|
if (StringUtils.isEmpty(data.getCompanyId())) {
|
data.setCompanyId(ContextUtil.getCompanyId());
|
}
|
if (StringUtils.isEmpty(data.getBatchId())) {
|
data.setBatchId(DateFormatUtils.format(data.getReceiveDate(), "yyyyMMddHHmm"));
|
}
|
if(StringUtils.isEmpty(batchTag)){
|
//自动生成
|
batchTag = "01";
|
}
|
|
if(StringUtils.isBlank(data.getCreateBy())){
|
data.setCreateBy(ContextUtil.getLoginUserName());
|
data.setCreateTime(new Date());
|
}
|
|
data.setUpdateBy(ContextUtil.getLoginUserName());
|
data.setUpdateTime(new Date());
|
|
data.setBatchTag(batchTag);
|
int insert = grainMapper.updateById(data);
|
if(insert == 0){
|
grainMapper.insert(data);
|
}
|
|
updateCacheGrainData(data);
|
}
|
|
|
/**
|
* 更新缓存中最新的粮情检测数据
|
*
|
* @param data 当前数据为标准的Grain数据，没有做更新点位信息
|
* @return key=参考编码
|
*/
|
public void updateCacheGrainData(Grain data) {
|
|
GrainData grainData = new GrainData();
|
BeanUtils.copyProperties(data, grainData);
|
|
String key = RedisConst.buildKey(data.getCompanyId(),
|
RedisConst.KEY_GRAIN, grainData.getDepotId());
|
redisCache.setCacheObject(key, grainData);
|
|
|
//推送大屏
|
String deptId = data.getDeptId();
|
if (null == deptId) {
|
Depot depot = depotService.getCacheDepot(data.getCompanyId(), data.getDepotId());
|
if (null == depot) {
|
return;
|
}
|
deptId = depot.getDeptId();
|
}
|
Map<String, GrainData> mapData = this.getCacheGrainDateMap(data.getCompanyId(), deptId);
|
WebSocketPacket packet = new WebSocketPacket();
|
packet.setBizType("screen");
|
packet.setCompanyId(ContextUtil.getCompanyId());
|
packet.setDeptId(deptId);
|
packet.setBizId("grain");
|
packet.setOrderResp(OrderRespEnum.ORDER_INPROGRESS.getCode());
|
packet.setData(mapData);
|
WebSocketServer.sendByPocket(packet);
|
}
|
|
/**
|
* 获取缓存中最新的粮情检测数据
|
*
|
* @param companyId
|
* @return key=参考编码
|
*/
|
public Map<String, GrainData> getCacheGrainDateMap(String companyId, String deptId) {
|
if (null == companyId || null == deptId) return null;
|
|
String pattern = RedisConst.buildKey(companyId, RedisConst.KEY_GRAIN) + "*";
|
|
Collection<String> keys = redisCache.keys(pattern);
|
|
if (null == keys || keys.isEmpty()) return null;
|
|
Map<String, GrainData> result = new HashMap<>();
|
GrainData data;
|
for (String key : keys) {
|
data = (GrainData) redisCache.getCacheObject(key);
|
if (null == data) continue;
|
|
result.put(data.getDepotId(), data);
|
}
|
return result;
|
}
|
|
}
|
