From d673c2d0cbeeeb3ab788607ecef3d3b953054d2d Mon Sep 17 00:00:00 2001 From: kaiyu <404897439@qq.com> Date: Wed, 23 Jun 2021 14:48:10 +0800 Subject: [PATCH] screen-manage 修复update语句bug --- screen-manage/src/main/java/com/moral/api/service/impl/VersionServiceImpl.java | 163 ++++++++++++++++++++++++++++++++++++++++++++++++------ 1 files changed, 144 insertions(+), 19 deletions(-) diff --git a/screen-manage/src/main/java/com/moral/api/service/impl/VersionServiceImpl.java b/screen-manage/src/main/java/com/moral/api/service/impl/VersionServiceImpl.java index 4e79d21..92e0bf6 100644 --- a/screen-manage/src/main/java/com/moral/api/service/impl/VersionServiceImpl.java +++ b/screen-manage/src/main/java/com/moral/api/service/impl/VersionServiceImpl.java @@ -4,9 +4,9 @@ import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.moral.api.config.mybatis.wrapper.NullFilterWrapper; -import com.moral.api.entity.User; -import com.moral.api.entity.Version; -import com.moral.api.entity.VersionSensorUnit; +import com.moral.api.entity.*; +import com.moral.api.mapper.OrganizationUnitAlarmMapper; +import com.moral.api.mapper.SensorMapper; import com.moral.api.mapper.VersionMapper; import com.moral.api.mapper.VersionSensorUnitMapper; import com.moral.api.pojo.dto.version.VersionDTO; @@ -40,6 +40,10 @@ VersionMapper versionMapper; @Autowired VersionSensorUnitMapper versionSensorUnitMapper; + @Autowired + SensorMapper sensorMapper; + @Autowired + OrganizationUnitAlarmMapper organizationUnitAlarmMapper; @Override public VersionQueryDTO query(VersionQueryForm form) { @@ -141,7 +145,7 @@ QueryWrapper<Version> wrapper = new QueryWrapper<>(); wrapper.setEntity(existVersion); Version existVersionResult = versionMapper.selectOne(wrapper); - if(!ObjectUtils.isEmpty(existVersionResult)){ + if (!ObjectUtils.isEmpty(existVersionResult)) { dto.setCode(ResponseCodeEnum.VERSION_EXIST.getCode()); dto.setMsg(ResponseCodeEnum.VERSION_EXIST.getMsg()); return dto; @@ -161,6 +165,16 @@ VersionDTO dto = new VersionDTO(); //������ Integer id = form.getId(); + //������������������������������������������������������������ + QueryWrapper<OrganizationUnitAlarm> queryWrapper = new QueryWrapper<>(); + queryWrapper.eq("is_delete",Constants.NOT_DELETE); + queryWrapper.eq("version_id",id); + List<OrganizationUnitAlarm> organizationUnitAlarms = organizationUnitAlarmMapper.selectList(queryWrapper); + if(!ObjectUtils.isEmpty(organizationUnitAlarms)){ + dto.setCode(ResponseCodeEnum.VERSION_USED.getCode()); + dto.setMsg(ResponseCodeEnum.VERSION_USED.getMsg()); + return dto; + } //������������������������������������������ Version oldVersion = versionMapper.selectById(id); //������������ @@ -170,9 +184,9 @@ versionMapper.updateById(deleteVersion); //������������������������������������ UpdateWrapper deleteSensorUnitWrapper = new UpdateWrapper(); - deleteSensorUnitWrapper.eq("version_id",id); - deleteSensorUnitWrapper.set("is_delete",Constants.DELETE); - versionSensorUnitMapper.update(null,deleteSensorUnitWrapper); + deleteSensorUnitWrapper.eq("version_id", id); + deleteSensorUnitWrapper.set("is_delete", Constants.DELETE); + versionSensorUnitMapper.update(null, deleteSensorUnitWrapper); //������������������ dto.setCode(ResponseCodeEnum.SUCCESS.getCode()); dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg()); @@ -187,15 +201,67 @@ //������ List<VersionSensorUnit> sensorUnits = form.getSensorUnits(); Integer versionId = form.getVersionId(); - //������������������������������������ - UpdateWrapper deleteWrapper = new UpdateWrapper(); - deleteWrapper.eq("version_id",versionId); - deleteWrapper.set("is_delete",Constants.DELETE); - versionSensorUnitMapper.update(null,deleteWrapper); - //��������������������������������� + //������������������id������code for (VersionSensorUnit sensorUnit : sensorUnits) { - sensorUnit.setVersionId(versionId); + Sensor sensor = sensorMapper.selectById(sensorUnit.getSensorId()); + sensorUnit.setSensorCode(sensor.getCode()); + } + //������������������������������������ + QueryWrapper<VersionSensorUnit> queryOldWrapper = new QueryWrapper<>(); + queryOldWrapper.eq("version_id",versionId); + queryOldWrapper.eq("is_delete",Constants.NOT_DELETE); + List<VersionSensorUnit> oldSensorUnits = versionSensorUnitMapper.selectList(queryOldWrapper); + //������������������������(������������������������) + List<VersionSensorUnit> insertList = new ArrayList<>(); + List<VersionSensorUnit> deleteList = new ArrayList<>(); + sensorUnits.forEach(value->{ + value.setVersionId(versionId); + if(!oldSensorUnits.contains(value)) + insertList.add(value); + }); + oldSensorUnits.forEach(value->{ + if(!sensorUnits.contains(value)) + deleteList.add(value); + }); + //������������ + UpdateWrapper deleteWrapper = new UpdateWrapper(); + List<Integer> deleteIds = new ArrayList<>(); + deleteList.forEach(value->deleteIds.add(value.getId())); + deleteWrapper.in("id",deleteIds); + deleteWrapper.set("is_delete", Constants.DELETE); + versionSensorUnitMapper.update(null, deleteWrapper); + //������������ + for (VersionSensorUnit sensorUnit : insertList) { versionSensorUnitMapper.insert(sensorUnit); + } + //��������������������������� + QueryWrapper<OrganizationUnitAlarm> queryOrgUnitAlarmWrapper = new QueryWrapper<>(); + queryOrgUnitAlarmWrapper.select("distinct organization_id"); + queryOrgUnitAlarmWrapper.eq("version_id",versionId); + queryOrgUnitAlarmWrapper.eq("is_delete",Constants.NOT_DELETE); + List<OrganizationUnitAlarm> organizationUnitAlarms = organizationUnitAlarmMapper.selectList(queryOrgUnitAlarmWrapper); + List<Integer> organizationIds = new ArrayList<>(); + organizationUnitAlarms.forEach(value->organizationIds.add(value.getOrganizationId())); + for (Integer organizationId : organizationIds) { + //������������ + for (VersionSensorUnit versionSensorUnit : deleteList) { + UpdateWrapper deleteOrganizationUnitAlarmWrapper = new UpdateWrapper(); + deleteOrganizationUnitAlarmWrapper.eq("organization_id",organizationId); + deleteOrganizationUnitAlarmWrapper.eq("version_id",versionId); + deleteOrganizationUnitAlarmWrapper.eq("sensor_code",versionSensorUnit.getSensorCode()); + deleteOrganizationUnitAlarmWrapper.set("is_delete",Constants.DELETE); + organizationUnitAlarmMapper.update(null,deleteOrganizationUnitAlarmWrapper); + } + //������������ + for (VersionSensorUnit versionSensorUnit : insertList) { + OrganizationUnitAlarm organizationUnitAlarm = new OrganizationUnitAlarm(); + organizationUnitAlarm.setOrganizationId(organizationId); + organizationUnitAlarm.setVersionId(versionId); + organizationUnitAlarm.setSensorCode(versionSensorUnit.getSensorCode()); + organizationUnitAlarm.setUnitKey(versionSensorUnit.getUnitKey()); + organizationUnitAlarm.setShowUnitKey(versionSensorUnit.getUnitKey()); + organizationUnitAlarmMapper.insert(organizationUnitAlarm); + } } //������������������ dto.setCode(ResponseCodeEnum.SUCCESS.getCode()); @@ -204,18 +270,77 @@ } @Override - public VersionDTO queryVersionById(Integer id) { + public VersionQueryDTO queryVersionById(VersionQueryByIdForm form) { //������������������ - VersionDTO dto = new VersionDTO(); + VersionQueryDTO dto = new VersionQueryDTO(); + //������ + Integer id = form.getId(); + Integer size = form.getSize(); + Integer page = form.getPage(); //������������������������������������ List<Integer> versionsIds = new ArrayList<>(); versionsIds.add(id); - List<Version> version = new ArrayList<>(); - version = versionMapper.queryVersionsAndSensorUnitByIds(versionsIds); + List<Version> versions = new ArrayList<>(); + versions = versionMapper.queryVersionsAndSensorUnitByIds(versionsIds); + //������������������������������������ + Version version = versions.get(0); + List<Sensor> querySensors = version.getSensors(); + if (!ObjectUtils.isEmpty(querySensors)) { + int startIndex = (page - 1) * size; + int endIndex = startIndex + size - 1; + List<Sensor> resultSensors = new ArrayList<>(); + + for (int i = startIndex; i <= endIndex; i++) { + if (i >= querySensors.size()) + break; + resultSensors.add(querySensors.get(i)); + } + version.setSensors(resultSensors); + } //������������������ dto.setCode(ResponseCodeEnum.SUCCESS.getCode()); dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg()); - dto.setVersion(version.get(0)); + if (ObjectUtils.isEmpty(querySensors)) { + dto.setTotal(0); + dto.setPages(0); + }else{ + dto.setTotal(querySensors.size()); + double querySize = (double)querySensors.size(); + double dSize = (double)size; + dto.setPages((int)Math.ceil(querySize/dSize)); + } + dto.setSize(size); + dto.setCurrent(page); + VersionDTO versionDTO = new VersionDTO(); + versionDTO.setVersion(version); + dto.setVersionDTOS(Arrays.asList(versionDTO)); return dto; } + + @Override + public VersionQueryDTO queryByOrganizationId(Integer organizationId) { + //������������������ + VersionQueryDTO dto = new VersionQueryDTO(); + //������������id + QueryWrapper<OrganizationUnitAlarm> queryVersionIdsWrapper = new QueryWrapper<>(); + queryVersionIdsWrapper.select("DISTINCT version_id").eq("is_delete",Constants.NOT_DELETE).eq("organization_id",organizationId); + List<OrganizationUnitAlarm> organizationUnitAlarms = organizationUnitAlarmMapper.selectList(queryVersionIdsWrapper); + List<Integer> versionIds = new ArrayList<>(); + for (OrganizationUnitAlarm organizationUnitAlarm : organizationUnitAlarms) { + versionIds.add(organizationUnitAlarm.getVersionId()); + } + //������������id������������ + List<Version> versions = new ArrayList<>(); + if(!ObjectUtils.isEmpty(versionIds)) + versions = versionMapper.selectBatchIds(versionIds); + //������������������ + List<VersionDTO> versionDTOS = new ArrayList<>(); + versions.forEach(value->versionDTOS.add(new VersionDTO(value))); + dto.setCode(ResponseCodeEnum.SUCCESS.getCode()); + dto.setMsg(ResponseCodeEnum.SUCCESS.getMsg()); + dto.setVersionDTOS(versionDTOS); + return dto; + } + + } -- Gitblit v1.8.0