zhangnaisong
2023-10-06 f8ca81e96bcb9a9a05531d606f1dd67953c7a976
ard-work/src/main/java/com/ruoyi/statistical/service/impl/StatisticalServiceImpl.java
@@ -1,5 +1,6 @@
package com.ruoyi.statistical.service.impl;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.ruoyi.alarmpoints.well.domain.ArdAlarmpointsWell;
import com.ruoyi.alarmpoints.well.mapper.ArdAlarmpointsWellMapper;
@@ -11,6 +12,7 @@
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.List;
import java.util.Map;
/**
@@ -69,26 +71,24 @@
                }
            }
        }
        //记录well表中oil_production为抽油机的数据有多少条
        QueryWrapper<ArdAlarmpointsWell> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("oil_production","抽油机");
        int num = wellMapper.selectCount(queryWrapper);
        SysConfig sysConfigUpd = sysConfigMapper.checkConfigKeyUnique("well");
        if(sysConfigUpd!=null){
            sysConfigUpd.setConfigValue(String.valueOf(num));
            sysConfigUpd.setUpdateBy(SecurityUtils.getUsername());
            sysConfigMapper.updateConfig(sysConfigUpd);
        }else {
            SysConfig sysConfig = new SysConfig();
            int randomNum = (int) (Math.random() * (1000 - 1 + 1)) + 1;
            sysConfig.setConfigId((long) randomNum);
            sysConfig.setConfigType("Y");
            sysConfig.setConfigName("抽水井");
            sysConfig.setConfigKey("well");
            sysConfig.setConfigValue(String.valueOf(num));
            sysConfig.setCreateBy(SecurityUtils.getUsername());
            sysConfigMapper.insertConfig(sysConfig);
        }
        return Results.succeed("添加成功!");
    }
    @Override
    public Results factorySelect() {
        JSONObject jsonObject = new JSONObject();
        QueryWrapper<SysConfig> queryWrapper = new QueryWrapper<>();
        queryWrapper.in("config_key","area","store","people","well","depot","unit");
        List<SysConfig> list = sysConfigMapper.selectList(queryWrapper);
        for (int i = 0; i < list.size(); i++) {
            SysConfig sysConfig = list.get(i);
            jsonObject.put(sysConfig.getConfigKey(),sysConfig.getConfigValue());
        }
        //记录well表中oil_production为抽油机的数据有多少条
        QueryWrapper<ArdAlarmpointsWell> queryWrapperWell = new QueryWrapper<>();
        queryWrapperWell.eq("oil_production","抽油机");
        int num = wellMapper.selectCount(queryWrapperWell);
        jsonObject.put("well",String.valueOf(num));
        return Results.succeed(jsonObject);
    }
}