package org.jeecg.modules.iot.controller;
|
|
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.extern.slf4j.Slf4j;
|
import org.jeecg.common.api.vo.Result;
|
import org.jeecg.common.aspect.annotation.AutoLog;
|
import org.jeecg.common.system.base.controller.JeecgController;
|
import org.jeecg.modules.iot.entity.*;
|
import org.jeecg.modules.iot.service.IEquipmentService;
|
import org.jeecg.modules.iot.service.IParameterGroupService;
|
import org.jeecg.modules.iot.service.IProjectClassifyService;
|
import org.jeecg.modules.iot.service.IServerDeployService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
|
import java.util.*;
|
|
|
/**
|
* @Description: 项目分类
|
* @Author: cuikaidong
|
* @Date: 2024-12-20
|
* @Version: V1.0
|
*/
|
@Api(tags = "项目分类")
|
@RestController
|
@RequestMapping("/projectClassify")
|
@Slf4j
|
public class ProjectClassifyController extends JeecgController<ProjectClassify, IProjectClassifyService> {
|
@Autowired
|
private IProjectClassifyService projectClassifyService;
|
@Autowired
|
private IServerDeployService serverDeployService;
|
@Autowired
|
private IEquipmentService equipmentService;
|
@Autowired
|
private IParameterGroupService parameterGroupService;
|
|
/**
|
* 查询项目分类树结构
|
*
|
* @return projectClassifyList
|
*/
|
@AutoLog(value = "查询项目分类树结构")
|
@ApiOperation(value = "查询项目分类树结构", notes = "查询项目分类树结构")
|
@GetMapping(value = "/tree")
|
public Result<?> queryPageList() {
|
// 树图标
|
Map<String, String> iconIot = new HashMap<>(); // iot图标
|
iconIot.put("icon", "iot");
|
Map<String, String> iconProject = new HashMap<>(); //
|
iconProject.put("icon", "classify");
|
Map<String, String> iconProject1 = new HashMap<>(); // 服务器图标
|
iconProject1.put("icon", "project");
|
Map<String, String> iconEmpty = new HashMap<>();
|
iconEmpty.put("icon", "empty");
|
Map<String, String> iconSolid = new HashMap<>();
|
iconSolid.put("icon", "solid");
|
Map<String, String> iconParameter = new HashMap<>();
|
iconParameter.put("icon", "");
|
// 查询数据
|
List<ProjectClassify> projectList = projectClassifyService.list();
|
List<ServerDeploy> serverDeployList = serverDeployService.list();
|
List<Equipment> equipmentList = equipmentService.list();
|
List<ParameterGroup> groupList = parameterGroupService.list();
|
// 树结构
|
List<IotTree> trees1 = new ArrayList<>();
|
List<IotTree> trees2 = new ArrayList<>();
|
List<IotTree> treeList = new ArrayList<>();
|
|
IotTree iotTree = new IotTree();
|
iotTree.setTitle("IOT采集平台");
|
iotTree.setKey("0");
|
iotTree.setSlots(iconIot);
|
for (ProjectClassify project : projectList) {
|
IotTree it = new IotTree();
|
it.setTitle(project.getProjectName());
|
it.setKey(project.getProjectCode());
|
it.setSlots(iconProject);
|
trees2 = new ArrayList<>();
|
for (ServerDeploy serverDeploy : serverDeployList) {
|
if (project.getId().equals(serverDeploy.getProjectClassifyId())) {
|
List<IotTree> treeList1 = new ArrayList<>();// 实设备
|
List<IotTree> treeList2 = new ArrayList<>(); //虚设备
|
List<IotTree> trees3 = new ArrayList<>(); // 分类
|
IotTree itProject = new IotTree();
|
itProject.setTitle(serverDeploy.getServerCode() + "-" + serverDeploy.getServerName());
|
itProject.setKey("fwq_" + serverDeploy.getServerCode());
|
itProject.setSlots(iconProject1);
|
// 实设备添加
|
IotTree empty = new IotTree();
|
empty.setTitle("实设备");
|
empty.setKey("solid_" + serverDeploy.getServerCode());
|
empty.setSlots(iconParameter);
|
// 虚设备添加
|
IotTree solid = new IotTree();
|
solid.setTitle("虚设备");
|
solid.setKey("empty_" + serverDeploy.getServerCode());
|
solid.setSlots(iconParameter);
|
// 遍历设备信息
|
equipmentList.forEach(eq -> {
|
IotTree emptyEquipment = new IotTree();
|
emptyEquipment.setTitle(eq.getEqptName());
|
List<IotTree> treeList4 = new ArrayList<>();// 参数组
|
// 遍历参数组信息
|
groupList.forEach(g -> {
|
if (eq.getId().equals(g.getEquipmentId())) {
|
IotTree groupTree = new IotTree();
|
groupTree.setTitle(g.getName());
|
groupTree.setKey("ssb_" + g.getId());
|
groupTree.setSlots(iconParameter);
|
treeList4.add(groupTree);
|
}
|
});
|
if (serverDeploy.getId().equals(eq.getServerId()) && eq.getEquipmentType().equals(1)) {
|
emptyEquipment.setSlots(iconSolid);
|
emptyEquipment.setKey("group_" + eq.getId());
|
emptyEquipment.setChildren(treeList4);
|
treeList1.add(emptyEquipment);
|
} else if (serverDeploy.getId().equals(eq.getServerId()) && eq.getEquipmentType().equals(0)) {
|
emptyEquipment.setKey("xxb_" + eq.getId());
|
emptyEquipment.setSlots(iconEmpty);
|
treeList2.add(emptyEquipment);
|
}
|
|
});
|
solid.setChildren(treeList2);
|
empty.setChildren(treeList1);
|
trees3.add(empty);
|
trees3.add(solid);
|
itProject.setChildren(trees3);
|
trees2.add(itProject);
|
}
|
}
|
|
it.setChildren(trees2);
|
trees1.add(it);
|
|
}
|
iotTree.setChildren(trees1);
|
treeList.add(iotTree);
|
return Result.ok(treeList);
|
}
|
|
/**
|
* 通过projectCode查询
|
*
|
* @param projectCode
|
* @return
|
*/
|
@ApiOperation(value = "项目分类-通过projectCode查询", notes = "项目分类-通过projectCode查询")
|
@GetMapping(value = "/queryByProjectCode")
|
public Result<?> queryByProjectCode(@RequestParam(name = "projectCode", required = true) String projectCode) {
|
ProjectClassify projectClassify = projectClassifyService.findByProjectCode(projectCode);
|
return Result.ok(projectClassify);
|
}
|
|
/**
|
* 添加
|
*
|
* @param projectClassify
|
* @return
|
*/
|
@AutoLog(value = "项目分类-添加")
|
@ApiOperation(value = "项目分类-添加", notes = "项目分类-添加")
|
@PostMapping(value = "/add")
|
public Result<?> add(@RequestBody ProjectClassify projectClassify) {
|
projectClassify.setProjectCode(randomDigit());
|
projectClassifyService.addProjectClassify(projectClassify);
|
return Result.ok("添加成功！");
|
}
|
|
/**
|
* 编辑
|
*
|
* @param projectClassify
|
* @return
|
*/
|
@AutoLog(value = "项目分类-编辑")
|
@ApiOperation(value = "项目分类-编辑", notes = "项目分类-编辑")
|
@PutMapping(value = "/edit")
|
public Result<?> edit(@RequestBody ProjectClassify projectClassify) {
|
projectClassifyService.updateById(projectClassify);
|
return Result.ok("编辑成功!");
|
}
|
|
/**
|
* 通过id删除
|
*
|
* @param id
|
* @return
|
*/
|
@AutoLog(value = "项目分类-通过id删除")
|
@ApiOperation(value = "项目分类-通过id删除", notes = "项目分类-通过id删除")
|
@DeleteMapping(value = "/delete")
|
public Result<?> delete(@RequestParam(name = "id", required = true) String id) {
|
return Result.ok(projectClassifyService.projectClassifyRemoveById(id));
|
}
|
|
private String randomDigit() {
|
Random random = new Random();
|
String randomNumber = "xm_" + random.nextInt(9000) + 1000;
|
ProjectClassify projectClassify = projectClassifyService.findByProjectCode(randomNumber);
|
if (projectClassify != null) {
|
randomDigit();
|
}
|
return randomNumber;
|
}
|
|
}
|
