From 23e43d94105c6f3676e879d2cc271d0bec63f081 Mon Sep 17 00:00:00 2001 From: kaiyu <404897439@qq.com> Date: Tue, 31 Aug 2021 13:13:41 +0800 Subject: [PATCH] screen-manage kafka消费者更换主题名称 --- screen-manage/src/main/java/com/moral/api/service/impl/DeviceServiceImpl.java | 214 +++++++++++++++++++++++++++++++++++++---------------- 1 files changed, 150 insertions(+), 64 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 9e06f36..9339dc2 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,26 +1,32 @@ package com.moral.api.service.impl; -import com.alibaba.fastjson.JSON; +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.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.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; @@ -44,6 +50,7 @@ * @since 2021-05-11 */ @Service +@Slf4j public class DeviceServiceImpl extends ServiceImpl<DeviceMapper, Device> implements DeviceService { @Autowired @@ -76,45 +83,31 @@ @Autowired private VersionSensorUnitMapper versionSensorUnitMapper; + @Autowired + private SpecialDeviceMapper specialDeviceMapper; + @Autowired + private AdjustDataUtils adjustDataUtils; /* * ���redis������������������ * */ private Map<String, Object> getDeviceInfoFromRedis(String mac) { - Map<String, Object> deviceInfo = (Map<String, Object>) redisTemplate.opsForValue().get(RedisConstants.DEVICE + mac); - return deviceInfo; + return (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.DEVICE, mac); } /* * ������������������redis */ private void setDeviceInfoToRedis(String mac, Map<String, Object> deviceInfo) { - redisTemplate.opsForValue().set(getDeviceKey(mac), deviceInfo); + redisTemplate.opsForHash().put(RedisConstants.DEVICE, mac, deviceInfo); } /* * ���redis������������������ */ private void delDeviceInfoFromRedis(String mac) { - redisTemplate.delete(getDeviceKey(mac)); - } - - /* - * ���������������������redis������key - */ - private String getDeviceKey(String mac) { - return keysConnect(RedisConstants.DEVICE, mac); - } - - //redis key������ - private String keysConnect(String... keys) { - StringBuilder key = new StringBuilder(keys[0]); - for (int i = 1; i < keys.length; i++) { - key.append("_"); - key.append(keys[i]); - } - return key.toString().toLowerCase(); + redisTemplate.opsForHash().delete(RedisConstants.DEVICE, mac); } @Override @@ -125,34 +118,38 @@ deviceMapper.insert(device); Map<String, Object> deviceInfo = selectDeviceInfoById(device.getId()); //��������������������������� - insertOrganizationUnitAlarm(orgId,device.getDeviceVersionId()); + insertOrganizationUnitAlarm(orgId, device.getDeviceVersionId()); //������������������������redis String mac = device.getMac(); //���redis��������������������� delDeviceInfoFromRedis(mac); //������������������redis setDeviceInfoToRedis(mac, deviceInfo); + //������deviceInfo������ + CacheUtils.refreshDeviceAlarmInfo(); //������������������ HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest(); StringBuilder content = new StringBuilder(); - content.append("������������������").append(device.getName()).append("���").append("mac���").append(mac); + 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��������������������� delDeviceInfoFromRedis(mac); //��������������������������� Integer versionId = device.getDeviceVersionId(); Integer orgId = device.getOrganizationId(); - deleteOrganizationUnitAlarm(orgId,versionId); + deleteOrganizationUnitAlarm(orgId, versionId); + //������deviceInfo������ + CacheUtils.refreshDeviceAlarmInfo(); //������������������ HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest(); StringBuilder content = new StringBuilder(); @@ -166,7 +163,7 @@ Integer deviceId = device.getId(); Device oldDevice = deviceMapper.selectById(deviceId); //���������������������������������������������������������������������������������id������������ - if(!ObjectUtils.isEmpty(device.getMonitorPointId())){ + if (!ObjectUtils.isEmpty(device.getMonitorPointId())) { MonitorPoint monitorPoint = monitorPointMapper.selectById(device.getMonitorPointId()); device.setOrganizationId(monitorPoint.getOrganizationId()); } @@ -174,23 +171,25 @@ 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); - } + 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); + delDeviceInfoFromRedis(oldDevice.getMac()); Map<String, Object> deviceInfo = selectDeviceInfoById(deviceId); //������������������redis setDeviceInfoToRedis(mac, deviceInfo); + //������deviceInfo������ + CacheUtils.refreshDeviceAlarmInfo(); //������������������ HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest(); StringBuilder content = new StringBuilder(); - content.append("������������������").append(mac).append("���"); + content.append("������������������" + oldDevice.getMac()).append("==>").append(mac).append("���"); Field[] fields = Device.class.getDeclaredFields(); for (Field field : fields) { if (field.getName().equals("id")) { @@ -207,7 +206,7 @@ Object o1 = method.invoke(oldDevice); Object o2 = method.invoke(device); if (o2 != null) { - content.append(fieldName).append("���").append(o1).append("-->").append(o2).append("���"); + content.append(fieldName).append("���").append(o1).append("-->").append(o2).append(":"); } } catch (Exception e) { @@ -276,13 +275,15 @@ queryWrapper.eq("monitor_point_id", mpId); } + //��������������������� if (name != null) { queryWrapper.like("name", name); } + + //mac������������ if (mac != null) { queryWrapper.like("mac", mac); } - //������������,������create_time������ if (order != null && orderType != null) { @@ -311,7 +312,6 @@ result.put("item", items); return result; } - @Override public Map<String, Object> selectDeviceInfoById(Integer deviceId) { @@ -382,6 +382,9 @@ 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); setDeviceInfoToRedis(mac, deviceInfo); @@ -405,7 +408,7 @@ @Override public Map<String, Object> getDeviceByMac(String mac) { Map<String, Object> deviceInfo = getDeviceInfoFromRedis(mac); - if (deviceInfo == null) { + if (ObjectUtils.isEmpty(deviceInfo)) { QueryWrapper<Device> queryWrapper = new QueryWrapper<>(); queryWrapper.eq("mac", mac).eq("is_delete", Constants.NOT_DELETE); Device device = deviceMapper.selectOne(queryWrapper); @@ -418,27 +421,110 @@ } @Override - public Map<String, Object> adjustDeviceData(Map<String, Object> deviceData, Map<String, Object> deviceInfo) { - return null; + public Map<String, Object> adjustDeviceData(Map<String, Object> deviceData) { + String mac = deviceData.remove("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 = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.AQI_DATA, areaCode); + if (ObjectUtils.isEmpty(aqiMap)) { + aqiMap = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.AQI_DATA, cityCode); + } + return adjustDataUtils.adjust(deviceData, adjustFormula, ObjectUtils.isEmpty(aqiMap) ? null : aqiMap); + } + deviceData.remove("DataTime"); + return deviceData; } @Override - public Map<String, Object> judgeDeviceState(Map<String, Object> deviceData, Map<String, Object> deviceInfo) { - return null; + public Map<String, Object> adjustSpecialDeviceData(Map<String, Object> deviceData) { + String mac = deviceData.remove("mac").toString(); + //���redis������������������ + Map<String, Object> adjustFormula = redisTemplate.opsForHash().entries(RedisConstants.ADJUST + "_" + mac); + if (!ObjectUtils.isEmpty(adjustFormula)) { + Organization organization = specialDeviceMapper.selectOrgByMac(mac); + Integer areaCode = organization.getAreaCode(); + Integer cityCode = organization.getCityCode(); + Map<String, Object> aqiMap = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.AQI_DATA, areaCode); + if (ObjectUtils.isEmpty(aqiMap)) { + aqiMap = (Map<String, Object>) redisTemplate.opsForHash().get(RedisConstants.AQI_DATA, cityCode); + } + return adjustDataUtils.adjust(deviceData, adjustFormula, ObjectUtils.isEmpty(aqiMap) ? null : aqiMap); + } + deviceData.remove("DataTime"); + return deviceData; } - private void insertOrganizationUnitAlarm(Integer orgId,Integer versionId){ + @Override + @Transactional + 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<Object> list = JSONObject.parseObject(alarmLevel, List.class); + String sensorCode = sensor.getCode(); + //������������ + String formula = sensor.getFormula(); + //������������������������ + String sensorValue = (String) deviceData.get(sensorCode); + double value = Double.parseDouble(sensorValue); + //������������ + if (formula != null) { + //������������������ + sensorValue = formula.replace("{0}", sensorValue); + expression = AviatorEvaluator.compile(sensorValue); + value = Double.parseDouble(expression.execute().toString()); + } + int sensorState = judgeState(list, value); + if (sensorState > state) { + state = sensorState; + } + } + //������������������ + UpdateWrapper<Device> updateWrapper = new UpdateWrapper<>(); + updateWrapper.eq("id", device.getId()).set("state", String.valueOf(state)); + deviceMapper.update(null, updateWrapper); + } + + //��������������������������� + private int judgeState(List<Object> levels, Double data) { + int state = 1; + for (int i = levels.size() - 1; i >= 0; i--) { + Double level = Double.parseDouble(levels.get(i).toString()); + if (data >= level) { + state = i + 1; + 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); + 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); + 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)){ + if (!ObjectUtils.isEmpty(versionSensorUnits)) { for (VersionSensorUnit versionSensorUnit : versionSensorUnits) { OrganizationUnitAlarm organizationUnitAlarm = new OrganizationUnitAlarm(); organizationUnitAlarm.setOrganizationId(orgId); @@ -452,19 +538,19 @@ } } - private void deleteOrganizationUnitAlarm(Integer orgId,Integer versionId){ + 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); + 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)){//������������������������������������������������������ + 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); + 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