|  |  |  | 
|---|
|  |  |  | 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.gb28181.bean.Device; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.bean.DeviceChannel; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.gb28181.event.DeviceOffLineDetector; | 
|---|
|  |  |  | 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.service.IDeviceService; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IRedisCatchStorage; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IVideoManagerStorager; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.storager.IVideoManagerStorage; | 
|---|
|  |  |  | import com.genersoft.iot.vmp.vmanager.bean.WVPResult; | 
|---|
|  |  |  | import com.github.pagehelper.PageInfo; | 
|---|
|  |  |  | import io.swagger.annotations.Api; | 
|---|
|  |  |  | 
|---|
|  |  |  | import org.springframework.web.bind.annotation.*; | 
|---|
|  |  |  | import org.springframework.web.context.request.async.DeferredResult; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | import java.util.UUID; | 
|---|
|  |  |  | import javax.sip.DialogState; | 
|---|
|  |  |  | import java.util.*; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Api(tags = "国标设备查询", value = "国标设备查询") | 
|---|
|  |  |  | @SuppressWarnings("rawtypes") | 
|---|
|  |  |  | 
|---|
|  |  |  | private final static Logger logger = LoggerFactory.getLogger(DeviceQuery.class); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IVideoManagerStorager storager; | 
|---|
|  |  |  | private IVideoManagerStorage storager; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IRedisCatchStorage redisCatchStorage; | 
|---|
|  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private DeferredResultHolder resultHolder; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private DeviceOffLineDetector offLineDetector; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IDeviceService deviceService; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private DynamicTask dynamicTask; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private SubscribeHolder subscribeHolder; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | * 使用ID查询国标设备 | 
|---|
|  |  |  | 
|---|
|  |  |  | @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), | 
|---|
|  |  |  | }) | 
|---|
|  |  |  | 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 channelType, | 
|---|
|  |  |  | @RequestParam(required = false) Boolean catalogUnderDevice) { | 
|---|
|  |  |  | //      if (logger.isDebugEnabled()) { | 
|---|
|  |  |  | //         logger.debug("查询视频设备通道API调用"); | 
|---|
|  |  |  | //      } | 
|---|
|  |  |  | 
|---|
|  |  |  | query = null; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | PageInfo pageResult = storager.queryChannelsByDeviceId(deviceId, query, channelType, online, page, count); | 
|---|
|  |  |  | PageInfo pageResult = storager.queryChannelsByDeviceId(deviceId, query, channelType, online, catalogUnderDevice, page, count); | 
|---|
|  |  |  | return new ResponseEntity<>(pageResult,HttpStatus.OK); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | @ApiImplicitParam(name="deviceId", value = "设备id", required = true, dataTypeClass = String.class), | 
|---|
|  |  |  | }) | 
|---|
|  |  |  | @PostMapping("/devices/{deviceId}/sync") | 
|---|
|  |  |  | public DeferredResult<ResponseEntity<Device>> devicesSync(@PathVariable String deviceId){ | 
|---|
|  |  |  | public WVPResult<SyncStatus> devicesSync(@PathVariable String deviceId){ | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (logger.isDebugEnabled()) { | 
|---|
|  |  |  | logger.debug("设备通道信息同步API调用,deviceId:" + deviceId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | Device device = storager.queryVideoDevice(deviceId); | 
|---|
|  |  |  | String key = DeferredResultHolder.CALLBACK_CMD_CATALOG + deviceId; | 
|---|
|  |  |  | String uuid = UUID.randomUUID().toString(); | 
|---|
|  |  |  | DeferredResult<ResponseEntity<Device>> result = new DeferredResult<ResponseEntity<Device>>(15*1000L); | 
|---|
|  |  |  | result.onTimeout(()->{ | 
|---|
|  |  |  | logger.warn(String.format("设备通道信息同步超时")); | 
|---|
|  |  |  | // 释放rtpserver | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setKey(key); | 
|---|
|  |  |  | msg.setId(uuid); | 
|---|
|  |  |  | msg.setData("Timeout"); | 
|---|
|  |  |  | resultHolder.invokeAllResult(msg); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | // 等待其他相同请求返回时一起返回 | 
|---|
|  |  |  | if (resultHolder.exist(key, null)) { | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | boolean status = deviceService.isSyncRunning(deviceId); | 
|---|
|  |  |  | // 已存在则返回进度 | 
|---|
|  |  |  | if (status) { | 
|---|
|  |  |  | WVPResult<SyncStatus> wvpResult = new WVPResult<>(); | 
|---|
|  |  |  | wvpResult.setCode(0); | 
|---|
|  |  |  | SyncStatus channelSyncStatus = deviceService.getChannelSyncStatus(deviceId); | 
|---|
|  |  |  | wvpResult.setData(channelSyncStatus); | 
|---|
|  |  |  | return wvpResult; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | cmder.catalogQuery(device, event -> { | 
|---|
|  |  |  | RequestMessage msg = new RequestMessage(); | 
|---|
|  |  |  | msg.setKey(key); | 
|---|
|  |  |  | msg.setId(uuid); | 
|---|
|  |  |  | msg.setData(String.format("同步通道失败,错误码: %s, %s", event.statusCode, event.msg)); | 
|---|
|  |  |  | resultHolder.invokeAllResult(msg); | 
|---|
|  |  |  | }); | 
|---|
|  |  |  | deviceService.sync(device); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | resultHolder.put(key, uuid, result); | 
|---|
|  |  |  | return result; | 
|---|
|  |  |  | WVPResult<SyncStatus> wvpResult = new WVPResult<>(); | 
|---|
|  |  |  | wvpResult.setCode(0); | 
|---|
|  |  |  | wvpResult.setMsg("开始同步"); | 
|---|
|  |  |  | return wvpResult; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | /** | 
|---|
|  |  |  | 
|---|
|  |  |  | if (logger.isDebugEnabled()) { | 
|---|
|  |  |  | logger.debug("设备信息删除API调用,deviceId:" + deviceId); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (offLineDetector.isOnline(deviceId)) { | 
|---|
|  |  |  | return new ResponseEntity<String>("不允许删除在线设备!", HttpStatus.NOT_ACCEPTABLE); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | // 清除redis记录 | 
|---|
|  |  |  | boolean isSuccess = storager.delete(deviceId); | 
|---|
|  |  |  | if (isSuccess) { | 
|---|
|  |  |  | redisCatchStorage.clearCatchByDeviceId(deviceId); | 
|---|
|  |  |  | // 停止此设备的订阅更新 | 
|---|
|  |  |  | Set<String> allKeys = dynamicTask.getAllKeys(); | 
|---|
|  |  |  | for (String key : allKeys) { | 
|---|
|  |  |  | if (key.startsWith(deviceId)) { | 
|---|
|  |  |  | dynamicTask.stop(key); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | JSONObject json = new JSONObject(); | 
|---|
|  |  |  | json.put("deviceId", deviceId); | 
|---|
|  |  |  | return new ResponseEntity<>(json.toString(),HttpStatus.OK); | 
|---|
|  |  |  | 
|---|
|  |  |  | @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 = String.class), | 
|---|
|  |  |  | @ApiImplicitParam(name="online", value = "是否在线", dataTypeClass = Boolean.class), | 
|---|
|  |  |  | @ApiImplicitParam(name="channelType", value = "通道类型, 子目录", dataTypeClass = Boolean.class), | 
|---|
|  |  |  | }) | 
|---|
|  |  |  | @GetMapping("/sub_channels/{deviceId}/{channelId}/channels") | 
|---|
|  |  |  | 
|---|
|  |  |  | int page, | 
|---|
|  |  |  | int count, | 
|---|
|  |  |  | @RequestParam(required = false) String query, | 
|---|
|  |  |  | @RequestParam(required = false) String online, | 
|---|
|  |  |  | @RequestParam(required = false) Boolean online, | 
|---|
|  |  |  | @RequestParam(required = false) Boolean channelType){ | 
|---|
|  |  |  |  | 
|---|
|  |  |  | //      if (logger.isDebugEnabled()) { | 
|---|
|  |  |  | 
|---|
|  |  |  | public ResponseEntity<PageInfo> updateTransport(@PathVariable String deviceId, @PathVariable String streamMode){ | 
|---|
|  |  |  | Device device = storager.queryVideoDevice(deviceId); | 
|---|
|  |  |  | device.setStreamMode(streamMode); | 
|---|
|  |  |  | storager.updateDevice(device); | 
|---|
|  |  |  | //      storager.updateDevice(device); | 
|---|
|  |  |  | deviceService.updateDevice(device); | 
|---|
|  |  |  | return new ResponseEntity<>(null,HttpStatus.OK); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | public ResponseEntity<WVPResult<String>> updateDevice(Device device){ | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (device != null && device.getDeviceId() != null) { | 
|---|
|  |  |  | Device deviceInStore = storager.queryVideoDevice(device.getDeviceId()); | 
|---|
|  |  |  | if (!StringUtils.isEmpty(device.getName())) deviceInStore.setName(device.getName()); | 
|---|
|  |  |  | if (!StringUtils.isEmpty(device.getCharset())) deviceInStore.setCharset(device.getCharset()); | 
|---|
|  |  |  | if (!StringUtils.isEmpty(device.getMediaServerId())) deviceInStore.setMediaServerId(device.getMediaServerId()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (deviceInStore.getSubscribeCycleForCatalog() <=0 && device.getSubscribeCycleForCatalog() > 0) { | 
|---|
|  |  |  | deviceInStore.setSubscribeCycleForCatalog(device.getSubscribeCycleForCatalog()); | 
|---|
|  |  |  | // 开启订阅 | 
|---|
|  |  |  | deviceService.addCatalogSubscribe(deviceInStore); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (deviceInStore.getSubscribeCycleForCatalog() > 0 && device.getSubscribeCycleForCatalog() <= 0) { | 
|---|
|  |  |  | deviceInStore.setSubscribeCycleForCatalog(device.getSubscribeCycleForCatalog()); | 
|---|
|  |  |  | // 取消订阅 | 
|---|
|  |  |  | deviceService.removeCatalogSubscribe(deviceInStore); | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | storager.updateDevice(deviceInStore); | 
|---|
|  |  |  | cmder.deviceInfoQuery(deviceInStore); | 
|---|
|  |  |  | // TODO 报警订阅相关的信息 | 
|---|
|  |  |  |  | 
|---|
|  |  |  | deviceService.updateDevice(device); | 
|---|
|  |  |  | //         cmder.deviceInfoQuery(device); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | WVPResult<String> result = new WVPResult<>(); | 
|---|
|  |  |  | result.setCode(0); | 
|---|
|  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @GetMapping("/{deviceId}/sync_status") | 
|---|
|  |  |  | @ApiOperation(value = "获取通道同步进度", notes = "获取通道同步进度") | 
|---|
|  |  |  | 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(0); | 
|---|
|  |  |  | wvpResult.setData(channelSyncStatus); | 
|---|
|  |  |  | if (channelSyncStatus.getErrorMsg() != null) { | 
|---|
|  |  |  | wvpResult.setMsg(channelSyncStatus.getErrorMsg()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | return wvpResult; | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @GetMapping("/{deviceId}/subscribe_info") | 
|---|
|  |  |  | @ApiOperation(value = "获取设备的订阅状态", notes = "获取设备的订阅状态") | 
|---|
|  |  |  | public WVPResult<Map<String, String>> getSubscribeInfo(@PathVariable String deviceId) { | 
|---|
|  |  |  | Set<String> allKeys = dynamicTask.getAllKeys(); | 
|---|
|  |  |  | Map<String, String> dialogStateMap = new HashMap<>(); | 
|---|
|  |  |  | for (String key : allKeys) { | 
|---|
|  |  |  | if (key.startsWith(deviceId)) { | 
|---|
|  |  |  | ISubscribeTask subscribeTask = (ISubscribeTask)dynamicTask.get(key); | 
|---|
|  |  |  | DialogState dialogState = subscribeTask.getDialogState(); | 
|---|
|  |  |  | if (dialogState == null) { | 
|---|
|  |  |  | continue; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (subscribeTask instanceof CatalogSubscribeTask) { | 
|---|
|  |  |  | dialogStateMap.put("catalog", dialogState.toString()); | 
|---|
|  |  |  | }else if (subscribeTask instanceof MobilePositionSubscribeTask) { | 
|---|
|  |  |  | dialogStateMap.put("mobilePosition", dialogState.toString()); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | WVPResult<Map<String, String>> wvpResult = new WVPResult<>(); | 
|---|
|  |  |  | wvpResult.setCode(0); | 
|---|
|  |  |  | wvpResult.setData(dialogStateMap); | 
|---|
|  |  |  | return wvpResult; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|