From 5d901b5e3f033e8b04e53420d68626cbd87431c8 Mon Sep 17 00:00:00 2001 From: 648540858 <648540858@qq.com> Date: 星期五, 06 五月 2022 10:12:34 +0800 Subject: [PATCH] 使用阿里代码规范。规范代码写法 --- src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMRESTfulUtils.java | 72 ++++++++++++++++++++++++++++++----- 1 files changed, 61 insertions(+), 11 deletions(-) diff --git a/src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMRESTfulUtils.java b/src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMRESTfulUtils.java index 71bb317..4b2bf48 100644 --- a/src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMRESTfulUtils.java +++ b/src/main/java/com/genersoft/iot/vmp/media/zlm/ZLMRESTfulUtils.java @@ -4,6 +4,7 @@ import com.alibaba.fastjson.JSONObject; import com.genersoft.iot.vmp.media.zlm.dto.MediaServerItem; import okhttp3.*; +import okhttp3.logging.HttpLoggingInterceptor; import org.jetbrains.annotations.NotNull; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -14,6 +15,7 @@ import java.util.HashMap; import java.util.Map; import java.util.Objects; +import java.util.concurrent.TimeUnit; @Component public class ZLMRESTfulUtils { @@ -24,11 +26,28 @@ void run(JSONObject response); } + private OkHttpClient getClient(){ + OkHttpClient.Builder httpClientBuilder = new OkHttpClient.Builder(); + if (logger.isDebugEnabled()) { + HttpLoggingInterceptor logging = new HttpLoggingInterceptor(message -> { + logger.debug("http璇锋眰鍙傛暟锛�" + message); + }); + logging.setLevel(HttpLoggingInterceptor.Level.BASIC); + // OkHttp閫茶娣诲姞鏀旀埅鍣╨oggingInterceptor + httpClientBuilder.addInterceptor(logging); + } + return httpClientBuilder.build(); + } + + public JSONObject sendPost(MediaServerItem mediaServerItem, String api, Map<String, Object> param, RequestCallback callback) { - OkHttpClient client = new OkHttpClient(); + OkHttpClient client = getClient(); + + if (mediaServerItem == null) { + return null; + } String url = String.format("http://%s:%s/index/api/%s", mediaServerItem.getIp(), mediaServerItem.getHttpPort(), api); JSONObject responseJSON = null; - logger.debug(url); FormBody.Builder builder = new FormBody.Builder(); builder.add("secret",mediaServerItem.getSecret()); @@ -50,10 +69,14 @@ try { Response response = client.newCall(request).execute(); if (response.isSuccessful()) { - String responseStr = response.body().string(); - if (responseStr != null) { + ResponseBody responseBody = response.body(); + if (responseBody != null) { + String responseStr = responseBody.string(); responseJSON = JSON.parseObject(responseStr); } + }else { + response.close(); + Objects.requireNonNull(response.body()).close(); } } catch (ConnectException e) { logger.error(String.format("杩炴帴ZLM澶辫触: %s, %s", e.getCause().getMessage(), e.getMessage())); @@ -73,6 +96,10 @@ } catch (IOException e) { logger.error(String.format("[ %s ]璇锋眰澶辫触: %s", url, e.getMessage())); } + + }else { + response.close(); + Objects.requireNonNull(response.body()).close(); } } @@ -92,7 +119,11 @@ public void sendGetForImg(MediaServerItem mediaServerItem, String api, Map<String, Object> params, String targetPath, String fileName) { String url = String.format("http://%s:%s/index/api/%s", mediaServerItem.getIp(), mediaServerItem.getHttpPort(), api); logger.debug(url); - HttpUrl.Builder httpBuilder = HttpUrl.parse(url).newBuilder(); + HttpUrl parseUrl = HttpUrl.parse(url); + if (parseUrl == null) { + return; + } + HttpUrl.Builder httpBuilder = parseUrl.newBuilder(); httpBuilder.addQueryParameter("secret", mediaServerItem.getSecret()); if (params != null) { @@ -115,16 +146,20 @@ if (targetPath != null) { File snapFolder = new File(targetPath); if (!snapFolder.exists()) { - snapFolder.mkdirs(); + if (!snapFolder.mkdirs()) { + logger.warn("{}璺緞鍒涘缓澶辫触", snapFolder.getAbsolutePath()); + } + } File snapFile = new File(targetPath + "/" + fileName); FileOutputStream outStream = new FileOutputStream(snapFile); - outStream.write(response.body().bytes()); + + outStream.write(Objects.requireNonNull(response.body()).bytes()); outStream.close(); } else { logger.error(String.format("[ %s ]璇锋眰澶辫触: %s %s", url, response.code(), response.message())); } - response.body().close(); + Objects.requireNonNull(response.body()).close(); } else { logger.error(String.format("[ %s ]璇锋眰澶辫触: %s %s", url, response.code(), response.message())); } @@ -138,9 +173,15 @@ public JSONObject getMediaList(MediaServerItem mediaServerItem, String app, String stream, String schema, RequestCallback callback){ Map<String, Object> param = new HashMap<>(); - if (app != null) param.put("app",app); - if (stream != null) param.put("stream",stream); - if (schema != null) param.put("schema",schema); + if (app != null) { + param.put("app",app); + } + if (stream != null) { + param.put("stream",stream); + } + if (schema != null) { + param.put("schema",schema); + } param.put("vhost","__defaultVhost__"); return sendPost(mediaServerItem, "getMediaList",param, callback); } @@ -216,6 +257,10 @@ return sendPost(mediaServerItem, "stopSendRtp",param, null); } + public JSONObject restartServer(MediaServerItem mediaServerItem) { + return sendPost(mediaServerItem, "restartServer",null, null); + } + public JSONObject addStreamProxy(MediaServerItem mediaServerItem, String app, String stream, String url, boolean enable_hls, boolean enable_mp4, String rtp_type) { Map<String, Object> param = new HashMap<>(); param.put("vhost", "__defaultVhost__"); @@ -224,6 +269,11 @@ param.put("url", url); param.put("enable_hls", enable_hls?1:0); param.put("enable_mp4", enable_mp4?1:0); + param.put("enable_rtmp", 1); + param.put("enable_fmp4", 1); + param.put("enable_audio", 1); + param.put("enable_rtsp", 1); + param.put("add_mute_audio", 1); param.put("rtp_type", rtp_type); return sendPost(mediaServerItem, "addStreamProxy",param, null); } -- Gitblit v1.8.0