package com.tievd.jyz.controller;
|
|
import cn.hutool.core.util.NumberUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.tievd.cube.commons.annotations.AutoLog;
|
import com.tievd.cube.commons.annotations.DictApi;
|
import com.tievd.cube.commons.base.CubeController;
|
import com.tievd.cube.commons.base.Result;
|
import com.tievd.cube.commons.constant.CommonConst;
|
import com.tievd.cube.commons.mybatisplus.QueryGenerator;
|
import com.tievd.cube.commons.utils.SystemContextUtil;
|
import com.tievd.cube.modules.system.model.LoginUser;
|
import com.tievd.cube.modules.system.model.SysDepartTreeModel;
|
import com.tievd.cube.modules.system.service.ISysDepartService;
|
import com.tievd.jyz.cache.DeviceCache;
|
import com.tievd.jyz.constants.SystemConstant;
|
import com.tievd.jyz.dto.DeviceDTO;
|
import com.tievd.jyz.entity.Device;
|
import com.tievd.jyz.handler.BaseDataSyncHandler;
|
import com.tievd.jyz.handler.DeviceStatusHandler;
|
import com.tievd.jyz.service.IDeviceService;
|
import io.swagger.v3.oas.annotations.Parameter;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.IOException;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* Device
|
*
|
* @author cube
|
* @version V2.0.0
|
* @since 2023-02-27
|
*/
|
@Slf4j
|
@DictApi
|
@RestController
|
@RequestMapping("/jyz/device")
|
public class DeviceController extends CubeController<Device, IDeviceService> {
|
|
@Autowired
|
private IDeviceService deviceService;
|
|
@Autowired
|
private DeviceCache deviceCache;
|
|
@Autowired
|
private ISysDepartService sysDepartService;
|
@Autowired
|
private BaseDataSyncHandler baseDataSyncHandler;
|
|
/**
|
* 分页列表查询
|
*/
|
@GetMapping("/list")
|
public Result<?> queryPageList(Device device,
|
@RequestParam(defaultValue = "1") Integer pageNo,
|
@RequestParam(defaultValue = "10") Integer pageSize,
|
HttpServletRequest req) {
|
QueryWrapper<Device> queryWrapper = QueryGenerator.initQueryWrapper(device, req.getParameterMap());
|
Page<Device> page = new Page<>(pageNo, pageSize);
|
IPage<Device> pageList = deviceService.page(page, queryWrapper);
|
return Result.ok(pageList);
|
}
|
|
@GetMapping("/tables")
|
public Result<?> tables(Device device,
|
@RequestParam(defaultValue = "1") Integer pageNo,
|
@RequestParam(defaultValue = "10") Integer pageSize,
|
@RequestParam(required = false) String searchKey,
|
@RequestParam(required = false) String parentCode,
|
@RequestParam(required = false) Byte status) {
|
|
QueryWrapper<Device> queryWrapper = new QueryWrapper<>();
|
if (StringUtils.isNotBlank(searchKey)) {
|
queryWrapper.like("name", searchKey);
|
}
|
if (StringUtils.isNotBlank(parentCode)) {
|
queryWrapper.likeRight("org_code", parentCode);
|
}
|
if (status != null) {
|
queryWrapper.eq("status", status);
|
}
|
Page<Device> page = new Page<>(pageNo, pageSize);
|
IPage<DeviceDTO> pageList = deviceService.tables(page, queryWrapper);
|
Map<String, String> cameraStatusMap = DeviceStatusHandler.getCameraStatusMap();
|
for (DeviceDTO deviceDTO : pageList.getRecords()) {
|
String sn = deviceDTO.getSn();
|
int onlineCount = 0;
|
for (Map.Entry<String, String> entry : cameraStatusMap.entrySet()) {
|
if (entry.getKey().contains(sn)) {
|
if (NumberUtil.parseInt(entry.getValue()) == SystemConstant.DEVICE_ONLINE) {
|
onlineCount++;
|
}
|
}
|
}
|
deviceDTO.setCameraOnlineCount(onlineCount);
|
}
|
return Result.ok(pageList);
|
}
|
|
@GetMapping("/allList")
|
public Result<?> allList(Device device, HttpServletRequest req) {
|
QueryWrapper<Device> queryWrapper = new QueryWrapper<>(device);
|
List<Device> deviceList = deviceService.list(queryWrapper);
|
return Result.ok(deviceList);
|
}
|
|
/**
|
* 添加
|
*/
|
@AutoLog("Device-添加")
|
@PostMapping("/add")
|
public Result<?> add(@RequestBody Device device) {
|
device.setStatus(SystemConstant.DEVICE_OFFLINE);
|
deviceService.save(device);
|
deviceCache.init();
|
return Result.ok();
|
}
|
|
/**
|
* 同步网关基础数据
|
* @param deviceSn 网关序列号
|
* @return
|
*/
|
@AutoLog("拉取/同步网关基础数据")
|
@GetMapping("/syncData")
|
public Result<?> syncData(@RequestParam String deviceSn) {
|
//同步网关基础数据
|
baseDataSyncHandler.pullData(SystemConstant.ALL_REQ_SOURCE,deviceSn);
|
return Result.ok();
|
}
|
|
/**
|
* 编辑
|
*/
|
@AutoLog("Device-编辑")
|
@PutMapping("/edit")
|
public Result<?> edit(@RequestBody Device device) {
|
deviceService.updateById(device);
|
deviceCache.init();
|
return Result.ok();
|
}
|
|
/**
|
* 通过id删除
|
*/
|
@AutoLog("Device-通过id删除")
|
@DeleteMapping("/delete")
|
public Result<?> delete(@RequestParam String id) {
|
deviceService.deleteById(id);
|
deviceCache.init();
|
return Result.ok();
|
}
|
|
/**
|
* 通过id查询
|
*/
|
@GetMapping("/queryById")
|
public Result<?> queryById(@RequestParam String id) {
|
Device device = deviceService.getById(id);
|
return Result.ok(device);
|
}
|
|
/**
|
* 导出excel
|
*/
|
@RequestMapping("/exportXls")
|
public void exportXls(HttpServletRequest request, HttpServletResponse response, Device device) throws IOException {
|
super.exportXls(request, response, device, "Device");
|
}
|
|
/**
|
* 通过excel导入数据
|
*/
|
@PostMapping("/importExcel")
|
public Result<?> importExcel(HttpServletRequest request) throws Exception {
|
Result<?> result = super.importExcel(request, Device.class);
|
deviceCache.init();
|
return result;
|
}
|
|
@GetMapping("/queryTreeList")
|
public Result<?> queryTreeList(@RequestParam(required = false) boolean myself) {
|
List<SysDepartTreeModel> list;
|
if (myself) {
|
LoginUser user = SystemContextUtil.currentLoginUser();
|
list = sysDepartService.queryMyDeptTreeList(user.getDepartIds());
|
} else {
|
list = sysDepartService.queryTreeList();
|
}
|
deviceService.assembleDevice(list);
|
return Result.ok(list);
|
}
|
|
@GetMapping("/searchBy")
|
public Result<List<SysDepartTreeModel>> searchBy(@Parameter(name = "关键词") String keyWord, @Parameter(name = "我的部门下搜索") @RequestParam(required = false) String myDeptSearch) {
|
//部门查询，myDeptSearch为1时为我的部门查询，登录用户为上级时查只查负责部门下数据
|
LoginUser user = SystemContextUtil.currentLoginUser();
|
String departIds = null;
|
if (user.getUserIdentity() != null && user.getUserIdentity().equals(CommonConst.USER_IDENTITY_2)) {
|
departIds = user.getDepartIds();
|
}
|
List<SysDepartTreeModel> treeList = sysDepartService.searchBy(keyWord, myDeptSearch, departIds);
|
if (treeList == null || treeList.size() == 0) {
|
return Result.error("未查询匹配数据！");
|
}
|
deviceService.assembleDevice(treeList);
|
return Result.ok(treeList);
|
}
|
}
|
