添加截图（快照）功能

panlinlin

2021-05-08 e48fa711a3664bece9b3e58840a75fe7c05bc47c

 src/main/java/com/genersoft/iot/vmp/gb28181/transmit/request/impl/RegisterRequestProcessor.java

@@ -16,6 +16,12 @@
import javax.sip.message.Request;

import javax.sip.message.Response;



import com.genersoft.iot.vmp.gb28181.bean.WvpSipDate;

import gov.nist.javax.sip.RequestEventExt;

import gov.nist.javax.sip.header.SIPDateHeader;

import gov.nist.javax.sip.message.SIPRequest;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.util.StringUtils;



import com.genersoft.iot.vmp.common.VideoManagerConstants;

@@ -23,7 +29,6 @@
import com.genersoft.iot.vmp.gb28181.auth.DigestServerAuthenticationHelper;

import com.genersoft.iot.vmp.gb28181.auth.RegisterLogicHandler;

import com.genersoft.iot.vmp.gb28181.bean.Device;

import com.genersoft.iot.vmp.gb28181.bean.Host;

import com.genersoft.iot.vmp.gb28181.event.EventPublisher;

import com.genersoft.iot.vmp.gb28181.transmit.request.SIPRequestAbstractProcessor;

import com.genersoft.iot.vmp.storager.IVideoManagerStorager;

@@ -39,6 +44,8 @@
 */

public class RegisterRequestProcessor extends SIPRequestAbstractProcessor {



   private Logger logger = LoggerFactory.getLogger(RegisterRequestProcessor.class);



   private SipConfig sipConfig;

   

   private RegisterLogicHandler handler;

@@ -46,17 +53,17 @@
   private IVideoManagerStorager storager;

   

   private EventPublisher publisher;

	
   /***



   /**

    * 收到注册请求 处理

    * 
    * @param request

    *            请求消息

    */ 
     * @param evt

    */

   @Override

   public void process(RequestEvent evt) {

      try {

         System.out.println("收到注册请求，开始处理");

         RequestEventExt evtExt = (RequestEventExt)evt;

         String requestAddress = evtExt.getRemoteIpAddress() + ":" + evtExt.getRemotePort();

         logger.info("[{}] 收到注册请求，开始处理", requestAddress);

         Request request = evt.getRequest();



         Response response = null; 

@@ -75,9 +82,9 @@
         if (authorhead == null || !passwordCorrect) {

            

            if (authorhead == null) {

               System.out.println("未携带授权头 回复401");

               logger.info("[{}] 未携带授权头 回复401", requestAddress);

            } else if (!passwordCorrect) {

               System.out.println("密码错误 回复401");

               logger.info("[{}] 密码错误 回复401", requestAddress);

            }

            response = getMessageFactory().createResponse(Response.UNAUTHORIZED, request);

            new DigestServerAuthenticationHelper().generateChallenge(getHeaderFactory(), response, sipConfig.getSipDomain());

@@ -86,36 +93,41 @@
         else if (passwordCorrect) {

            response = getMessageFactory().createResponse(Response.OK, request);

            // 添加date头

            response.addHeader(getHeaderFactory().createDateHeader(Calendar.getInstance(Locale.ENGLISH)));

            SIPDateHeader dateHeader = new SIPDateHeader();

            // 使用自己修改的

            WvpSipDate wvpSipDate = new WvpSipDate(Calendar.getInstance(Locale.ENGLISH).getTimeInMillis());

            dateHeader.setDate(wvpSipDate);

            response.addHeader(dateHeader);



            ExpiresHeader expiresHeader = (ExpiresHeader) request.getHeader(Expires.NAME);

            // 添加Contact头

            response.addHeader(request.getHeader(ContactHeader.NAME));

            // 添加Expires头

            response.addHeader(request.getExpires());

            

            // 1.获取到通信地址等信息，保存到Redis

            // 获取到通信地址等信息

            FromHeader fromHeader = (FromHeader) request.getHeader(FromHeader.NAME);

            ViaHeader viaHeader = (ViaHeader) request.getHeader(ViaHeader.NAME);

            String received = viaHeader.getReceived();

            int rPort = viaHeader.getRPort();

            // 本地模拟设备 received 为空 rPort 为 -1

            // 解析本地地址替代

            if (StringUtils.isEmpty(received) || rPort == -1) {

               received = viaHeader.getHost();

               rPort = viaHeader.getPort();

            }

            //

            Host host = new Host();

            host.setIp(received);

            host.setPort(rPort);

            host.setAddress(received.concat(":").concat(String.valueOf(rPort)));

            AddressImpl address = (AddressImpl) fromHeader.getAddress();

            SipUri uri = (SipUri) address.getURI();

            String deviceId = uri.getUser();

            device = new Device();

            device.setStreamMode("UDP");

            device.setDeviceId(deviceId);

            device.setHost(host);

            device = storager.queryVideoDevice(deviceId);

            if (device == null) {

               device = new Device();

               device.setStreamMode("UDP");

               device.setDeviceId(deviceId);

            }

            device.setIp(received);

            device.setPort(rPort);

            device.setHostAddress(received.concat(":").concat(String.valueOf(rPort)));

            // 注销成功

            if (expiresHeader != null && expiresHeader.getExpires() == 0) {

               registerFlag = 2;

@@ -137,13 +149,16 @@
         // 注册成功

         // 保存到redis

         // 下发catelog查询目录

         if (registerFlag == 1 && device != null) {

            System.out.println("注册成功! deviceId:" + device.getDeviceId());

         if (registerFlag == 1 ) {

            logger.info("[{}] 注册成功! deviceId:" + device.getDeviceId(), requestAddress);

            device.setRegisterTimeMillis(System.currentTimeMillis());

            storager.updateDevice(device);

            publisher.onlineEventPublish(device.getDeviceId(), VideoManagerConstants.EVENT_ONLINE_REGISTER);



            // 重新注册更新设备和通道，以免设备替换或更新后信息无法更新

            handler.onRegister(device);

         } else if (registerFlag == 2) {

            System.out.println("注销成功! deviceId:" + device.getDeviceId());

            logger.info("[{}] 注销成功! deviceId:" + device.getDeviceId(), requestAddress);

            publisher.outlineEventPublish(device.getDeviceId(), VideoManagerConstants.EVENT_OUTLINE_UNREGISTER);

         }

      } catch (SipException | InvalidArgumentException | NoSuchAlgorithmException | ParseException e) {