From ec95e52c6f84c0a6cbdcbaecc2465b00ede6696d Mon Sep 17 00:00:00 2001 From: jinpengyong <jpy123456> Date: Thu, 24 Jun 2021 14:03:59 +0800 Subject: [PATCH] 数据校准工具类 --- screen-manage/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java | 277 ++++++++++++++++++++++++++++++++++++++++++++++++------ 1 files changed, 243 insertions(+), 34 deletions(-) diff --git a/screen-manage/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java b/screen-manage/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java index c5573f4..fcf29d3 100644 --- a/screen-manage/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java +++ b/screen-manage/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java @@ -1,34 +1,32 @@ package com.moral.api.service.impl; +import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; -import com.moral.api.entity.Device; -import com.moral.api.entity.ManageAccount; -import com.moral.api.entity.MonitorPoint; -import com.moral.api.entity.Organization; -import com.moral.api.entity.SysDictData; -import com.moral.api.entity.SysDictType; -import com.moral.api.entity.Version; -import com.moral.api.mapper.DeviceMapper; -import com.moral.api.mapper.ManageAccountMapper; -import com.moral.api.mapper.MonitorPointMapper; -import com.moral.api.mapper.OrganizationMapper; -import com.moral.api.mapper.SysDictDataMapper; -import com.moral.api.mapper.SysDictTypeMapper; +import com.googlecode.aviator.AviatorEvaluator; +import com.googlecode.aviator.Expression; +import com.moral.api.entity.*; +import com.moral.api.mapper.*; import com.moral.api.pojo.vo.device.DeviceVO; import com.moral.api.service.DeviceService; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.moral.api.util.CacheUtils; +import com.moral.api.util.AdjustDataUtils; import com.moral.api.util.LogUtils; import com.moral.constant.Constants; -import com.moral.redis.RedisUtil; +import com.moral.constant.RedisConstants; import com.moral.util.ConvertUtils; import com.moral.util.DateUtils; +import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.data.redis.core.RedisTemplate; import org.springframework.stereotype.Service; import org.springframework.transaction.annotation.Transactional; +import org.springframework.util.ObjectUtils; +import org.springframework.util.StringUtils; import org.springframework.web.context.request.RequestContextHolder; import org.springframework.web.context.request.ServletRequestAttributes; @@ -52,6 +50,7 @@ * @since 2021-05-11 */ @Service +@Slf4j public class DeviceServiceImpl extends ServiceImpl<DeviceMapper, Device> implements DeviceService { @Autowired @@ -75,6 +74,39 @@ @Autowired private LogUtils logUtils; + @Autowired + private RedisTemplate redisTemplate; + + @Autowired + private OrganizationUnitAlarmMapper organizationUnitAlarmMapper; + + @Autowired + private VersionSensorUnitMapper versionSensorUnitMapper; + + @Autowired + private AdjustDataUtils adjustDataUtils; + + /* + * ���redis������������������ + * */ + private Map<String, Object> getDeviceInfoFromRedis(String mac) { + return (Map<String, Object>) redisTemplate.opsForValue().get(RedisConstants.DEVICE + mac); + } + + /* + * ������������������redis + */ + private void setDeviceInfoToRedis(String mac, Map<String, Object> deviceInfo) { + redisTemplate.opsForValue().set(RedisConstants.DEVICE + mac, deviceInfo); + } + + /* + * ���redis������������������ + */ + private void delDeviceInfoFromRedis(String mac) { + redisTemplate.delete(RedisConstants.DEVICE + mac); + } + @Override @Transactional public void insert(Device device) { @@ -82,25 +114,39 @@ device.setOrganizationId(orgId); deviceMapper.insert(device); Map<String, Object> deviceInfo = selectDeviceInfoById(device.getId()); + //��������������������������� + insertOrganizationUnitAlarm(orgId, device.getDeviceVersionId()); //������������������������redis - RedisUtil.set("device_" + device.getMac(), deviceInfo); + String mac = device.getMac(); + //���redis��������������������� + delDeviceInfoFromRedis(mac); + //������������������redis + setDeviceInfoToRedis(mac, deviceInfo); + //������deviceInfo������ + CacheUtils.flushDeviceAlarmInfo(); //������������������ HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest(); StringBuilder content = new StringBuilder(); - content.append("������������������").append(device.getName()).append("���").append("mac���").append(device.getMac()); + content.append("������������������").append(device.getName()).append("���").append("mac���").append(mac); logUtils.saveOperationForManage(request, content.toString(), Constants.INSERT_OPERATE_TYPE); } @Override @Transactional public void delete(Integer deviceId) { + Device device = deviceMapper.selectById(deviceId); UpdateWrapper<Device> updateWrapper = new UpdateWrapper<>(); updateWrapper.eq("id", deviceId).set("is_delete", Constants.DELETE); deviceMapper.update(null, updateWrapper); - Device device = deviceMapper.selectById(deviceId); String mac = device.getMac(); - //������redis - RedisUtil.del("device_" + mac); + //���redis��������������������� + delDeviceInfoFromRedis(mac); + //��������������������������� + Integer versionId = device.getDeviceVersionId(); + Integer orgId = device.getOrganizationId(); + deleteOrganizationUnitAlarm(orgId, versionId); + //������deviceInfo������ + CacheUtils.flushDeviceAlarmInfo(); //������������������ HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest(); StringBuilder content = new StringBuilder(); @@ -113,12 +159,30 @@ public void update(Device device) { Integer deviceId = device.getId(); Device oldDevice = deviceMapper.selectById(deviceId); + //���������������������������������������������������������������������������������id������������ + if (!ObjectUtils.isEmpty(device.getMonitorPointId())) { + MonitorPoint monitorPoint = monitorPointMapper.selectById(device.getMonitorPointId()); + device.setOrganizationId(monitorPoint.getOrganizationId()); + } deviceMapper.updateById(device); - String mac = deviceMapper.selectById(deviceId).getMac(); - //������redis - RedisUtil.del("device_" + mac); + Device updateDevice = deviceMapper.selectById(deviceId); + String mac = updateDevice.getMac(); + //��������������������������� + Integer oldOrgId = oldDevice.getOrganizationId(); + Integer newOrgId = updateDevice.getOrganizationId(); + Integer oldVersionId = oldDevice.getDeviceVersionId(); + Integer newVersionId = updateDevice.getDeviceVersionId(); + if (!oldOrgId.equals(newOrgId) || !oldVersionId.equals(newVersionId)) { + deleteOrganizationUnitAlarm(oldOrgId, oldVersionId); + insertOrganizationUnitAlarm(newOrgId, newVersionId); + } + //���redis��������������������� + delDeviceInfoFromRedis(mac); Map<String, Object> deviceInfo = selectDeviceInfoById(deviceId); - RedisUtil.set("device_" + mac, deviceInfo); + //������������������redis + setDeviceInfoToRedis(mac, deviceInfo); + //������deviceInfo������ + CacheUtils.flushDeviceAlarmInfo(); //������������������ HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest(); StringBuilder content = new StringBuilder(); @@ -177,17 +241,13 @@ } @Override - public List<Map<String, Object>> selectDevicesByOrgId(Integer orgId) { - QueryWrapper<Device> queryWrapper = new QueryWrapper<>(); - queryWrapper.eq("is_delete", Constants.NOT_DELETE).eq("organization_id", orgId); - return deviceMapper.selectMaps(queryWrapper); + public Map<String, Object> selectDevicesByOrgId(Map<String, Object> parameters) { + return selectDevices(parameters); } @Override - public List<Map<String, Object>> selectDevicesByMpId(Integer mpId) { - QueryWrapper<Device> queryWrapper = new QueryWrapper<>(); - queryWrapper.eq("is_delete", Constants.NOT_DELETE).eq("monitor_point_id", mpId); - return deviceMapper.selectMaps(queryWrapper); + public Map<String, Object> selectDevicesByMpId(Map<String, Object> parameters) { + return selectDevices(parameters); } @Override @@ -199,9 +259,25 @@ Object orderType = parameters.get("orderType"); Object name = parameters.get("name"); Object mac = parameters.get("mac"); + Object orgId = parameters.get("organizationId"); + Object mpId = parameters.get("monitorPointId"); + + //������������������������������ + if (orgId != null) { + queryWrapper.eq("organization_id", orgId); + } + + //������������������������������ + if (mpId != null) { + queryWrapper.eq("monitor_point_id", mpId); + } + + //��������������������� if (name != null) { queryWrapper.like("name", name); } + + //mac������������ if (mac != null) { queryWrapper.like("mac", mac); } @@ -234,11 +310,10 @@ return result; } - @Override public Map<String, Object> selectDeviceInfoById(Integer deviceId) { String mac = deviceMapper.selectById(deviceId).getMac(); - Map<String, Object> deviceInfo = (Map<String, Object>) RedisUtil.get("device_" + mac); + Map<String, Object> deviceInfo = getDeviceInfoFromRedis(mac); //������redis������ if (deviceInfo != null) { return deviceInfo; @@ -254,6 +329,9 @@ deviceInfo.put("latitude", device.getLatitude()); deviceInfo.put("createTime", DateUtils.dateToDateString(device.getCreateTime())); deviceInfo.put("installTime", device.getInstallTime() == null ? null : DateUtils.dateToDateString(device.getInstallTime())); + + //������������ + deviceInfo.put("extend", device.getExtend()); //������ deviceInfo.put("profession", device.getProfession()); @@ -301,8 +379,12 @@ MonitorPoint monitorPoint = device.getMonitorPoint(); mpInfo.put("id", monitorPoint.getId()); mpInfo.put("name", monitorPoint.getName()); + mpInfo.put("areaCode", monitorPoint.getAreaCode()); + mpInfo.put("cityCode", monitorPoint.getCityCode()); + mpInfo.put("provinceCode", monitorPoint.getProvinceCode()); deviceInfo.put("monitorPoint", mpInfo); - RedisUtil.set("device_" + mac, deviceInfo); + + setDeviceInfoToRedis(mac, deviceInfo); return deviceInfo; } @@ -320,4 +402,131 @@ return monitorPointMapper.selectMaps(queryWrapper); } + @Override + public Map<String, Object> getDeviceByMac(String mac) { + Map<String, Object> deviceInfo = getDeviceInfoFromRedis(mac); + if (deviceInfo == null) { + QueryWrapper<Device> queryWrapper = new QueryWrapper<>(); + queryWrapper.eq("mac", mac).eq("is_delete", Constants.NOT_DELETE); + Device device = deviceMapper.selectOne(queryWrapper); + if (device != null) { + deviceInfo = selectDeviceInfoById(device.getId()); + setDeviceInfoToRedis(mac, deviceInfo); + } + } + return deviceInfo; + } + + @Override + public Map<String, Object> adjustDeviceData(Map<String, Object> deviceData) { + String mac = deviceData.get("mac").toString(); + //���redis������������������ + Map<String, Object> adjustFormula = redisTemplate.opsForHash().entries(RedisConstants.ADJUST + mac); + if (!ObjectUtils.isEmpty(adjustFormula)) { + Map<String, Object> deviceInfo = getDeviceByMac(mac); + Map<String, Object> monitorPoint = (Map<String, Object>) deviceInfo.get("monitorPoint"); + Object areaCode = monitorPoint.get("areaCode"); + Object cityCode = monitorPoint.get("cityCode"); + + Map<String, Object> aqiMap = redisTemplate.opsForHash().entries(RedisConstants.AQI_DATA + areaCode); + if (ObjectUtils.isEmpty(aqiMap)) { + aqiMap = redisTemplate.opsForHash().entries(RedisConstants.AQI_DATA + cityCode); + } + return adjustDataUtils.adjust(deviceData, adjustFormula, ObjectUtils.isEmpty(aqiMap) ? null : aqiMap); + } + return deviceData; + } + + @Override + public void judgeDeviceState(Map<String, Object> deviceData) { + String mac = deviceData.remove("mac").toString(); + Device device = (Device) redisTemplate.opsForHash().get(RedisConstants.DEVICE_INFO, mac); + Version version = device.getVersion(); + List<Sensor> sensors = version.getSensors(); + Expression expression; + int state = 1; + for (Sensor sensor : sensors) { + //������������������ + String alarmLevel = sensor.getAlarmLevel(); + if (StringUtils.isEmpty(alarmLevel)) { + continue; + } + List<Double> list = JSONObject.parseObject(alarmLevel, List.class); + String sensorCode = sensor.getCode(); + //������������ + String formula = sensor.getFormula(); + //������������������������ + String sensorValue = (String) deviceData.get(sensorCode); + double value = Double.parseDouble(String.format("%.3f", sensorValue)); + //������������ + if (formula != null) { + //������������������ + sensorValue = formula.replace("{0}", sensorValue); + expression = AviatorEvaluator.compile(sensorValue); + value = Double.parseDouble(String.format("%.3f", expression.execute())); + } + int sensorState = judgeState(list, value); + if (sensorState > state) { + state = sensorState; + } + } + //������������������ + UpdateWrapper<Device> updateWrapper = new UpdateWrapper<>(); + updateWrapper.eq("id", device.getId()).set("state", state); + deviceMapper.update(null, updateWrapper); + } + + //��������������������������� + private int judgeState(List<Double> levels, Double data) { + int state = 1; + for (int i = levels.size() - 1; i >= 0; i--) { + Double level = levels.get(i); + if (data >= level) { + state = i + 2; + break; + } + } + return state; + } + + private void insertOrganizationUnitAlarm(Integer orgId, Integer versionId) { + QueryWrapper<OrganizationUnitAlarm> queryOrganizationVersionWrapper = new QueryWrapper<>(); + queryOrganizationVersionWrapper.eq("organization_id", orgId); + queryOrganizationVersionWrapper.eq("version_id", versionId); + queryOrganizationVersionWrapper.eq("is_delete", Constants.NOT_DELETE); + List<OrganizationUnitAlarm> organizationUnitAlarms = organizationUnitAlarmMapper.selectList(queryOrganizationVersionWrapper); + if (ObjectUtils.isEmpty(organizationUnitAlarms)) { + QueryWrapper<VersionSensorUnit> queryVersionSensorUnitWrapper = new QueryWrapper<>(); + queryVersionSensorUnitWrapper.eq("version_id", versionId); + queryVersionSensorUnitWrapper.eq("is_delete", Constants.NOT_DELETE); + List<VersionSensorUnit> versionSensorUnits = versionSensorUnitMapper.selectList(queryVersionSensorUnitWrapper); + if (!ObjectUtils.isEmpty(versionSensorUnits)) { + for (VersionSensorUnit versionSensorUnit : versionSensorUnits) { + OrganizationUnitAlarm organizationUnitAlarm = new OrganizationUnitAlarm(); + organizationUnitAlarm.setOrganizationId(orgId); + organizationUnitAlarm.setVersionId(versionId); + organizationUnitAlarm.setSensorCode(versionSensorUnit.getSensorCode()); + organizationUnitAlarm.setUnitKey(versionSensorUnit.getUnitKey()); + organizationUnitAlarm.setShowUnitKey(versionSensorUnit.getUnitKey()); + organizationUnitAlarmMapper.insert(organizationUnitAlarm); + } + } + } + } + + private void deleteOrganizationUnitAlarm(Integer orgId, Integer versionId) { + QueryWrapper<Device> queryOrganizationVersionWrapper = new QueryWrapper<>(); + queryOrganizationVersionWrapper.eq("organization_id", orgId); + queryOrganizationVersionWrapper.eq("device_version_id", versionId); + queryOrganizationVersionWrapper.eq("is_delete", Constants.NOT_DELETE); + List<Device> devices = deviceMapper.selectList(queryOrganizationVersionWrapper); + if (ObjectUtils.isEmpty(devices)) {//������������������������������������������������������ + UpdateWrapper deleteWrapper = new UpdateWrapper(); + deleteWrapper.eq("organization_id", orgId); + deleteWrapper.eq("version_id", versionId); + deleteWrapper.eq("is_delete", Constants.NOT_DELETE); + deleteWrapper.set("is_delete", Constants.DELETE); + organizationUnitAlarmMapper.update(null, deleteWrapper); + } + } } -- Gitblit v1.8.0