📦 project api config

lunasaw · Dec 3, 2023 · db63180 · db63180
1 parent 5713a1c
commit db63180
Show file tree

Hide file tree

Showing 13 changed files with 469 additions and 47 deletions.
diff --git a/README.md b/README.md
@@ -8,7 +8,9 @@
 ZLMediaKit的spring-boot starter, 对[ZLMediaKit](https://github.com/ZLMediaKit/ZLMediaKit)的API进行了封装，方便使用。支持集群化管理。
 
 [文档链接](https://github.com/lunasaw/zlm-spring-boot-starter/blob/master/zlm-api.md)
+
 [API文档](https://lunasaw.github.io/zlm-spring-boot-starter/)
+![img.png](images/img.png)
 
 > 引入spring-boot-starter
 

diff --git a/images/img.png b/images/img.png
diff --git a/src/main/java/io/github/lunasaw/zlm/api/ZlmRestService.java b/src/main/java/io/github/lunasaw/zlm/api/ZlmRestService.java
@@ -1,22 +1,34 @@
 package io.github.lunasaw.zlm.api;
 
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Optional;
+
+import io.github.lunasaw.zlm.entity.rtp.*;
+import org.apache.commons.lang3.StringUtils;
+import org.apache.hc.core5.http.ClassicHttpResponse;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
 import com.alibaba.fastjson2.JSON;
 import com.alibaba.fastjson2.TypeReference;
 import com.google.common.collect.Maps;
 import com.luna.common.check.Assert;
+import com.luna.common.file.FileTools;
 import com.luna.common.net.HttpUtils;
+
 import io.github.lunasaw.zlm.config.ZlmProperties;
 import io.github.lunasaw.zlm.constant.ApiConstants;
 import io.github.lunasaw.zlm.entity.*;
 import io.github.lunasaw.zlm.entity.req.MediaReq;
-import org.apache.commons.lang3.StringUtils;
-import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.stereotype.Service;
-
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
-import java.util.Optional;
+import io.github.lunasaw.zlm.entity.req.RecordReq;
+import io.github.lunasaw.zlm.entity.req.SnapshotReq;
 
 /**
  * @author luna
@@ -313,76 +325,111 @@ public static ServerResponse<MediaInfo> getMediaInfo(String host, String secret,
         });
     }
 
+    public static ServerResponse<Mp4RecordFile> getMp4RecordFile(String host, String secret, RecordReq recordReq) {
+        return getMp4RecordFile(host, secret, recordReq.toMap());
+    }
+
     /**
      * 获取流信息
      */
-    public static ServerResponse getMp4RecordFile(String host, String secret, Map<String, String> params) {
+    public static ServerResponse<Mp4RecordFile> getMp4RecordFile(String host, String secret, Map<String, String> params) {
         String s = doApi(host, secret, ApiConstants.GET_MP4_RECORD_FILE, params);
-        return JSON.parseObject(s, new TypeReference<ServerResponse>() {
+        return JSON.parseObject(s, new TypeReference<ServerResponse<Mp4RecordFile>>() {
         });
     }
 
     /**
      * 删除录像文件夹
      */
-    public static ServerResponse deleteRecordDirectory(String host, String secret, Map<String, String> params) {
+    public static DeleteRecordDirectory deleteRecordDirectory(String host, String secret, Map<String, String> params) {
         String s = doApi(host, secret, ApiConstants.DELETE_RECORD_DIRECTORY, params);
-        return JSON.parseObject(s, new TypeReference<ServerResponse>() {
+        return JSON.parseObject(s, new TypeReference<DeleteRecordDirectory>() {
         });
     }
 
+    public static ServerResponse<String> startRecord(String host, String secret, RecordReq recordReq) {
+        return startRecord(host, secret, recordReq.getSaveMp4Map());
+    }
+
     /**
      * 开始录制
      */
-    public static ServerResponse startRecord(String host, String secret, Map<String, String> params) {
+    public static ServerResponse<String> startRecord(String host, String secret, Map<String, String> params) {
         String s = doApi(host, secret, ApiConstants.START_RECORD, params);
-        return JSON.parseObject(s, new TypeReference<ServerResponse>() {
+        return JSON.parseObject(s, new TypeReference<ServerResponse<String>>() {
         });
     }
 
+    public static ServerResponse<String> setRecordSpeed(String host, String secret, RecordReq recordReq) {
+        return setRecordSpeed(host, secret, recordReq.getRecordSpeedMap());
+    }
+
     /**
      * 设置录像速度
      */
-    public static ServerResponse setRecordSpeed(String host, String secret, Map<String, String> params) {
+    public static ServerResponse<String> setRecordSpeed(String host, String secret, Map<String, String> params) {
         String s = doApi(host, secret, ApiConstants.SET_RECORD_SPEED, params);
-        return JSON.parseObject(s, new TypeReference<ServerResponse>() {
+        return JSON.parseObject(s, new TypeReference<ServerResponse<String>>() {
         });
     }
 
+    public static ServerResponse<String> seekRecordStamp(String host, String secret, RecordReq recordReq) {
+        return seekRecordStamp(host, secret, recordReq.getSeekRecordStampMap());
+    }
+
     /**
      * 设置录像流播放位置
      */
-    public static ServerResponse seekRecordStamp(String host, String secret, Map<String, String> params) {
+    public static ServerResponse<String> seekRecordStamp(String host, String secret, Map<String, String> params) {
         String s = doApi(host, secret, ApiConstants.SEEK_RECORD_STAMP, params);
-        return JSON.parseObject(s, new TypeReference<ServerResponse>() {
+        return JSON.parseObject(s, new TypeReference<ServerResponse<String>>() {
         });
     }
 
+    public static ServerResponse<String> stopRecord(String host, String secret, RecordReq recordReq) {
+        return stopRecord(host, secret, recordReq.getStopRecordMap());
+    }
+
     /**
      * 停止录制
      */
-    public static ServerResponse stopRecord(String host, String secret, Map<String, String> params) {
+    public static ServerResponse<String> stopRecord(String host, String secret, Map<String, String> params) {
         String s = doApi(host, secret, ApiConstants.STOP_RECORD, params);
-        return JSON.parseObject(s, new TypeReference<ServerResponse>() {
+        return JSON.parseObject(s, new TypeReference<ServerResponse<String>>() {
         });
     }
 
+    public static ServerResponse<String> isRecording(String host, String secret, RecordReq recordReq) {
+        return isRecording(host, secret, recordReq.getIsRecordingMap());
+    }
+
     /**
      * 是否正在录制
      */
-    public static ServerResponse isRecording(String host, String secret, Map<String, String> params) {
+    public static ServerResponse<String> isRecording(String host, String secret, Map<String, String> params) {
         String s = doApi(host, secret, ApiConstants.IS_RECORDING, params);
-        return JSON.parseObject(s, new TypeReference<ServerResponse>() {
+        return JSON.parseObject(s, new TypeReference<ServerResponse<String>>() {
         });
     }
 
+    public static String getSnap(String host, String secret, SnapshotReq snapshotReq) {
+        return getSnap(host, secret, snapshotReq.toMap(), snapshotReq.getSavePath());
+    }
+
     /**
      * 获取截图
      */
-    public static ServerResponse getSnap(String host, String secret, Map<String, String> params) {
-        String s = doApi(host, secret, ApiConstants.GET_SNAP, params);
-        return JSON.parseObject(s, new TypeReference<ServerResponse>() {
-        });
+    public static String getSnap(String host, String secret, Map<String, String> params, String file) {
+        Path path = Paths.get(file);
+        boolean exists = Files.exists(path);
+        if (!exists) {
+            try {
+                Files.createFile(path);
+            } catch (IOException e) {
+                return null;
+            }
+        }
+        return doApiImg(host, secret, ApiConstants.GET_SNAP, params, path.toFile());
     }
 
     /**
@@ -396,46 +443,64 @@ public static ServerResponse getMp4RecordSummary(String host, String secret, Map
 
     /**
      * 获取rtp推流信息
+     * 
+     * @param streamId RTP的ssrc，16进制字符串或者是流的id(openRtpServer接口指定)
      */
-    public static ServerResponse getRtpInfo(String host, String secret, Map<String, String> params) {
+    public static RtpInfo getRtpInfo(String host, String secret, String streamId) {
+        Map<String, String> params = new HashMap<>();
+        params.put("stream_id", streamId);
         String s = doApi(host, secret, ApiConstants.GET_RTP_INFO, params);
-        return JSON.parseObject(s, new TypeReference<ServerResponse>() {
+        return JSON.parseObject(s, new TypeReference<RtpInfo>() {
         });
     }
 
+    public static OpenRtpServerResult openRtpServer(String host, String secret, OpenRtpServerReq req) {
+        return openRtpServer(host, secret, req.toMap());
+    }
+
     /**
      * 创建RTP服务器
      */
-    public static ServerResponse openRtpServer(String host, String secret, Map<String, String> params) {
+    public static OpenRtpServerResult openRtpServer(String host, String secret, Map<String, String> params) {
         String s = doApi(host, secret, ApiConstants.OPEN_RTP_SERVER, params);
-        return JSON.parseObject(s, new TypeReference<ServerResponse>() {
+        return JSON.parseObject(s, new TypeReference<OpenRtpServerResult>() {
         });
     }
 
+    public static OpenRtpServerResult openRtpServerMultiplex(String host, String secret, OpenRtpServerReq req) {
+        return openRtpServerMultiplex(host, secret, req.toMap());
+    }
+
     /**
      * 创建多路复用RTP服务器
      */
-    public static ServerResponse openRtpServerMultiplex(String host, String secret, Map<String, String> params) {
+    public static OpenRtpServerResult openRtpServerMultiplex(String host, String secret, Map<String, String> params) {
         String s = doApi(host, secret, ApiConstants.OPEN_RTP_SERVER_MULTIPLEX, params);
-        return JSON.parseObject(s, new TypeReference<ServerResponse>() {
+        return JSON.parseObject(s, new TypeReference<OpenRtpServerResult>() {
         });
     }
 
+    public static OpenRtpServerResult connectRtpServer(String host, String secret, ConnectRtpServerReq req) {
+        return connectRtpServer(host, secret, req.toMap());
+    }
+
     /**
      * 连接RTP服务器
      */
-    public static ServerResponse connectRtpServer(String host, String secret, Map<String, String> params) {
+    public static OpenRtpServerResult connectRtpServer(String host, String secret, Map<String, String> params) {
         String s = doApi(host, secret, ApiConstants.CONNECT_RTP_SERVER, params);
-        return JSON.parseObject(s, new TypeReference<ServerResponse>() {
+        return JSON.parseObject(s, new TypeReference<OpenRtpServerResult>() {
         });
     }
 
     /**
      * 关闭RTP服务器
      */
-    public static ServerResponse closeRtpServer(String host, String secret, Map<String, String> params) {
+    public static CloseRtpServerResult closeRtpServer(String host, String secret, String streamId) {
+        Map<String, String> params = new HashMap<>();
+        params.put("stream_id", streamId);
         String s = doApi(host, secret, ApiConstants.CLOSE_RTP_SERVER, params);
-        return JSON.parseObject(s, new TypeReference<ServerResponse>() {
+        return JSON.parseObject(s, new TypeReference<CloseRtpServerResult>() {
         });
     }
 
@@ -565,14 +630,6 @@ public static ServerResponse loadMP4File(String host, String secret, Map<String,
         });
     }
 
-
-    public static void main(String[] args) {
-//        ServerResponse zlm = getServerConfig(URL, SECRET);
-//        System.out.println(zlm);
-        ServerResponse<List<String>> version = getApiList(URL, SECRET, new HashMap<>());
-        System.out.println(version);
-    }
-
     public static String doApi(String host, String secret, String path, Map<String, String> params) {
         Assert.notNull(host, "host is null");
         Assert.notNull(path, "api is null");
@@ -583,4 +640,22 @@ public static String doApi(String host, String secret, String path, Map<String,
         return HttpUtils.doPostHander(host, path, new HashMap<>(), params, StringUtils.EMPTY);
     }
 
+    public static String doApiImg(String host, String secret, String path, Map<String, String> params, File file) {
+        Assert.notNull(host, "host is null");
+        Assert.notNull(path, "api is null");
+        Assert.notNull(secret, "secret is null");
+
+        params = Optional.ofNullable(params).orElse(new HashMap<>());
+        params.put("secret", secret);
+        ClassicHttpResponse classicHttpResponse = HttpUtils.doPost(host, path, new HashMap<>(), params, StringUtils.EMPTY);
+        if (classicHttpResponse == null) {
+            return null;
+        }
+        byte[] bytes = HttpUtils.checkResponseStreamAndGetResult(classicHttpResponse);
+        if (bytes == null) {
+            return null;
+        }
+        FileTools.write(bytes, file.getAbsolutePath());
+        return file.getAbsolutePath();
+    }
 }
diff --git a/src/main/java/io/github/lunasaw/zlm/entity/DeleteRecordDirectory.java b/src/main/java/io/github/lunasaw/zlm/entity/DeleteRecordDirectory.java
@@ -0,0 +1,15 @@
+package io.github.lunasaw.zlm.entity;
+
+import lombok.Data;
+
+/**
+ * @author weidian
+ * @version 1.0
+ * @date 2023/12/2
+ * @description:
+ */
+@Data
+public class DeleteRecordDirectory extends ServerResponse<String> {
+    private String path;
+
+}
diff --git a/src/main/java/io/github/lunasaw/zlm/entity/Mp4RecordFile.java b/src/main/java/io/github/lunasaw/zlm/entity/Mp4RecordFile.java
@@ -0,0 +1,20 @@
+package io.github.lunasaw.zlm.entity;
+
+import lombok.Data;
+
+import java.util.List;
+
+/**
+ * @author weidian
+ * @version 1.0
+ * @date 2023/12/3
+ * @description:
+ */
+@Data
+public class Mp4RecordFile {
+
+    private List<String> paths;
+
+    private String       rootPath;
+
+}
diff --git a/src/main/java/io/github/lunasaw/zlm/entity/req/MediaReq.java b/src/main/java/io/github/lunasaw/zlm/entity/req/MediaReq.java
@@ -1,11 +1,10 @@
 package io.github.lunasaw.zlm.entity.req;
 
-import com.alibaba.fastjson.JSONObject;
-import lombok.Data;
-
 import java.util.HashMap;
 import java.util.Map;
 
+import lombok.Data;
+
 /**
  * @author weidian
  * @version 1.0