| | |
| | | import org.apache.http.HttpHost; |
| | | import org.apache.http.HttpRequest; |
| | | import org.apache.http.HttpResponse; |
| | | import org.junit.jupiter.api.Order; |
| | | import org.mitre.dsmiley.httpproxy.ProxyServlet; |
| | | import org.slf4j.Logger; |
| | | import org.slf4j.LoggerFactory; |
| | |
| | | import org.springframework.context.annotation.Bean; |
| | | import org.springframework.context.annotation.Configuration; |
| | | import org.springframework.util.ObjectUtils; |
| | | import org.springframework.util.StringUtils; |
| | | |
| | | import javax.servlet.ServletException; |
| | | import javax.servlet.http.HttpServletRequest; |
| | | import javax.servlet.http.HttpServletResponse; |
| | | import java.io.IOException; |
| | | import java.net.ConnectException; |
| | | |
| | |
| | | */ |
| | | @SuppressWarnings(value = {"rawtypes", "unchecked"}) |
| | | @Configuration |
| | | @Order(1) |
| | | public class ProxyServletConfig { |
| | | |
| | | private final static Logger logger = LoggerFactory.getLogger(ProxyServletConfig.class); |
| | |
| | | } catch (IOException ioException) { |
| | | if (ioException instanceof ConnectException) { |
| | | logger.error("zlm 连接失败"); |
| | | } else if (ioException instanceof ClientAbortException) { |
| | | logger.error("zlm: 用户已中断连接,代理终止"); |
| | | } else { |
| | | } else { |
| | | logger.error("zlm 代理失败: ", e); |
| | | } |
| | | } catch (RuntimeException exception){ |
| | |
| | | MediaServerItem getMediaInfoByUri(String uri){ |
| | | String[] split = uri.split("/"); |
| | | String mediaServerId = split[2]; |
| | | if ("default".equals(mediaServerId)) { |
| | | if ("default".equalsIgnoreCase(mediaServerId)) { |
| | | return mediaServerService.getDefaultMediaServer(); |
| | | }else { |
| | | return mediaServerService.getOne(mediaServerId); |
| | |
| | | protected String rewriteQueryStringFromRequest(HttpServletRequest servletRequest, String queryString) { |
| | | String queryStr = super.rewriteQueryStringFromRequest(servletRequest, queryString); |
| | | MediaServerItem mediaInfo = getMediaInfoByUri(servletRequest.getRequestURI()); |
| | | String remoteHost = String.format("http://%s:%s", mediaInfo.getIp(), mediaInfo.getHttpPort()); |
| | | if (mediaInfo != null) { |
| | | if (!ObjectUtils.isEmpty(queryStr)) { |
| | | queryStr += "&remoteHost=" + remoteHost; |
| | | }else { |
| | | queryStr = "remoteHost=" + remoteHost; |
| | | } |
| | | if (mediaInfo == null) { |
| | | return null; |
| | | } |
| | | String remoteHost = String.format("http://%s:%s", mediaInfo.getStreamIp(), mediaInfo.getRecordAssistPort()); |
| | | if (!ObjectUtils.isEmpty(queryStr)) { |
| | | queryStr += "&remoteHost=" + remoteHost; |
| | | }else { |
| | | queryStr = "remoteHost=" + remoteHost; |
| | | } |
| | | return queryStr; |
| | | } |
| | |
| | | } catch (IOException ioException) { |
| | | if (ioException instanceof ConnectException) { |
| | | logger.error("录像服务 连接失败"); |
| | | } else if (ioException instanceof ClientAbortException) { |
| | | logger.error("录像服务:用户已中断连接,代理终止"); |
| | | } else { |
| | | }else if (ioException instanceof ClientAbortException) { |
| | | /** |
| | | * TODO 使用这个代理库实现代理在遇到代理视频文件时,如果是206结果,会遇到报错蛋市目前功能正常, |
| | | * TODO 暂时去除异常处理。后续使用其他代理框架修改测试 |
| | | */ |
| | | |
| | | }else { |
| | | logger.error("录像服务 代理失败: ", e); |
| | | } |
| | | } catch (RuntimeException exception){ |
| | |
| | | MediaServerItem getMediaInfoByUri(String uri){ |
| | | String[] split = uri.split("/"); |
| | | String mediaServerId = split[2]; |
| | | if ("default".equals(mediaServerId)) { |
| | | if ("default".equalsIgnoreCase(mediaServerId)) { |
| | | return mediaServerService.getDefaultMediaServer(); |
| | | }else { |
| | | return mediaServerService.getOne(mediaServerId); |