接口优化

zhanghua

2023-03-16 7fc2d4db0603bfe826a8ddb8259ecfd88d30f192

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package com.ycl.util;
 
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.ycl.api.CommonResult;
import com.ycl.dto.v1.Req.FirstReq;
import com.ycl.dto.v1.Req.SecondReq;
import com.ycl.dto.v1.Res.FirstRes;
import com.ycl.utils.MD5Util;
import lombok.SneakyThrows;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.*;
import org.springframework.http.client.ClientHttpResponse;
import org.springframework.http.client.SimpleClientHttpRequestFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.client.DefaultResponseErrorHandler;
import org.springframework.web.client.RestTemplate;
 
import java.net.URI;
 
@Component
public class VideoUtil {
 
    @Value("${videoPoint.url}")
    private String url;
 
    @Value("${videoPoint.userName}")
    private String userName;
 
    @Value("${videoPoint.passWord}")
    private String passWord;
 
    @Value("${videoPoint.ip}")
    private String ip;
 
    /*
     * @Description  subType:类型int，选填。码流类型，0:主码流、1:辅流1、2:辅流2。默认为0主码流。scheme:类型string，选填。协议类型，支持RTSP、FLV_HTTP、HLS三种，默认RTSP。
     * @Param [code, scheme, subType]
     * @return java.lang.String
     **/
    @SneakyThrows
    public String getVideo(String code, String scheme, Integer subType) {
        // System.out.println("开始视频请求：" + code);
        try {
 
            String clientType = "winpc";
            String reqUrl = "/videoService/accounts/authorize";
            String deviceUrl = "/videoService/devicesManager/devices";
//        String monitorUrl = "/videoService/realmonitor/uri?subType=" + subType + "&scheme=" + scheme + "&channelId=";
 
            String monitorUrl = "/videoService/realmonitor/uri?trackId=&encryptedInfo=&userId=&urlType=1&keyIndex=&regionId=&packtype=Private&planId=&subType=" + subType + "&isroute=false&streamType=&locationEnable=false&domainId=&duration=600&scheme=RTSP&dataType=2&extend=&channelId=";
            if (!"RTSP".equals(scheme)) {
                monitorUrl = "/videoService/realmonitor/uri?subType=" + subType + "&scheme=" + scheme + "&channelId=";
            }
            RestTemplate restTemplate = new RestTemplate();
            restTemplate.setErrorHandler(new DefaultResponseErrorHandler() {
                @Override
                public boolean hasError(HttpStatus status) {
                    return super.hasError(status);
                }
 
                @Override
                public void handleError(URI url, HttpMethod method, ClientHttpResponse response) {
 
                }
            });
            SimpleClientHttpRequestFactory simpleClientHttpRequestFactory = new SimpleClientHttpRequestFactory();
            simpleClientHttpRequestFactory.setOutputStreaming(false);
            restTemplate.setRequestFactory(simpleClientHttpRequestFactory);
 
            //请求头
            HttpHeaders httpHeaders = new HttpHeaders();
            httpHeaders.setContentType(MediaType.APPLICATION_JSON);
 
            //第一次权限请求体
            FirstReq firstReq = new FirstReq();
            firstReq.setIpAddress(ip);
            firstReq.setUserName(userName);
            firstReq.setClientType(clientType);
            ObjectMapper objectMapper = new ObjectMapper();
 
            HttpEntity<String> firstEntity = new HttpEntity<>(objectMapper.writeValueAsString(firstReq), httpHeaders);
            ResponseEntity<String> responseEntity = restTemplate.postForEntity(url + reqUrl, firstEntity, String.class);
 
            FirstRes firstRes = objectMapper.readValue(responseEntity.getBody(), FirstRes.class);
 
            String signature = MD5Util.md5Encrypt(passWord);
            signature = MD5Util.md5Encrypt(userName + signature);
            signature = MD5Util.md5Encrypt(signature);
            signature = MD5Util.md5Encrypt(userName + ":" + firstRes.getRealm() + ":" + signature);
            signature = MD5Util.md5Encrypt(signature + ":" + firstRes.getRandomKey());
 
            SecondReq secondReq = new SecondReq();
            secondReq.setIpAddress(ip);
            secondReq.setEncryptType(firstRes.getEncryptType());
            secondReq.setSignature(signature);
            secondReq.setRandomKey(firstRes.getRandomKey());
            secondReq.setClientType(clientType);
            secondReq.setUserName(userName);
 
            HttpEntity<String> secondEntity = new HttpEntity<>(objectMapper.writeValueAsString(secondReq), httpHeaders);
            ResponseEntity<String> secondResEntity = restTemplate.postForEntity(url + reqUrl, secondEntity, String.class);
            HttpHeaders deviceHeaders = new HttpHeaders();
            deviceHeaders.add("X-Subject-Token", objectMapper.readTree(secondResEntity.getBody()).get("token").textValue());
            HttpEntity<Object> deviceEntity = new HttpEntity<>(deviceHeaders);
            // System.out.println("视频设备请求地址：" + url + deviceUrl + "/" + code);
 
            ResponseEntity<String> exchange = restTemplate.exchange(url + deviceUrl + "/" + code, HttpMethod.GET, deviceEntity, String.class);
            // System.out.println("开始视频身份请求结果：" + exchange.getBody());
            String channelId = objectMapper.readTree(objectMapper.readTree(objectMapper.readTree(exchange.getBody()).get("encoderUnit").toString()).get("channels").get(0).toString()).get("channelId").textValue();
            // System.out.println("视频请求地址：" + url + monitorUrl + channelId);
            ResponseEntity<String> monitorReq = restTemplate.exchange(url + monitorUrl + channelId, HttpMethod.GET, deviceEntity, String.class);
            // System.out.println("视频请求返回参数：" + monitorReq.getBody());
            return objectMapper.readTree(monitorReq.getBody()).get("url").textValue();
 
        } catch (Exception ex) {
            return "获取地址失败，请检查code值";
        }
    }
}