64850858
2021-07-16 89a9ab4534f10a224f70e546db838423e84a1965
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
package com.genersoft.iot.vmp.vmanager.record;
 
import com.genersoft.iot.vmp.conf.MediaConfig;
import com.genersoft.iot.vmp.media.zlm.ZLMServerConfig;
import com.genersoft.iot.vmp.media.zlm.dto.IMediaServerItem;
import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem;
import com.genersoft.iot.vmp.service.IMediaServerService;
import com.genersoft.iot.vmp.storager.IRedisCatchStorage;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.client.HttpClientErrorException;
import org.springframework.web.client.RestTemplate;
 
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
 
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
 
@RestController
@RequestMapping("/record_proxy")
public class RecoderProxyController {
 
 
    // private final static Logger logger = LoggerFactory.getLogger(ZLMHTTPProxyController.class);
 
    @Autowired
    private IRedisCatchStorage redisCatchStorage;
    @Autowired
    private IMediaServerService mediaServerService;
 
    @Autowired
    private MediaConfig mediaConfig;
 
    @ResponseBody
    @RequestMapping(value = "/**/**/**", produces = "application/json;charset=UTF-8")
    public Object proxy(HttpServletRequest request, HttpServletResponse response) throws UnsupportedEncodingException{
 
 
        String baseRequestURI = request.getRequestURI();
        String[] split = baseRequestURI.split("/");
        if (split.length <= 2) {
            response.setStatus(HttpStatus.NOT_FOUND.value());
            return null;
        }
        String mediaId = split[2];
        if (StringUtils.isEmpty(mediaId)){
            response.setStatus(HttpStatus.BAD_REQUEST.value());
            return null;
        }
        // 后续改为根据Id获取对应的ZLM
        IMediaServerItem mediaInfo = mediaServerService.getOne(mediaId);
        if (mediaInfo == null) {
            response.setStatus(HttpStatus.NOT_FOUND.value());
            return null;
        }
        String requestURI = String.format("http://%s:%s%s?%s",
                mediaInfo.getSdpIp(),
                mediaConfig.getRecordAssistPort(),
                baseRequestURI.substring(baseRequestURI.indexOf(mediaId) + mediaId.length()),
                URLDecoder.decode(request.getQueryString(), "UTF-8")
        );
        // 发送请求
        RestTemplate restTemplate = new RestTemplate();
        //将指定的url返回的参数自动封装到自定义好的对应类对象中
        Object result = null;
        try {
            result = restTemplate.getForObject(requestURI,Object.class);
 
        }catch (HttpClientErrorException httpClientErrorException) {
            response.setStatus(httpClientErrorException.getStatusCode().value());
        }
        return result;
    }
}