添加录像收藏相关接口

648540858
1 parent adbac9f9
Showing 4 changed files with 144 additions and 1 deletions
src/main/java/top/panll/assist/controller/RecordController.java
src/main/java/top/panll/assist/dto/MergeOrCutTaskInfo.java
src/main/java/top/panll/assist/dto/SignInfo.java
src/main/java/top/panll/assist/service/VideoFileService.java
@@ -14,6 +14,7 @@ import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
 import top.panll.assist.controller.bean.WVPResult;
 import top.panll.assist.dto.MergeOrCutTaskInfo;
+import top.panll.assist.dto.SignInfo;
 import top.panll.assist.dto.SpaceInfo;
 import top.panll.assist.service.VideoFileService;
 import top.panll.assist.utils.PageInfo;
@@ -290,6 +291,70 @@ public class RecordController {
     }
     /**
+     * 收藏录像（被收藏的录像不会被清理任务清理）
+     */
+    @ApiOperation("收藏录像（被收藏的录像不会被清理任务清理）")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name="app", value = "应用名", required = true, dataTypeClass = String.class),
+            @ApiImplicitParam(name="stream", value = "流ID", required = true, dataTypeClass = String.class),
+    })
+    @GetMapping(value = "/file/collection/add")
+    @ResponseBody
+    public WVPResult<String> collection(
+            @RequestParam(required = true) String app,
+            @RequestParam(required = true) String stream){
+
+        boolean collectionResult = videoFileService.collection(app, stream);
+        WVPResult<String> result = new WVPResult<>();
+        result.setCode(0);
+        result.setMsg(collectionResult ?"success":"error");
+        return result;
+    }
+
+    /**
+     * 移除收藏录像
+     */
+    @ApiOperation("移除收藏录像")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name="app", value = "应用名", required = true, dataTypeClass = String.class),
+            @ApiImplicitParam(name="stream", value = "流ID", required = true, dataTypeClass = String.class),
+    })
+    @GetMapping(value = "/file/collection/remove")
+    @ResponseBody
+    public WVPResult<String> removeCollection(
+            @RequestParam(required = true) String app,
+            @RequestParam(required = true) String stream){
+
+        boolean collectionResult = videoFileService.removeCollection(app, stream);
+        WVPResult<String> result = new WVPResult<>();
+        result.setCode(0);
+        result.setMsg(collectionResult ?"success":"error");
+        return result;
+    }
+
+    /**
+     * 收藏录像列表
+     */
+    @ApiOperation("收藏录像列表")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name="app", value = "应用名", required = false, dataTypeClass = String.class),
+            @ApiImplicitParam(name="stream", value = "流ID", required = false, dataTypeClass = String.class),
+    })
+    @GetMapping(value = "/file/collection/list")
+    @ResponseBody
+    public WVPResult<List<SignInfo>> collectionList(
+            @RequestParam(required = false) String app,
+            @RequestParam(required = false) String stream){
+
+        List<SignInfo> signInfos = videoFileService.getCollectionList(app, stream);
+        WVPResult<List<SignInfo>> result = new WVPResult<>();
+        result.setCode(0);
+        result.setMsg(signInfos != null ?"success":"error");
+        result.setData(signInfos);
+        return result;
+    }
+
+    /**
      * 中止视频裁剪合并任务列表
      */
     @ApiOperation("中止视频裁剪合并任务列表(暂不支持)")
 package top.panll.assist.dto;
-import java.util.Date;
 public class MergeOrCutTaskInfo {
     private String id;
+package top.panll.assist.dto;
+
+public class SignInfo {
+    private String app;
+    private String stream;
+
+    public String getApp() {
+        return app;
+    }
+
+    public void setApp(String app) {
+        this.app = app;
+    }
+
+    public String getStream() {
+        return stream;
+    }
+
+    public void setStream(String stream) {
+        this.stream = stream;
+    }
+}
@@ -3,12 +3,14 @@ package top.panll.assist.service;
 import net.bramp.ffmpeg.FFprobe;
 import net.bramp.ffmpeg.probe.FFmpegProbeResult;
 import net.bramp.ffmpeg.progress.Progress;
+import org.apache.commons.io.FileUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.redis.core.RedisTemplate;
 import org.springframework.stereotype.Service;
 import org.springframework.util.DigestUtils;
+import top.panll.assist.dto.SignInfo;
 import top.panll.assist.dto.SpaceInfo;
 import top.panll.assist.utils.RedisUtil;
 import top.panll.assist.dto.MergeOrCutTaskInfo;
@@ -497,4 +499,59 @@ public class VideoFileService {
 //        }
         return false;
     }
+
+    public boolean collection(String app, String stream) {
+        File streamFile = new File(userSettings.getRecord() + File.separator + app + File.separator + stream);
+        boolean result = false;
+        if (streamFile.exists() && streamFile.isDirectory() && streamFile.canWrite()) {
+            File signFile = new File(streamFile.getAbsolutePath() + File.separator + "sign");
+            try {
+                result = signFile.createNewFile();
+            } catch (IOException e) {
+                logger.error("[收藏文件]失败，{}/{}", app, stream);
+            }
+        }
+        return result;
+    }
+
+    public boolean removeCollection(String app, String stream) {
+        File signFile = new File(userSettings.getRecord() + File.separator + app + File.separator + stream + File.separator + "sign");
+        boolean result = false;
+        if (signFile.exists() && signFile.isFile()) {
+            result = signFile.delete();
+        }
+        return result;
+    }
+
+    public List<SignInfo> getCollectionList(String app, String stream) {
+        List<File> appList = this.getAppList(true);
+        List<SignInfo> result = new ArrayList<>();
+        if (appList.size() > 0) {
+            for (File appFile : appList) {
+                if (app != null) {
+                    if (!app.equals(appFile.getName())) {
+                        continue;
+                    }
+                }
+                List<File> streamList = getStreamList(appFile, true);
+                if (streamList.size() > 0) {
+                    for (File streamFile : streamList) {
+                        if (stream != null) {
+                            if (!stream.equals(streamFile.getName())) {
+                                continue;
+                            }
+                        }
+                        File signFile = new File(streamFile.getAbsolutePath() + File.separator + "sign");
+                        if (signFile.exists()) {
+                            SignInfo signInfo = new SignInfo();
+                            signInfo.setApp(appFile.getName());
+                            signInfo.setStream(streamFile.getName());
+                            result.add(signInfo);
+                        }
+                    }
+                }
+            }
+        }
+        return result;
+    }
 }