648540858
2022-08-19 1a9e49d9ff210e39f6297150db758906a4f02e6f
src/main/java/com/genersoft/iot/vmp/vmanager/gb28181/device/DeviceQuery.java
@@ -12,15 +12,16 @@
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.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.WVPResult;
import com.github.pagehelper.PageInfo;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
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;
@@ -39,7 +40,7 @@
import java.nio.file.Files;
import java.util.*;
@Api(tags = "国标设备查询", value = "国标设备查询")
@Tag(name  = "国标设备查询", description = "国标设备查询")
@SuppressWarnings("rawtypes")
@CrossOrigin
@RestController
@@ -50,6 +51,9 @@
   
   @Autowired
   private IVideoManagerStorage storager;
   @Autowired
   private IDeviceChannelService deviceChannelService;
   @Autowired
   private IRedisCatchStorage redisCatchStorage;
@@ -74,16 +78,10 @@
    * @param deviceId 国标ID
    * @return 国标设备
    */
   @ApiOperation("使用ID查询国标设备")
   @ApiImplicitParams({
         @ApiImplicitParam(name = "deviceId", value = "设备ID", required = true, dataTypeClass = String.class),
   })
   @Operation(summary = "查询国标设备")
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @GetMapping("/devices/{deviceId}")
   public ResponseEntity<Device> devices(@PathVariable String deviceId){
//      if (logger.isDebugEnabled()) {
//         logger.debug("查询视频设备API调用,deviceId:" + deviceId);
//      }
      
      Device device = storager.queryVideoDevice(deviceId);
      return new ResponseEntity<>(device,HttpStatus.OK);
@@ -95,17 +93,11 @@
    * @param count 每页查询数量
    * @return 分页国标列表
    */
   @ApiOperation("分页查询国标设备")
   @ApiImplicitParams({
         @ApiImplicitParam(name = "page", value = "当前页", required = true, dataTypeClass = Integer.class),
         @ApiImplicitParam(name = "count", value = "每页查询数量", required = true, dataTypeClass = Integer.class),
   })
   @Operation(summary = "分页查询国标设备")
   @Parameter(name = "page", description = "当前页", required = true)
   @Parameter(name = "count", description = "每页查询数量", required = true)
   @GetMapping("/devices")
   public PageInfo<Device> devices(int page, int count){
//      if (logger.isDebugEnabled()) {
//         logger.debug("查询所有视频设备API调用");
//      }
      
      return storager.queryVideoDeviceList(page, count);
   }
@@ -119,28 +111,24 @@
    * @param query 查询内容
    * @param online 是否在线  在线 true / 离线 false
    * @param channelType 设备 false/子目录 true
    * @param catalogUnderDevice 是否直属与设备的目录
    * @return 通道列表
    */
   @ApiOperation("分页查询通道")
   @GetMapping("/devices/{deviceId}/channels")
   @ApiImplicitParams({
         @ApiImplicitParam(name="deviceId", value = "设备id", required = true ,dataTypeClass = String.class),
         @ApiImplicitParam(name="page", value = "当前页", required = true ,dataTypeClass = Integer.class),
         @ApiImplicitParam(name="count", value = "每页查询数量", required = true ,dataTypeClass = Integer.class),
         @ApiImplicitParam(name="query", value = "查询内容" ,dataTypeClass = String.class),
         @ApiImplicitParam(name="online", value = "是否在线"  ,dataTypeClass = Boolean.class),
         @ApiImplicitParam(name="channelType", value = "设备/子目录-> false/true" ,dataTypeClass = Boolean.class),
         @ApiImplicitParam(name="catalogUnderDevice", value = "是否直属与设备的目录" ,dataTypeClass = Boolean.class),
   })
   @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 ResponseEntity<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) {
//      if (logger.isDebugEnabled()) {
//         logger.debug("查询视频设备通道API调用");
//      }
      if (StringUtils.isEmpty(query)) {
         query = null;
      }
@@ -154,10 +142,8 @@
    * @param deviceId 设备id
    * @return
    */
   @ApiOperation("同步设备通道")
   @ApiImplicitParams({
         @ApiImplicitParam(name="deviceId", value = "设备id", required = true, dataTypeClass = String.class),
   })
   @Operation(summary = "同步设备通道")
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @PostMapping("/devices/{deviceId}/sync")
   public WVPResult<SyncStatus> devicesSync(@PathVariable String deviceId){
      
@@ -187,10 +173,8 @@
    * @param deviceId 设备id
    * @return
    */
   @ApiOperation("移除设备")
   @ApiImplicitParams({
         @ApiImplicitParam(name="deviceId", value = "设备id", required = true, dataTypeClass = String.class),
   })
   @Operation(summary = "移除设备")
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @DeleteMapping("/devices/{deviceId}/delete")
   public ResponseEntity<String> delete(@PathVariable String deviceId){
      
@@ -206,6 +190,11 @@
         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);
            }
         }
@@ -229,16 +218,14 @@
    * @param channelType 通道类型
    * @return 子通道列表
    */
   @ApiOperation("分页查询子目录通道")
   @ApiImplicitParams({
         @ApiImplicitParam(name="deviceId", value = "设备id", required = true, dataTypeClass = String.class),
         @ApiImplicitParam(name="channelId", value = "通道id", required = true, dataTypeClass = String.class),
         @ApiImplicitParam(name="page", value = "当前页", required = true, dataTypeClass = Integer.class),
         @ApiImplicitParam(name="count", value = "每页条数", required = true, dataTypeClass = Integer.class),
         @ApiImplicitParam(name="query", value = "查询内容", dataTypeClass = String.class),
         @ApiImplicitParam(name="online", value = "是否在线", dataTypeClass = Boolean.class),
         @ApiImplicitParam(name="channelType", value = "通道类型, 子目录", dataTypeClass = Boolean.class),
   })
   @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 ResponseEntity<PageInfo> subChannels(@PathVariable String deviceId,
                                      @PathVariable String channelId,
@@ -248,9 +235,6 @@
                                      @RequestParam(required = false) Boolean online,
                                      @RequestParam(required = false) Boolean channelType){
//      if (logger.isDebugEnabled()) {
//         logger.debug("查询所有视频通道API调用");
//      }
      DeviceChannel deviceChannel = storager.queryChannel(deviceId,channelId);
      if (deviceChannel == null) {
         PageInfo<DeviceChannel> deviceChannelPageResult = new PageInfo<>();
@@ -267,14 +251,12 @@
    * @param channel 通道
    * @return
    */
   @ApiOperation("更新通道信息")
   @ApiImplicitParams({
         @ApiImplicitParam(name="deviceId", value = "设备id", required = true, dataTypeClass = String.class),
         @ApiImplicitParam(name="channel", value = "通道", required = true, dataTypeClass = String.class),
   })
   @Operation(summary = "更新通道信息")
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @Parameter(name = "channel", description = "通道信息", required = true)
   @PostMapping("/channel/update/{deviceId}")
   public ResponseEntity<PageInfo> updateChannel(@PathVariable String deviceId,DeviceChannel channel){
      storager.updateChannel(deviceId, channel);
   public ResponseEntity updateChannel(@PathVariable String deviceId,DeviceChannel channel){
      deviceChannelService.updateChannel(deviceId, channel);
      return new ResponseEntity<>(null,HttpStatus.OK);
   }
@@ -284,17 +266,14 @@
    * @param streamMode 数据流传输模式
    * @return
    */
   @ApiOperation("修改数据流传输模式")
   @ApiImplicitParams({
         @ApiImplicitParam(name = "deviceId", value = "设备id", required = true, dataTypeClass = String.class),
         @ApiImplicitParam(name = "streamMode", value = "数据流传输模式, 取值:" +
               "UDP(udp传输),TCP-ACTIVE(tcp主动模式,暂不支持),TCP-PASSIVE(tcp被动模式)", dataTypeClass = String.class),
   })
   @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 ResponseEntity<PageInfo> updateTransport(@PathVariable String deviceId, @PathVariable String streamMode){
   public ResponseEntity updateTransport(@PathVariable String deviceId, @PathVariable String streamMode){
      Device device = storager.queryVideoDevice(deviceId);
      device.setStreamMode(streamMode);
//      storager.updateDevice(device);
      deviceService.updateDevice(device);
      return new ResponseEntity<>(null,HttpStatus.OK);
   }
@@ -304,20 +283,13 @@
    * @param device 设备信息
    * @return
    */
   @ApiOperation("更新设备信息")
   @ApiImplicitParams({
         @ApiImplicitParam(name = "device", value = "设备信息", required = true, dataTypeClass = Device.class)
   })
   @Operation(summary = "更新设备信息")
   @Parameter(name = "device", description = "设备", required = true)
   @PostMapping("/device/update/")
   public ResponseEntity<WVPResult<String>> updateDevice(Device device){
      if (device != null && device.getDeviceId() != null) {
         // TODO 报警订阅相关的信息
         deviceService.updateDevice(device);
//         cmder.deviceInfoQuery(device);
      }
      WVPResult<String> result = new WVPResult<>();
      result.setCode(0);
@@ -330,10 +302,8 @@
    * 
    * @param deviceId 设备id
    */
   @ApiOperation("设备状态查询")
   @ApiImplicitParams({
         @ApiImplicitParam(name = "deviceId", value = "设备id", required = true, dataTypeClass = String.class),
   })
   @Operation(summary = "设备状态查询")
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @GetMapping("/devices/{deviceId}/status")
   public DeferredResult<ResponseEntity<String>> deviceStatusApi(@PathVariable String deviceId) {
      if (logger.isDebugEnabled()) {
@@ -342,6 +312,11 @@
      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;
      }
      cmder.deviceStatusQuery(device, event -> {
         RequestMessage msg = new RequestMessage();
         msg.setId(uuid);
@@ -349,7 +324,6 @@
         msg.setData(String.format("获取设备状态失败,错误码: %s, %s", event.statusCode, event.msg));
         resultHolder.invokeResult(msg);
      });
        DeferredResult<ResponseEntity<String>> result = new DeferredResult<ResponseEntity<String>>(2*1000L);
      result.onTimeout(()->{
         logger.warn(String.format("获取设备状态超时"));
         // 释放rtpserver
@@ -374,16 +348,14 @@
    * @param endTime      报警发生终止时间(可选)
    * @return            true = 命令发送成功
    */
   @ApiOperation("设备报警查询")
   @ApiImplicitParams({
         @ApiImplicitParam(name = "deviceId", value = "设备id", required = true, dataTypeClass = String.class),
         @ApiImplicitParam(name = "startPriority", value = "报警起始级别", dataTypeClass = String.class),
         @ApiImplicitParam(name = "endPriority", value = "报警终止级别", dataTypeClass = String.class),
         @ApiImplicitParam(name = "alarmMethod", value = "报警方式条件", dataTypeClass = String.class),
         @ApiImplicitParam(name = "alarmType", value = "报警类型", dataTypeClass = String.class),
         @ApiImplicitParam(name = "startTime", value = "报警发生起始时间", dataTypeClass = String.class),
         @ApiImplicitParam(name = "endTime", value = "报警发生终止时间", dataTypeClass = String.class),
   })
   @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, 
@@ -421,7 +393,8 @@
   @GetMapping("/{deviceId}/sync_status")
   @ApiOperation(value = "获取通道同步进度", notes = "获取通道同步进度")
   @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<>();
@@ -439,7 +412,8 @@
   }
   @GetMapping("/{deviceId}/subscribe_info")
   @ApiOperation(value = "获取设备的订阅状态", notes = "获取设备的订阅状态")
   @Operation(summary = "获取设备的订阅状态")
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   public WVPResult<Map<String, String>> getSubscribeInfo(@PathVariable String deviceId) {
      Set<String> allKeys = dynamicTask.getAllKeys();
      Map<String, String> dialogStateMap = new HashMap<>();
@@ -464,7 +438,9 @@
   }
   @GetMapping("/snap/{deviceId}/{channelId}")
   @ApiOperation(value = "请求截图", notes = "请求截图")
   @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 {
@@ -475,4 +451,98 @@
         resp.setStatus(HttpServletResponse.SC_NOT_FOUND);
      }
   }
   /**
    * 查询国标树
    * @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 = "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) 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);
      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, int page, int count){
      if (page <= 0) {
         page = 1;
      }
      List<DeviceChannel> treeData = deviceService.queryVideoDeviceInTreeNode(deviceId, parentId);
      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);
   }
}