|  |  |  | 
|---|
|  |  |  | import org.slf4j.LoggerFactory; | 
|---|
|  |  |  | import org.springframework.beans.factory.InitializingBean; | 
|---|
|  |  |  | import org.springframework.beans.factory.annotation.Autowired; | 
|---|
|  |  |  | import org.springframework.beans.factory.annotation.Qualifier; | 
|---|
|  |  |  | import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; | 
|---|
|  |  |  | import org.springframework.stereotype.Component; | 
|---|
|  |  |  | import org.springframework.util.StringUtils; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | 
|---|
|  |  |  | import javax.sip.SipException; | 
|---|
|  |  |  | import javax.sip.message.Response; | 
|---|
|  |  |  | import java.text.ParseException; | 
|---|
|  |  |  | import java.text.SimpleDateFormat; | 
|---|
|  |  |  | import java.util.ArrayList; | 
|---|
|  |  |  | import java.util.Iterator; | 
|---|
|  |  |  | import java.util.List; | 
|---|
|  |  |  | import java.util.concurrent.ConcurrentLinkedQueue; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Component | 
|---|
|  |  |  | public class CatalogResponseMessageHandler extends SIPRequestProcessorParent implements InitializingBean, IMessageHandler { | 
|---|
|  |  |  | 
|---|
|  |  |  | private Logger logger = LoggerFactory.getLogger(CatalogResponseMessageHandler.class); | 
|---|
|  |  |  | private final String cmdType = "Catalog"; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | private SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); | 
|---|
|  |  |  | private boolean taskQueueHandlerRun = false; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private ResponseMessageHandler responseMessageHandler; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | private ConcurrentLinkedQueue<HandlerCatchData> taskQueue = new ConcurrentLinkedQueue<>(); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IVideoManagerStorage storager; | 
|---|
|  |  |  | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private IRedisCatchStorage redisCatchStorage; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Qualifier("taskExecutor") | 
|---|
|  |  |  | @Autowired | 
|---|
|  |  |  | private ThreadPoolTaskExecutor taskExecutor; | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public void afterPropertiesSet() throws Exception { | 
|---|
|  |  |  | responseMessageHandler.addHandler(cmdType, this); | 
|---|
|  |  |  | 
|---|
|  |  |  |  | 
|---|
|  |  |  | @Override | 
|---|
|  |  |  | public void handForDevice(RequestEvent evt, Device device, Element element) { | 
|---|
|  |  |  | String key = DeferredResultHolder.CALLBACK_CMD_CATALOG + device.getDeviceId(); | 
|---|
|  |  |  | Element rootElement = null; | 
|---|
|  |  |  | taskQueue.offer(new HandlerCatchData(evt, device, element)); | 
|---|
|  |  |  | // 回复200 OK | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | rootElement = getRootElement(evt, device.getCharset()); | 
|---|
|  |  |  | Element deviceListElement = rootElement.element("DeviceList"); | 
|---|
|  |  |  | Element sumNumElement = rootElement.element("SumNum"); | 
|---|
|  |  |  | Element snElement = rootElement.element("SN"); | 
|---|
|  |  |  | if (snElement == null || sumNumElement == null || deviceListElement == null) { | 
|---|
|  |  |  | responseAck(evt, Response.BAD_REQUEST, "xml error"); | 
|---|
|  |  |  | return; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | int sumNum = Integer.parseInt(sumNumElement.getText()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | if (sumNum == 0) { | 
|---|
|  |  |  | // 数据已经完整接收 | 
|---|
|  |  |  | storager.cleanChannelsForDevice(device.getDeviceId()); | 
|---|
|  |  |  | catalogDataCatch.setChannelSyncEnd(device.getDeviceId(), null); | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | Iterator<Element> deviceListIterator = deviceListElement.elementIterator(); | 
|---|
|  |  |  | if (deviceListIterator != null) { | 
|---|
|  |  |  | List<DeviceChannel> channelList = new ArrayList<>(); | 
|---|
|  |  |  | // 遍历DeviceList | 
|---|
|  |  |  | while (deviceListIterator.hasNext()) { | 
|---|
|  |  |  | Element itemDevice = deviceListIterator.next(); | 
|---|
|  |  |  | Element channelDeviceElement = itemDevice.element("DeviceID"); | 
|---|
|  |  |  | if (channelDeviceElement == null) { | 
|---|
|  |  |  | continue; | 
|---|
|  |  |  | responseAck(evt, Response.OK); | 
|---|
|  |  |  | } catch (SipException e) { | 
|---|
|  |  |  | throw new RuntimeException(e); | 
|---|
|  |  |  | } catch (InvalidArgumentException e) { | 
|---|
|  |  |  | throw new RuntimeException(e); | 
|---|
|  |  |  | } catch (ParseException e) { | 
|---|
|  |  |  | throw new RuntimeException(e); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | if (!taskQueueHandlerRun) { | 
|---|
|  |  |  | taskQueueHandlerRun = true; | 
|---|
|  |  |  | taskExecutor.execute(()-> { | 
|---|
|  |  |  | while (!taskQueue.isEmpty()) { | 
|---|
|  |  |  | HandlerCatchData take = taskQueue.poll(); | 
|---|
|  |  |  | String key = DeferredResultHolder.CALLBACK_CMD_CATALOG + take.getDevice().getDeviceId(); | 
|---|
|  |  |  | Element rootElement = null; | 
|---|
|  |  |  | try { | 
|---|
|  |  |  | rootElement = getRootElement(take.getEvt(), take.getDevice().getCharset()); | 
|---|
|  |  |  | Element deviceListElement = rootElement.element("DeviceList"); | 
|---|
|  |  |  | Element sumNumElement = rootElement.element("SumNum"); | 
|---|
|  |  |  | Element snElement = rootElement.element("SN"); | 
|---|
|  |  |  | if (snElement == null || sumNumElement == null || deviceListElement == null) { | 
|---|
|  |  |  | responseAck(take.getEvt(), Response.BAD_REQUEST, "xml error"); | 
|---|
|  |  |  | return; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | //by brewswang | 
|---|
|  |  |  | //                        if (NumericUtil.isDouble(XmlUtil.getText(itemDevice, "Longitude"))) {//如果包含位置信息,就更新一下位置 | 
|---|
|  |  |  | //                            processNotifyMobilePosition(evt, itemDevice); | 
|---|
|  |  |  | //                        } | 
|---|
|  |  |  | DeviceChannel deviceChannel = XmlUtil.channelContentHander(itemDevice); | 
|---|
|  |  |  | deviceChannel.setDeviceId(device.getDeviceId()); | 
|---|
|  |  |  | int sumNum = Integer.parseInt(sumNumElement.getText()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | channelList.add(deviceChannel); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | int sn = Integer.parseInt(snElement.getText()); | 
|---|
|  |  |  | catalogDataCatch.put(device.getDeviceId(), sn, sumNum, device, channelList); | 
|---|
|  |  |  | logger.info("收到来自设备【{}】的通道: {}个,{}/{}", device.getDeviceId(), channelList.size(), catalogDataCatch.get(device.getDeviceId()) == null ? 0 :catalogDataCatch.get(device.getDeviceId()).size(), sumNum); | 
|---|
|  |  |  | if (catalogDataCatch.get(device.getDeviceId()).size() == sumNum) { | 
|---|
|  |  |  | // 数据已经完整接收 | 
|---|
|  |  |  | boolean resetChannelsResult = storager.resetChannels(device.getDeviceId(), catalogDataCatch.get(device.getDeviceId())); | 
|---|
|  |  |  | if (!resetChannelsResult) { | 
|---|
|  |  |  | String errorMsg = "接收成功,写入失败,共" + sumNum + "条,已接收" + catalogDataCatch.get(device.getDeviceId()).size() + "条"; | 
|---|
|  |  |  | catalogDataCatch.setChannelSyncEnd(device.getDeviceId(), errorMsg); | 
|---|
|  |  |  | if (sumNum == 0) { | 
|---|
|  |  |  | // 数据已经完整接收 | 
|---|
|  |  |  | storager.cleanChannelsForDevice(take.getDevice().getDeviceId()); | 
|---|
|  |  |  | catalogDataCatch.setChannelSyncEnd(take.getDevice().getDeviceId(), null); | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | catalogDataCatch.setChannelSyncEnd(device.getDeviceId(), null); | 
|---|
|  |  |  | Iterator<Element> deviceListIterator = deviceListElement.elementIterator(); | 
|---|
|  |  |  | if (deviceListIterator != null) { | 
|---|
|  |  |  | List<DeviceChannel> channelList = new ArrayList<>(); | 
|---|
|  |  |  | // 遍历DeviceList | 
|---|
|  |  |  | while (deviceListIterator.hasNext()) { | 
|---|
|  |  |  | Element itemDevice = deviceListIterator.next(); | 
|---|
|  |  |  | Element channelDeviceElement = itemDevice.element("DeviceID"); | 
|---|
|  |  |  | if (channelDeviceElement == null) { | 
|---|
|  |  |  | continue; | 
|---|
|  |  |  | } | 
|---|
|  |  |  | //by brewswang | 
|---|
|  |  |  | //                        if (NumericUtil.isDouble(XmlUtil.getText(itemDevice, "Longitude"))) {//如果包含位置信息,就更新一下位置 | 
|---|
|  |  |  | //                            processNotifyMobilePosition(evt, itemDevice); | 
|---|
|  |  |  | //                        } | 
|---|
|  |  |  | DeviceChannel deviceChannel = XmlUtil.channelContentHander(itemDevice, device); | 
|---|
|  |  |  | deviceChannel.setDeviceId(take.getDevice().getDeviceId()); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | channelList.add(deviceChannel); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | int sn = Integer.parseInt(snElement.getText()); | 
|---|
|  |  |  | catalogDataCatch.put(take.getDevice().getDeviceId(), sn, sumNum, take.getDevice(), channelList); | 
|---|
|  |  |  | logger.info("收到来自设备【{}】的通道: {}个,{}/{}", take.getDevice().getDeviceId(), channelList.size(), catalogDataCatch.get(take.getDevice().getDeviceId()) == null ? 0 :catalogDataCatch.get(take.getDevice().getDeviceId()).size(), sumNum); | 
|---|
|  |  |  | if (catalogDataCatch.get(take.getDevice().getDeviceId()).size() == sumNum) { | 
|---|
|  |  |  | // 数据已经完整接收 | 
|---|
|  |  |  | boolean resetChannelsResult = storager.resetChannels(take.getDevice().getDeviceId(), catalogDataCatch.get(take.getDevice().getDeviceId())); | 
|---|
|  |  |  | if (!resetChannelsResult) { | 
|---|
|  |  |  | String errorMsg = "接收成功,写入失败,共" + sumNum + "条,已接收" + catalogDataCatch.get(take.getDevice().getDeviceId()).size() + "条"; | 
|---|
|  |  |  | catalogDataCatch.setChannelSyncEnd(take.getDevice().getDeviceId(), errorMsg); | 
|---|
|  |  |  | }else { | 
|---|
|  |  |  | catalogDataCatch.setChannelSyncEnd(take.getDevice().getDeviceId(), null); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } catch (DocumentException e) { | 
|---|
|  |  |  | e.printStackTrace(); | 
|---|
|  |  |  | } catch (InvalidArgumentException e) { | 
|---|
|  |  |  | e.printStackTrace(); | 
|---|
|  |  |  | } catch (ParseException e) { | 
|---|
|  |  |  | e.printStackTrace(); | 
|---|
|  |  |  | } catch (SipException e) { | 
|---|
|  |  |  | e.printStackTrace(); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  | // 回复200 OK | 
|---|
|  |  |  | responseAck(evt, Response.OK); | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } catch (DocumentException e) { | 
|---|
|  |  |  | e.printStackTrace(); | 
|---|
|  |  |  | } catch (InvalidArgumentException e) { | 
|---|
|  |  |  | e.printStackTrace(); | 
|---|
|  |  |  | } catch (ParseException e) { | 
|---|
|  |  |  | e.printStackTrace(); | 
|---|
|  |  |  | } catch (SipException e) { | 
|---|
|  |  |  | e.printStackTrace(); | 
|---|
|  |  |  | taskQueueHandlerRun = false; | 
|---|
|  |  |  | }); | 
|---|
|  |  |  |  | 
|---|
|  |  |  | } | 
|---|
|  |  |  | } | 
|---|
|  |  |  |  | 
|---|