forked from uav/ard-uav-server
ard-sample/src/main/java/com/ard/sample/manage/service/impl/DeviceServiceImpl.java
@@ -1,10 +1,12 @@ package com.ard.sample.manage.service.impl; import com.alibaba.fastjson2.JSONObject; import com.ard.common.annotation.DataScope; import com.ard.common.annotation.DataScopePlus; import com.ard.system.authorize.service.AuthorizeService; import com.baomidou.mybatisplus.core.conditions.Wrapper; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.ard.common.core.domain.entity.SysDept; @@ -34,6 +36,7 @@ import com.fasterxml.jackson.core.type.TypeReference; import com.fasterxml.jackson.databind.JsonNode; import com.fasterxml.jackson.databind.ObjectMapper; import com.fasterxml.jackson.databind.node.ObjectNode; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Qualifier; @@ -348,6 +351,64 @@ // payloads // subDevice.setPayloadsList(payloadService.getDevicePayloadEntitiesByDeviceSn(gateway.getChildDeviceSn())); } //region 单独写一个方法做权限验证，由于有好多定时任务也在实时获取设备列表，原方法不适用再添加权限注解 @Override public List<DeviceDTO> getDockList() { List<DeviceDTO> devicesList = this.getDockListByParams( DeviceQueryParam.builder() .domains(List.of(DeviceDomainEnum.GATEWAY.getVal(), DeviceDomainEnum.DOCK.getVal())) .build()); devicesList.stream() .filter(gateway -> DeviceDomainEnum.DOCK.getVal() == gateway.getDomain() || RedisOpsUtils.checkExist(RedisConst.DEVICE_ONLINE_PREFIX + gateway.getDeviceSn())) .forEach(this::spliceDeviceTopoPer); return devicesList; } @Override public void spliceDeviceTopoPer(DeviceDTO gateway) { gateway.setStatus(RedisOpsUtils.checkExist(RedisConst.DEVICE_ONLINE_PREFIX + gateway.getDeviceSn())); // sub device if (!StringUtils.hasText(gateway.getChildDeviceSn())) { return; } DeviceDTO subDevice = getDevicesByParams(DeviceQueryParam.builder().deviceSn(gateway.getChildDeviceSn()).build()).get(0); subDevice.setStatus(RedisOpsUtils.checkExist(RedisConst.DEVICE_ONLINE_PREFIX + subDevice.getDeviceSn())); gateway.setChildren(subDevice); } @Override public List<DeviceDTO> getDockListByParams(DeviceQueryParam param){ // 在mapper中添加查询方法做权限验证 return mapper.selectListPer( new LambdaQueryWrapper<DeviceEntity>() .eq(StringUtils.hasText(param.getDeviceSn()), DeviceEntity::getDeviceSn, param.getDeviceSn()) .eq(param.getDeviceType() != null, DeviceEntity::getDeviceType, param.getDeviceType()) .eq(param.getSubType() != null, DeviceEntity::getSubType, param.getSubType()) .eq(StringUtils.hasText(param.getChildSn()), DeviceEntity::getChildSn, param.getChildSn()) .and(!CollectionUtils.isEmpty(param.getDomains()), wrapper -> { for (Integer domain : param.getDomains()) { wrapper.eq(DeviceEntity::getDomain, domain).or(); } }) .eq(param.getBoundStatus() != null, DeviceEntity::getBoundStatus, param.getBoundStatus()) .orderBy(param.isOrderBy(), param.isAsc(), DeviceEntity::getId)) .stream() .map(this::deviceEntityConvertToDTO) .collect(Collectors.toList()); } //endregion @Override public Optional<TopologyDeviceDTO> getDeviceTopoForPilot(String sn) { @@ -874,11 +935,17 @@ Long pageSize, Integer domain, Long deptId) { Page<DeviceEntity> pagination = mapper.selectPage(new Page<>(page, pageSize), new LambdaQueryWrapper<DeviceEntity>() .eq(DeviceEntity::getDomain, domain) .eq(deptId != null, DeviceEntity::getDeptId, deptId) .isNotNull(DeviceEntity::getDeptId)); // .eq(DeviceEntity::getBoundStatus, true)); new QueryWrapper<DeviceEntity>() .eq("m.domain", domain) .eq(deptId != null, "m.dept_id", deptId) .isNotNull("m.dept_id")); //以前的写法 // new LambdaQueryWrapper<DeviceEntity>() // .eq(DeviceEntity::getDomain, domain) // .eq(deptId != null, DeviceEntity::getDeptId, deptId) // .isNotNull(DeviceEntity::getDeptId); List<DeviceDTO> devicesList = pagination.getRecords().stream().map(this::deviceEntityConvertToDTO) .peek(device -> { device.setStatus(RedisOpsUtils.checkExist(RedisConst.DEVICE_ONLINE_PREFIX + device.getDeviceSn())); @@ -1006,6 +1073,16 @@ this.deviceOnePropertySet(topic, propertyEnum, Map.entry(propertyEnum.getProperty(), param)); return; } if(param.size()==1){ this.deviceOnePropertySetSilent(topic,propertyEnum,param); return; } // if(Objects.equals(param.fieldNames().next(), "silent_mode")) // { // this.deviceOnePropertySet(topic, propertyEnum, Map.entry(propertyEnum.getProperty(), param.get("silent_mode"))); // return; // } // If there are multiple parameters, set them separately. for (Iterator<Map.Entry<String, JsonNode>> filed = param.fields(); filed.hasNext(); ) { Map.Entry<String, JsonNode> node = filed.next(); @@ -1013,12 +1090,34 @@ if (!isPublish) { continue; } this.deviceOnePropertySet(topic, propertyEnum, Map.entry(propertyEnum.getProperty(), node)); } } /** * 为静音模式单独列出来一个方法，只有一个值的时候都可以调用，不限于静音模式 * */ @Override public void deviceOnePropertySetSilent(String topic, DeviceSetPropertyEnum propertyEnum, JsonNode value) { // if (Objects.isNull(value) || Objects.isNull(value.getValue())) { // throw new IllegalArgumentException(CommonErrorEnum.ILLEGAL_ARGUMENT.getErrorMsg()); // } Map reply = messageSender.publishWithReply( Map.class, topic, CommonTopicResponse.builder() .bid(UUID.randomUUID().toString()) .tid(UUID.randomUUID().toString()) .timestamp(System.currentTimeMillis()) .data(value) .build(), 2); } @Override public void deviceOnePropertySet(String topic, DeviceSetPropertyEnum propertyEnum, Map.Entry<String, Object> value) { if (Objects.isNull(value) || Objects.isNull(value.getValue())) { @@ -1035,20 +1134,29 @@ .build(), 2); while (true) { reply = (Map<String, Object>) reply.get(value.getKey()); if (value.getValue() instanceof JsonNode) { break; } value = (Map.Entry) value.getValue(); } SetReply setReply = objectMapper.convertValue(reply, SetReply.class); if (SetReplyStatusResultEnum.SUCCESS.getVal() != setReply.getResult()) { throw new RuntimeException("Failed to set " + value.getKey() + "; Error Code: " + setReply.getResult()); } // 接收返回信息在serviceReply中，订阅thing/product/7CTDLCE00A1UPE/property/set_reply上报的信息，此处不处理 // while (true) { // reply = (Map<String, Object>) reply.get(value.getKey()); // if (value.getValue() instanceof JsonNode) { // break; // } // value = (Map.Entry) value.getValue(); // } // SetReply setReply = objectMapper.convertValue(reply, SetReply.class); // if (SetReplyStatusResultEnum.SUCCESS.getVal() != setReply.getResult()) { // throw new RuntimeException("Failed to set " + value.getKey() + "; Error Code: " + setReply.getResult()); // } } // public JsonNode getJsonNode(JsonNode jsonNode) { // ObjectNode parentNode = objectMapper.createObjectNode(); // parentNode.put("bid", UUID.randomUUID().toString()); // parentNode.set("data", jsonNode); // parentNode.put("tid", UUID.randomUUID().toString()); // parentNode.put("timestamp", System.currentTimeMillis()); // return parentNode; // } @Override public Boolean checkDeviceOnline(String sn) { @@ -1221,9 +1329,9 @@ messageSender.publish(topic, builder.build()); } @DataScopePlus(deptAlias="d") @Override public List<DeviceEntity> selectList(Wrapper<DeviceEntity> queryWrapper) { return mapper.selectList(queryWrapper); } // @DataScopePlus(deptAlias="d") // @Override // public List<DeviceEntity> selectList(Wrapper<DeviceEntity> queryWrapper) { // return mapper.selectList(queryWrapper); // } }
