648540858
2024-03-28 1de344674afd6bb35b51b165bbad76dbe6299b7e
src/main/java/com/genersoft/iot/vmp/vmanager/gb28181/device/DeviceQuery.java
old mode 100644 new mode 100755
@@ -3,6 +3,7 @@
import com.alibaba.fastjson2.JSONObject;
import com.genersoft.iot.vmp.conf.DynamicTask;
import com.genersoft.iot.vmp.conf.exception.ControllerException;
import com.genersoft.iot.vmp.conf.security.JwtUtils;
import com.genersoft.iot.vmp.gb28181.bean.Device;
import com.genersoft.iot.vmp.gb28181.bean.DeviceChannel;
import com.genersoft.iot.vmp.gb28181.bean.SyncStatus;
@@ -14,6 +15,7 @@
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.service.IInviteStreamService;
import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
import com.genersoft.iot.vmp.storager.IVideoManagerStorage;
import com.genersoft.iot.vmp.vmanager.bean.BaseTree;
@@ -22,8 +24,10 @@
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.security.SecurityRequirement;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.apache.commons.compress.utils.IOUtils;
import org.apache.ibatis.annotations.Options;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
@@ -34,6 +38,7 @@
import org.springframework.web.bind.annotation.*;
import org.springframework.web.context.request.async.DeferredResult;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import javax.sip.InvalidArgumentException;
import javax.sip.SipException;
@@ -46,7 +51,7 @@
@Tag(name  = "国标设备查询", description = "国标设备查询")
@SuppressWarnings("rawtypes")
@CrossOrigin
@RestController
@RequestMapping("/api/device/query")
public class DeviceQuery {
@@ -61,6 +66,9 @@
   @Autowired
   private IRedisCatchStorage redisCatchStorage;
   @Autowired
   private IInviteStreamService inviteStreamService;
   
   @Autowired
   private SIPCommander cmder;
@@ -79,7 +87,7 @@
    * @param deviceId 国标ID
    * @return 国标设备
    */
   @Operation(summary = "查询国标设备")
   @Operation(summary = "查询国标设备", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @GetMapping("/devices/{deviceId}")
   public Device devices(@PathVariable String deviceId){
@@ -93,12 +101,14 @@
    * @param count 每页查询数量
    * @return 分页国标列表
    */
   @Operation(summary = "分页查询国标设备")
   @Operation(summary = "分页查询国标设备", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "page", description = "当前页", required = true)
   @Parameter(name = "count", description = "每页查询数量", required = true)
   @GetMapping("/devices")
   @Options()
   public PageInfo<Device> devices(int page, int count){
//      if (page == null) page = 0;
//      if (count == null) count = 20;
      return storager.queryVideoDeviceList(page, count,null);
   }
@@ -115,7 +125,7 @@
    * @return 通道列表
    */
   @GetMapping("/devices/{deviceId}/channels")
   @Operation(summary = "分页查询通道")
   @Operation(summary = "分页查询通道", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @Parameter(name = "page", description = "当前页", required = true)
   @Parameter(name = "count", description = "每页查询数量", required = true)
@@ -123,7 +133,7 @@
   @Parameter(name = "online", description = "是否在线")
   @Parameter(name = "channelType", description = "设备/子目录-> false/true")
   @Parameter(name = "catalogUnderDevice", description = "是否直属与设备的目录")
   public PageInfo channels(@PathVariable String deviceId,
   public PageInfo<DeviceChannel> channels(@PathVariable String deviceId,
                                    int page, int count,
                                    @RequestParam(required = false) String query,
                                    @RequestParam(required = false) Boolean online,
@@ -141,7 +151,7 @@
    * @param deviceId 设备id
    * @return
    */
   @Operation(summary = "同步设备通道")
   @Operation(summary = "同步设备通道", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @GetMapping("/devices/{deviceId}/sync")
   public WVPResult<SyncStatus> devicesSync(@PathVariable String deviceId){
@@ -169,7 +179,7 @@
    * @param deviceId 设备id
    * @return
    */
   @Operation(summary = "移除设备")
   @Operation(summary = "移除设备", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @DeleteMapping("/devices/{deviceId}/delete")
   public String delete(@PathVariable String deviceId){
@@ -181,7 +191,7 @@
      // 清除redis记录
      boolean isSuccess = deviceService.delete(deviceId);
      if (isSuccess) {
         redisCatchStorage.clearCatchByDeviceId(deviceId);
         inviteStreamService.clearInviteInfo(deviceId);
         // 停止此设备的订阅更新
         Set<String> allKeys = dynamicTask.getAllKeys();
         for (String key : allKeys) {
@@ -189,7 +199,7 @@
               Runnable runnable = dynamicTask.get(key);
               if (runnable instanceof ISubscribeTask) {
                  ISubscribeTask subscribeTask = (ISubscribeTask) runnable;
                  subscribeTask.stop();
                  subscribeTask.stop(null);
               }
               dynamicTask.stop(key);
            }
@@ -214,7 +224,7 @@
    * @param channelType 通道类型
    * @return 子通道列表
    */
   @Operation(summary = "分页查询子目录通道")
   @Operation(summary = "分页查询子目录通道", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @Parameter(name = "channelId", description = "通道国标编号", required = true)
   @Parameter(name = "page", description = "当前页", required = true)
@@ -223,7 +233,7 @@
   @Parameter(name = "online", description = "是否在线")
   @Parameter(name = "channelType", description = "设备/子目录-> false/true")
   @GetMapping("/sub_channels/{deviceId}/{channelId}/channels")
   public PageInfo subChannels(@PathVariable String deviceId,
   public PageInfo<DeviceChannel> subChannels(@PathVariable String deviceId,
                                      @PathVariable String channelId,
                                      int page,
                                      int count,
@@ -237,8 +247,7 @@
         return deviceChannelPageResult;
      }
      PageInfo pageResult = storager.querySubChannels(deviceId, channelId, query, channelType, online, page, count);
      return pageResult;
      return storager.querySubChannels(deviceId, channelId, query, channelType, online, page, count);
   }
   /**
@@ -247,12 +256,20 @@
    * @param channel 通道
    * @return
    */
   @Operation(summary = "更新通道信息")
   @Operation(summary = "更新通道信息", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @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);
   }
   @Operation(summary = "修改通道的码流类型", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @Parameter(name = "channel", description = "通道信息", required = true)
   @PostMapping("/channel/stream/identification/update/")
   public void updateChannelStreamIdentification(DeviceChannel channel){
      deviceChannelService.updateChannelStreamIdentification(channel);
   }
   /**
@@ -261,7 +278,7 @@
    * @param streamMode 数据流传输模式
    * @return
    */
   @Operation(summary = "修改数据流传输模式")
   @Operation(summary = "修改数据流传输模式", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @Parameter(name = "streamMode", description = "数据流传输模式, 取值:" +
         "UDP(udp传输),TCP-ACTIVE(tcp主动模式,暂不支持),TCP-PASSIVE(tcp被动模式)", required = true)
@@ -277,7 +294,7 @@
    * @param device 设备信息
    * @return
    */
   @Operation(summary = "添加设备信息")
   @Operation(summary = "添加设备信息", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "device", description = "设备", required = true)
   @PostMapping("/device/add/")
   public void addDevice(Device device){
@@ -299,7 +316,7 @@
    * @param device 设备信息
    * @return
    */
   @Operation(summary = "更新设备信息")
   @Operation(summary = "更新设备信息", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "device", description = "设备", required = true)
   @PostMapping("/device/update/")
   public void updateDevice(Device device){
@@ -314,7 +331,7 @@
    * 
    * @param deviceId 设备id
    */
   @Operation(summary = "设备状态查询")
   @Operation(summary = "设备状态查询", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @GetMapping("/devices/{deviceId}/status")
   public DeferredResult<ResponseEntity<String>> deviceStatusApi(@PathVariable String deviceId) {
@@ -365,7 +382,7 @@
    * @param endTime      报警发生终止时间(可选)
    * @return            true = 命令发送成功
    */
   @Operation(summary = "设备状态查询")
   @Operation(summary = "设备报警查询", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   @Parameter(name = "startPriority", description = "报警起始级别")
   @Parameter(name = "endPriority", description = "报警终止级别")
@@ -415,7 +432,7 @@
   @GetMapping("/{deviceId}/sync_status")
   @Operation(summary = "获取通道同步进度")
   @Operation(summary = "获取通道同步进度", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   public WVPResult<SyncStatus> getSyncStatus(@PathVariable String deviceId) {
      SyncStatus channelSyncStatus = deviceService.getChannelSyncStatus(deviceId);
@@ -435,7 +452,7 @@
   }
   @GetMapping("/{deviceId}/subscribe_info")
   @Operation(summary = "获取设备的订阅状态")
   @Operation(summary = "获取设备的订阅状态", security = @SecurityRequirement(name = JwtUtils.HEADER))
   @Parameter(name = "deviceId", description = "设备国标编号", required = true)
   public WVPResult<Map<String, Integer>> getSubscribeInfo(@PathVariable String deviceId) {
      Set<String> allKeys = dynamicTask.getAllKeys();
@@ -460,12 +477,16 @@
   @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) {
   @Parameter(name = "mark", description = "标识", required = false)
   public void getSnap(HttpServletResponse resp, @PathVariable String deviceId, @PathVariable String channelId, @RequestParam(required = false) String mark) {
      try {
         final InputStream in = Files.newInputStream(new File("snap" + File.separator + deviceId + "_" + channelId + ".jpg").toPath());
         final InputStream in = Files.newInputStream(new File("snap" + File.separator + deviceId + "_" + channelId + (mark == null? ".jpg": ("_" + mark + ".jpg"))).toPath());
         resp.setContentType(MediaType.IMAGE_PNG_VALUE);
         ServletOutputStream outputStream = resp.getOutputStream();
         IOUtils.copy(in, resp.getOutputStream());
         in.close();
         outputStream.close();
      } catch (IOException e) {
         resp.setStatus(HttpServletResponse.SC_NOT_FOUND);
      }