From 2e98b20bea4463e4465e3c19059d0744a09aec06 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Tue, 27 Jun 2023 14:34:41 +0800
Subject: [PATCH] gb28181版本升级-补充
---
src/main/java/com/genersoft/iot/vmp/vmanager/gb28181/device/DeviceQuery.java | 692 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 692 insertions(+), 0 deletions(-)
diff --git a/src/main/java/com/genersoft/iot/vmp/vmanager/gb28181/device/DeviceQuery.java b/src/main/java/com/genersoft/iot/vmp/vmanager/gb28181/device/DeviceQuery.java
new file mode 100644
index 0000000..571c86e
--- /dev/null
+++ b/src/main/java/com/genersoft/iot/vmp/vmanager/gb28181/device/DeviceQuery.java
@@ -0,0 +1,692 @@
+package com.genersoft.iot.vmp.vmanager.gb28181.device;
+
+import com.alibaba.fastjson.JSONObject;
+import com.genersoft.iot.vmp.conf.DynamicTask;
+import com.genersoft.iot.vmp.conf.exception.ControllerException;
+import com.genersoft.iot.vmp.gb28181.bean.Device;
+import com.genersoft.iot.vmp.gb28181.bean.DeviceChannel;
+import com.genersoft.iot.vmp.gb28181.bean.SubscribeHolder;
+import com.genersoft.iot.vmp.gb28181.bean.SyncStatus;
+import com.genersoft.iot.vmp.gb28181.task.ISubscribeTask;
+import com.genersoft.iot.vmp.gb28181.task.impl.CatalogSubscribeTask;
+import com.genersoft.iot.vmp.gb28181.task.impl.MobilePositionSubscribeTask;
+import com.genersoft.iot.vmp.gb28181.transmit.callback.DeferredResultHolder;
+import com.genersoft.iot.vmp.gb28181.transmit.callback.RequestMessage;
+import com.genersoft.iot.vmp.gb28181.transmit.cmd.impl.SIPCommander;
+import com.genersoft.iot.vmp.gb28181.vo.DeviceChannelVO;
+import com.genersoft.iot.vmp.service.IDeviceChannelService;
+import com.genersoft.iot.vmp.service.IDeviceService;
+import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
+import com.genersoft.iot.vmp.storager.IVideoManagerStorage;
+import com.genersoft.iot.vmp.vmanager.bean.BaseTree;
+import com.genersoft.iot.vmp.vmanager.bean.ErrorCode;
+import com.genersoft.iot.vmp.vmanager.bean.WVPResult;
+import com.genersoft.iot.vmp.vmanager.gb28181.vo.PageVO;
+import com.github.pagehelper.PageInfo;
+import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.Parameter;
+import io.swagger.v3.oas.annotations.tags.Tag;
+import org.apache.commons.compress.utils.IOUtils;
+import org.apache.http.HttpResponse;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
+import org.springframework.util.ObjectUtils;
+import org.springframework.util.StringUtils;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.web.context.request.async.DeferredResult;
+
+import javax.servlet.http.HttpServletResponse;
+import javax.sip.DialogState;
+import javax.sip.InvalidArgumentException;
+import javax.sip.SipException;
+import java.io.*;
+import java.nio.file.Files;
+import java.text.ParseException;
+import java.util.*;
+
+@Tag(name = "国标设备查询", description = "国标设备查询")
+@SuppressWarnings("rawtypes")
+@CrossOrigin
+@RestController
+@RequestMapping("/api/device/query")
+public class DeviceQuery {
+
+ private final static Logger logger = LoggerFactory.getLogger(DeviceQuery.class);
+
+ @Autowired
+ private IVideoManagerStorage storager;
+
+ @Autowired
+ private IDeviceChannelService deviceChannelService;
+
+ @Autowired
+ private IRedisCatchStorage redisCatchStorage;
+
+ @Autowired
+ private SIPCommander cmder;
+
+ @Autowired
+ private DeferredResultHolder resultHolder;
+
+ @Autowired
+ private IDeviceService deviceService;
+
+ @Autowired
+ private DynamicTask dynamicTask;
+
+ /**
+ * 使用ID查询国标设备
+ * @param deviceId 国标ID
+ * @return 国标设备
+ */
+ @Operation(summary = "查询国标设备")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @GetMapping("/devices/{deviceId}")
+ public Device devices(@PathVariable String deviceId, HttpServletResponse response){
+ //跨域设置
+ crossDomain(response);
+ return storager.queryVideoDevice(deviceId);
+ }
+
+ /**
+ * 分页查询国标设备
+ * @param page 当前页
+ * @param count 每页查询数量
+ * @return 分页国标列表
+ */
+ @Operation(summary = "分页查询国标设备")
+ @Parameter(name = "page", description = "当前页", required = true)
+ @Parameter(name = "count", description = "每页查询数量", required = true)
+ @GetMapping("/devices")
+ public PageInfo<Device> devices(int page, int count,@RequestParam(required = false) String name, HttpServletResponse response){
+ //跨域设置
+ crossDomain(response);
+ return storager.queryVideoDeviceList(page, count,name);
+ }
+
+ /**
+ * 分页查询通道数
+ *
+ * @param deviceId 设备id
+ * @param page 当前页
+ * @param count 每页条数
+ * @param query 查询内容
+ * @param online 是否在线 在线 true / 离线 false
+ * @param channelType 设备 false/子目录 true
+ * @param catalogUnderDevice 是否直属与设备的目录
+ * @return 通道列表
+ */
+ @GetMapping("/devices/{deviceId}/channels")
+ @Operation(summary = "分页查询通道")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @Parameter(name = "page", description = "当前页", required = true)
+ @Parameter(name = "count", description = "每页查询数量", required = true)
+ @Parameter(name = "query", description = "查询内容")
+ @Parameter(name = "online", description = "是否在线")
+ @Parameter(name = "channelType", description = "设备/子目录-> false/true")
+ @Parameter(name = "catalogUnderDevice", description = "是否直属与设备的目录")
+ public PageInfo channels(@PathVariable String deviceId,
+ int page, int count,
+ @RequestParam(required = false) String query,
+ @RequestParam(required = false) Boolean online,
+ @RequestParam(required = false) Boolean channelType,
+ @RequestParam(required = false) Boolean catalogUnderDevice,
+ @RequestParam(required = false) String deptId,
+ HttpServletResponse response) {
+ //跨域设置
+ crossDomain(response);
+ if (ObjectUtils.isEmpty(query)) {
+ query = null;
+ }
+
+ return storager.queryChannelsByDeviceId(deviceId, query, channelType, online, catalogUnderDevice, page, count,deptId);
+ }
+
+ /**
+ * 分页查询通道数(车辆/摄像头)
+ *
+ * @param deviceId 设备id
+ * @param page 当前页
+ * @param count 每页条数
+ * @param query 查询内容
+ * @param type 类型
+ * @param online 是否在线 在线 true / 离线 false
+ * @param channelType 设备 false/子目录 true
+ * @return 通道列表
+ */
+ @GetMapping("/devices/{deviceId}/channelsCar")
+ @Operation(summary = "分页查询通道")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @Parameter(name = "page", description = "当前页", required = true)
+ @Parameter(name = "count", description = "每页查询数量", required = true)
+ @Parameter(name = "query", description = "查询内容")
+ @Parameter(name="type", description = "设备类型" )
+ @Parameter(name = "online", description = "是否在线")
+ @Parameter(name = "channelType", description = "设备/子目录-> false/true")
+ @Parameter(name = "catalogUnderDevice", description = "是否直属与设备的目录")
+ public PageInfo channelsCar(@PathVariable String deviceId,
+ int page, int count,
+ @RequestParam(required = false) String query,
+ @RequestParam(required = false) String type,
+ @RequestParam(required = false) String dwbh,
+ @RequestParam(required = false) Boolean online,
+ @RequestParam(required = false) Boolean channelType,
+ @RequestParam(required = false) Boolean catalogUnderDevice, HttpServletResponse response) {
+ //跨域设置
+ crossDomain(response);
+ if (ObjectUtils.isEmpty(query)) {
+ query = null;
+ }
+ //查询并返回
+ return storager.queryChannelsCarByDeviceId(deviceId, query,type, channelType,dwbh, online, catalogUnderDevice, page, count);
+ }
+ /**
+ * 查询所有通道数(车辆/摄像头)
+ *
+ * @param deviceId 设备id
+ * @param query 查询内容
+ * @param type 类型
+ * @param online 是否在线 在线 true / 离线 false
+ * @param channelType 设备 false/子目录 true
+ * @return 通道列表
+ */
+ @GetMapping("/devices/{deviceId}/channelsCarAll")
+ @Operation(summary = "分页查询通道")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @Parameter(name = "page", description = "当前页", required = true)
+ @Parameter(name = "count", description = "每页查询数量", required = true)
+ @Parameter(name = "query", description = "查询内容")
+ @Parameter(name="type", description = "设备类型" )
+ @Parameter(name = "online", description = "是否在线")
+ @Parameter(name = "channelType", description = "设备/子目录-> false/true")
+ @Parameter(name = "catalogUnderDevice", description = "是否直属与设备的目录")
+ public List<DeviceChannelVO> channelsCarAll(@PathVariable String deviceId,
+ @RequestParam(required = false) String query,
+ @RequestParam(required = false) String type,
+ @RequestParam(required = false) String dwbh,
+ @RequestParam(required = false) Boolean online,
+ @RequestParam(required = false) Boolean channelType,
+ @RequestParam(required = false) Boolean catalogUnderDevice, HttpServletResponse response) {
+ //跨域设置
+ crossDomain(response);
+ if (ObjectUtils.isEmpty(query)) {
+ query = null;
+ }
+
+ return storager.channelsCarAll(deviceId, query,type, channelType,dwbh, online, catalogUnderDevice);
+ }
+
+ /**
+ * 设备面查询
+ * @param page
+ * @return
+ */
+ @PostMapping("/SpatialQueryList")
+ @Operation(summary = "设备面查询")
+ public PageInfo SpatialQueryList(@RequestBody PageVO page, HttpServletResponse response) {
+ //跨域设置
+ crossDomain(response);
+ return storager.SpatialQueryList(page);
+ }
+
+ /**
+ * 设备面查询-不分页
+ * @param page
+ * @return
+ */
+ @PostMapping("/spatialQueryListAll")
+ @Operation(summary = "设备面查询")
+ public Object spatialQueryListAll(@RequestBody PageVO page, HttpServletResponse response) {
+ //跨域设置
+ crossDomain(response);
+ return storager.spatialQueryListAll(page);
+ }
+
+ /**
+ * 同步设备通道
+ * @param deviceId 设备id
+ * @return
+ */
+ @Operation(summary = "同步设备通道")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @PostMapping("/devices/{deviceId}/sync")
+ public WVPResult<SyncStatus> devicesSync(@PathVariable String deviceId){
+
+ if (logger.isDebugEnabled()) {
+ logger.debug("设备通道信息同步API调用，deviceId：" + deviceId);
+ }
+ Device device = storager.queryVideoDevice(deviceId);
+ boolean status = deviceService.isSyncRunning(deviceId);
+ // 已存在则返回进度
+ if (status) {
+ SyncStatus channelSyncStatus = deviceService.getChannelSyncStatus(deviceId);
+ return WVPResult.success(channelSyncStatus);
+ }
+ deviceService.sync(device);
+
+ WVPResult<SyncStatus> wvpResult = new WVPResult<>();
+ wvpResult.setCode(0);
+ wvpResult.setMsg("开始同步");
+ return wvpResult;
+ }
+
+ /**
+ * 移除设备
+ * @param deviceId 设备id
+ * @return
+ */
+ @Operation(summary = "移除设备")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @DeleteMapping("/devices/{deviceId}/delete")
+ public String delete(@PathVariable String deviceId){
+
+ if (logger.isDebugEnabled()) {
+ logger.debug("设备信息删除API调用，deviceId：" + deviceId);
+ }
+
+ // 清除redis记录
+ boolean isSuccess = deviceService.delete(deviceId);
+ if (isSuccess) {
+ redisCatchStorage.clearCatchByDeviceId(deviceId);
+ // 停止此设备的订阅更新
+ Set<String> allKeys = dynamicTask.getAllKeys();
+ for (String key : allKeys) {
+ if (key.startsWith(deviceId)) {
+ Runnable runnable = dynamicTask.get(key);
+ if (runnable instanceof ISubscribeTask) {
+ ISubscribeTask subscribeTask = (ISubscribeTask) runnable;
+ subscribeTask.stop();
+ }
+ dynamicTask.stop(key);
+ }
+ }
+ JSONObject json = new JSONObject();
+ json.put("deviceId", deviceId);
+ return json.toString();
+ } else {
+ logger.warn("设备信息删除API调用失败！");
+ throw new ControllerException(ErrorCode.ERROR100.getCode(), "设备信息删除API调用失败！");
+ }
+ }
+
+ /**
+ * 分页查询子目录通道
+ * @param deviceId 通道id
+ * @param channelId 通道id
+ * @param page 当前页
+ * @param count 每页条数
+ * @param query 查询内容
+ * @param online 是否在线
+ * @param channelType 通道类型
+ * @return 子通道列表
+ */
+ @Operation(summary = "分页查询子目录通道")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @Parameter(name = "channelId", description = "通道国标编号", required = true)
+ @Parameter(name = "page", description = "当前页", required = true)
+ @Parameter(name = "count", description = "每页查询数量", required = true)
+ @Parameter(name = "query", description = "查询内容")
+ @Parameter(name = "online", description = "是否在线")
+ @Parameter(name = "channelType", description = "设备/子目录-> false/true")
+ @GetMapping("/sub_channels/{deviceId}/{channelId}/channels")
+ public PageInfo subChannels(@PathVariable String deviceId,
+ @PathVariable String channelId,
+ int page,
+ int count,
+ @RequestParam(required = false) String query,
+ @RequestParam(required = false) String deptId,
+ @RequestParam(required = false) Boolean online,
+ @RequestParam(required = false) Boolean channelType){
+
+ DeviceChannel deviceChannel = storager.queryChannel(deviceId,channelId);
+ if (deviceChannel == null) {
+ PageInfo<DeviceChannel> deviceChannelPageResult = new PageInfo<>();
+ return deviceChannelPageResult;
+ }
+
+ PageInfo pageResult = storager.querySubChannels(deviceId, channelId, query, channelType, online, page, count,deptId);
+ return pageResult;
+ }
+
+ /**
+ * 更新通道信息
+ * @param deviceId 设备id
+ * @param channel 通道
+ * @return
+ */
+ @Operation(summary = "更新通道信息")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @Parameter(name = "channel", description = "通道信息", required = true)
+ @PostMapping("/channel/update/{deviceId}")
+ public void updateChannel(@PathVariable String deviceId,DeviceChannel channel){
+ deviceChannelService.updateChannel(deviceId, channel);
+ }
+
+ /**
+ * 修改数据流传输模式
+ * @param deviceId 设备id
+ * @param streamMode 数据流传输模式
+ * @return
+ */
+ @Operation(summary = "修改数据流传输模式")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @Parameter(name = "streamMode", description = "数据流传输模式, 取值：" +
+ "UDP（udp传输），TCP-ACTIVE（tcp主动模式,暂不支持），TCP-PASSIVE（tcp被动模式）", required = true)
+ @PostMapping("/transport/{deviceId}/{streamMode}")
+ public void updateTransport(@PathVariable String deviceId, @PathVariable String streamMode){
+ Device device = deviceService.getDevice(deviceId);
+ device.setStreamMode(streamMode);
+ deviceService.updateCustomDevice(device);
+ }
+
+ /**
+ * 添加设备信息
+ * @param device 设备信息
+ * @return
+ */
+ @Operation(summary = "添加设备信息")
+ @Parameter(name = "device", description = "设备", required = true)
+ @PostMapping("/device/add/")
+ public void addDevice(Device device){
+
+ if (device == null || device.getDeviceId() == null) {
+ throw new ControllerException(ErrorCode.ERROR400);
+ }
+
+ // 查看deviceId是否存在
+ boolean exist = deviceService.isExist(device.getDeviceId());
+ if (exist) {
+ throw new ControllerException(ErrorCode.ERROR100.getCode(), "设备编号已存在");
+ }
+ deviceService.addDevice(device);
+ }
+
+ /**
+ * 更新设备信息
+ * @param device 设备信息
+ * @return
+ */
+ @Operation(summary = "更新设备信息")
+ @Parameter(name = "device", description = "设备", required = true)
+ @PostMapping("/device/update/")
+ public void updateDevice(Device device){
+
+ if (device != null && device.getDeviceId() != null) {
+ deviceService.updateCustomDevice(device);
+ }
+ }
+
+ /**
+ * 设备状态查询请求API接口
+ *
+ * @param deviceId 设备id
+ */
+ @Operation(summary = "设备状态查询")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @GetMapping("/devices/{deviceId}/status")
+ public DeferredResult<ResponseEntity<String>> deviceStatusApi(@PathVariable String deviceId) {
+ if (logger.isDebugEnabled()) {
+ logger.debug("设备状态查询API调用");
+ }
+ Device device = storager.queryVideoDevice(deviceId);
+ String uuid = UUID.randomUUID().toString();
+ String key = DeferredResultHolder.CALLBACK_CMD_DEVICESTATUS + deviceId;
+ DeferredResult<ResponseEntity<String>> result = new DeferredResult<ResponseEntity<String>>(2*1000L);
+ if(device == null) {
+ result.setResult(new ResponseEntity(String.format("设备%s不存在", deviceId),HttpStatus.OK));
+ return result;
+ }
+ try {
+ cmder.deviceStatusQuery(device, event -> {
+ RequestMessage msg = new RequestMessage();
+ msg.setId(uuid);
+ msg.setKey(key);
+ msg.setData(String.format("获取设备状态失败，错误码： %s, %s", event.statusCode, event.msg));
+ resultHolder.invokeResult(msg);
+ });
+ } catch (InvalidArgumentException | SipException | ParseException e) {
+ logger.error("[命令发送失败] 获取设备状态: {}", e.getMessage());
+ throw new ControllerException(ErrorCode.ERROR100.getCode(), "命令发送失败: " + e.getMessage());
+ }
+ result.onTimeout(()->{
+ logger.warn(String.format("获取设备状态超时"));
+ // 释放rtpserver
+ RequestMessage msg = new RequestMessage();
+ msg.setId(uuid);
+ msg.setKey(key);
+ msg.setData("Timeout. Device did not response to this command.");
+ resultHolder.invokeResult(msg);
+ });
+ resultHolder.put(DeferredResultHolder.CALLBACK_CMD_DEVICESTATUS + deviceId, uuid, result);
+ return result;
+ }
+
+ /**
+ * 设备报警查询请求API接口
+ * @param deviceId 设备id
+ * @param startPriority 报警起始级别（可选）
+ * @param endPriority 报警终止级别（可选）
+ * @param alarmMethod 报警方式条件（可选）
+ * @param alarmType 报警类型
+ * @param startTime 报警发生起始时间（可选）
+ * @param endTime 报警发生终止时间（可选）
+ * @return true = 命令发送成功
+ */
+ @Operation(summary = "设备状态查询")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @Parameter(name = "startPriority", description = "报警起始级别")
+ @Parameter(name = "endPriority", description = "报警终止级别")
+ @Parameter(name = "alarmMethod", description = "报警方式条件")
+ @Parameter(name = "alarmType", description = "报警类型")
+ @Parameter(name = "startTime", description = "报警发生起始时间")
+ @Parameter(name = "endTime", description = "报警发生终止时间")
+ @GetMapping("/alarm/{deviceId}")
+ public DeferredResult<ResponseEntity<String>> alarmApi(@PathVariable String deviceId,
+ @RequestParam(required = false) String startPriority,
+ @RequestParam(required = false) String endPriority,
+ @RequestParam(required = false) String alarmMethod,
+ @RequestParam(required = false) String alarmType,
+ @RequestParam(required = false) String startTime,
+ @RequestParam(required = false) String endTime) {
+ if (logger.isDebugEnabled()) {
+ logger.debug("设备报警查询API调用");
+ }
+ Device device = storager.queryVideoDevice(deviceId);
+ String key = DeferredResultHolder.CALLBACK_CMD_ALARM + deviceId;
+ String uuid = UUID.randomUUID().toString();
+ try {
+ cmder.alarmInfoQuery(device, startPriority, endPriority, alarmMethod, alarmType, startTime, endTime, event -> {
+ RequestMessage msg = new RequestMessage();
+ msg.setId(uuid);
+ msg.setKey(key);
+ msg.setData(String.format("设备报警查询失败，错误码： %s, %s",event.statusCode, event.msg));
+ resultHolder.invokeResult(msg);
+ });
+ } catch (InvalidArgumentException | SipException | ParseException e) {
+ logger.error("[命令发送失败] 设备报警查询: {}", e.getMessage());
+ throw new ControllerException(ErrorCode.ERROR100.getCode(), "命令发送失败: " + e.getMessage());
+ }
+ DeferredResult<ResponseEntity<String>> result = new DeferredResult<ResponseEntity<String >> (3 * 1000L);
+ result.onTimeout(()->{
+ logger.warn(String.format("设备报警查询超时"));
+ // 释放rtpserver
+ RequestMessage msg = new RequestMessage();
+ msg.setId(uuid);
+ msg.setKey(key);
+ msg.setData("设备报警查询超时");
+ resultHolder.invokeResult(msg);
+ });
+ resultHolder.put(DeferredResultHolder.CALLBACK_CMD_ALARM + deviceId, uuid, result);
+ return result;
+ }
+
+
+ @GetMapping("/{deviceId}/sync_status")
+ @Operation(summary = "获取通道同步进度")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ public WVPResult<SyncStatus> getSyncStatus(@PathVariable String deviceId) {
+ SyncStatus channelSyncStatus = deviceService.getChannelSyncStatus(deviceId);
+ WVPResult<SyncStatus> wvpResult = new WVPResult<>();
+ if (channelSyncStatus == null) {
+ wvpResult.setCode(-1);
+ wvpResult.setMsg("同步尚未开始");
+ }else {
+ wvpResult.setCode(ErrorCode.SUCCESS.getCode());
+ wvpResult.setMsg(ErrorCode.SUCCESS.getMsg());
+ wvpResult.setData(channelSyncStatus);
+ if (channelSyncStatus.getErrorMsg() != null) {
+ wvpResult.setMsg(channelSyncStatus.getErrorMsg());
+ }
+ }
+ return wvpResult;
+ }
+
+ @GetMapping("/{deviceId}/subscribe_info")
+ @Operation(summary = "获取设备的订阅状态")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ public WVPResult<Map<String, Integer>> getSubscribeInfo(@PathVariable String deviceId) {
+ Set<String> allKeys = dynamicTask.getAllKeys();
+ Map<String, Integer> dialogStateMap = new HashMap<>();
+ for (String key : allKeys) {
+ if (key.startsWith(deviceId)) {
+ ISubscribeTask subscribeTask = (ISubscribeTask)dynamicTask.get(key);
+ if (subscribeTask instanceof CatalogSubscribeTask) {
+ dialogStateMap.put("catalog", 1);
+ }else if (subscribeTask instanceof MobilePositionSubscribeTask) {
+ dialogStateMap.put("mobilePosition", 1);
+ }
+ }
+ }
+ WVPResult<Map<String, Integer>> wvpResult = new WVPResult<>();
+ wvpResult.setCode(0);
+ wvpResult.setData(dialogStateMap);
+ return wvpResult;
+ }
+
+ @GetMapping("/snap/{deviceId}/{channelId}")
+ @Operation(summary = "请求截图")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @Parameter(name = "channelId", description = "通道国标编号", required = true)
+ public void getSnap(HttpServletResponse resp, @PathVariable String deviceId, @PathVariable String channelId) {
+
+ try {
+ final InputStream in = Files.newInputStream(new File("snap" + File.separator + deviceId + "_" + channelId + ".jpg").toPath());
+ resp.setContentType(MediaType.IMAGE_PNG_VALUE);
+ IOUtils.copy(in, resp.getOutputStream());
+ } catch (IOException e) {
+ resp.setStatus(HttpServletResponse.SC_NOT_FOUND);
+ }
+ }
+
+ /**
+ * 查询国标树
+ * @param deviceId 设备ID
+ * @param parentId 父ID
+ * @param deptId 辖区id
+ * @param page 当前页
+ * @param count 每页条数
+ * @return 国标设备
+ */
+ @Operation(summary = "查询国标树")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @Parameter(name = "parentId", description = "父级国标编号")
+ @Parameter(name = "deptId", description = "辖区id")
+ @Parameter(name = "onlyCatalog", description = "只获取目录")
+ @Parameter(name = "page", description = "当前页", required = true)
+ @Parameter(name = "count", description = "每页条数", required = true)
+ @GetMapping("/tree/{deviceId}")
+ public ResponseEntity<PageInfo> getTree(@PathVariable String deviceId,
+ @RequestParam(required = false) String parentId,
+ @RequestParam(required = false) String deptId,
+ @RequestParam(required = false) Boolean onlyCatalog,
+ int page, int count){
+
+
+ if (page <= 0) {
+ page = 1;
+ }
+ if (onlyCatalog == null) {
+ onlyCatalog = false;
+ }
+
+ List<BaseTree<DeviceChannel>> treeData = deviceService.queryVideoDeviceTree(deviceId, parentId, onlyCatalog,deptId);
+ if (treeData == null || (page - 1) * count > treeData.size()) {
+ PageInfo<BaseTree<DeviceChannel>> pageInfo = new PageInfo<>();
+ pageInfo.setPageNum(page);
+ pageInfo.setTotal(treeData == null? 0 : treeData.size());
+ pageInfo.setSize(0);
+ pageInfo.setList(new ArrayList<>());
+ return new ResponseEntity<>(pageInfo,HttpStatus.OK);
+ }
+
+ int toIndex = Math.min(page * count, treeData.size());
+ // 处理分页
+ List<BaseTree<DeviceChannel>> trees = treeData.subList((page - 1) * count, toIndex);
+ PageInfo<BaseTree<DeviceChannel>> pageInfo = new PageInfo<>();
+ pageInfo.setPageNum(page);
+ pageInfo.setTotal(treeData.size());
+ pageInfo.setSize(trees.size());
+ pageInfo.setList(trees);
+
+ return new ResponseEntity<>(pageInfo,HttpStatus.OK);
+ }
+
+ /**
+ * 查询国标树下的通道
+ * @param deviceId 设备ID
+ * @param parentId 父ID
+ * @param page 当前页
+ * @param count 每页条数
+ * @return 国标设备
+ */
+ @Operation(summary = "查询国标树下的通道")
+ @Parameter(name = "deviceId", description = "设备国标编号", required = true)
+ @Parameter(name = "parentId", description = "父级国标编号")
+ @Parameter(name = "page", description = "当前页", required = true)
+ @Parameter(name = "count", description = "每页条数", required = true)
+ @GetMapping("/tree/channel/{deviceId}")
+ public ResponseEntity<PageInfo> getChannelInTreeNode(@PathVariable String deviceId,
+ @RequestParam(required = false) String parentId,
+ @RequestParam(required = false) String deptId,
+ int page, int count){
+
+ if (page <= 0) {
+ page = 1;
+ }
+
+ List<DeviceChannel> treeData = deviceService.queryVideoDeviceInTreeNode(deviceId, parentId,deptId);
+ if (treeData == null || (page - 1) * count > treeData.size()) {
+ PageInfo<BaseTree<DeviceChannel>> pageInfo = new PageInfo<>();
+ pageInfo.setPageNum(page);
+ pageInfo.setTotal(treeData == null? 0 : treeData.size());
+ pageInfo.setSize(0);
+ pageInfo.setList(new ArrayList<>());
+ return new ResponseEntity<>(pageInfo,HttpStatus.OK);
+ }
+
+ int toIndex = Math.min(page * count, treeData.size());
+ // 处理分页
+ List<DeviceChannel> trees = treeData.subList((page - 1) * count, toIndex);
+ PageInfo<DeviceChannel> pageInfo = new PageInfo<>();
+ pageInfo.setPageNum(page);
+ pageInfo.setTotal(treeData.size());
+ pageInfo.setSize(trees.size());
+ pageInfo.setList(trees);
+
+ return new ResponseEntity<>(pageInfo,HttpStatus.OK);
+ }
+
+ /**
+ * 配置跨域设置
+ * @param response
+ */
+ private void crossDomain(HttpServletResponse response) {
+ response.setHeader("Access-Control-Allow-Origin", "*");
+ response.setHeader("Access-Control-Allow-Methods", "POST, GET, OPTIONS, DELETE");
+ response.setHeader("Access-Control-Allow-Credentials","true");
+ }
+}
--
Gitblit v1.9.3