质押合同页面重构

sgj

2026-03-20 58f63e3a124680d0b1f571f5d0a1e3ed308d43be

 fzzy-igdss-core/src/main/java/com/fzzy/igds/service/GrainService.java

@@ -2,21 +2,25 @@

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.data.InoutParam;
import com.fzzy.igds.domain.Depot;
import com.fzzy.igds.domain.Grain;
import com.fzzy.igds.domain.InoutRecord;
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.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.*;

/**
 * @Description
@@ -29,6 +33,10 @@

    @Resource
    private GrainMapper grainMapper;
    @Resource
    private RedisCache redisCache;
    @Resource
    private DepotService depotService;

    /**
     * 查询数据
@@ -93,4 +101,100 @@
        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;
    }

}