添加磁盘空间查询接口

648540858
1 parent 72434a5d
Showing 4 changed files with 81 additions and 0 deletions
src/main/java/top/panll/assist/config/StartConfig.java
src/main/java/top/panll/assist/controller/RecordController.java
src/main/java/top/panll/assist/dto/SpaceInfo.java
src/main/java/top/panll/assist/service/VideoFileService.java
@@ -35,6 +35,9 @@ public class StartConfig implements CommandLineRunner {
     @Override
     public void run(String... args) {
         String record = userSettings.getRecord();
+        if (!record.endsWith(File.separator)) {
+            userSettings.setRecord(userSettings.getRecord() + File.separator);
+        }
         File recordFile = new File(record);
         if (!recordFile.exists() || !recordFile.isDirectory()) {
             logger.error("[userSettings.record]配置错误，请检查路径是否存在");
 package top.panll.assist.controller;
  
+import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -9,6 +10,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.SpaceInfo;
 import top.panll.assist.service.VideoFileService;
 import top.panll.assist.utils.PageInfo;
  
@@ -271,4 +273,19 @@ public class RecordController {
  
         return new ResponseEntity<String>(ret.toString(), HttpStatus.OK);
     }
+
+    /**
+     * 磁盘空间查询
+     * @return
+     */
+    @ResponseBody
+    @GetMapping(value = "/space", produces = "application/json;charset=UTF-8")
+    public ResponseEntity<String> getSpace() {
+        JSONObject ret = new JSONObject();
+        ret.put("code", 0);
+        ret.put("msg", "success");
+        SpaceInfo spaceInfo = videoFileService.getSpaceInfo();
+        ret.put("data", JSON.toJSON(spaceInfo));
+        return new ResponseEntity<>(ret.toString(), HttpStatus.OK);
+    }
 }
+package top.panll.assist.dto;
+
+public class SpaceInfo {
+    private long total;
+    private long free;
+
+    public long getTotal() {
+        return total;
+    }
+
+    public void setTotal(long total) {
+        this.total = total;
+    }
+
+    public long getFree() {
+        return free;
+    }
+
+    public void setFree(long free) {
+        this.free = free;
+    }
+
+}
@@ -11,6 +11,7 @@ import org.springframework.context.annotation.Bean;
 import org.springframework.data.redis.core.StringRedisTemplate;
 import org.springframework.stereotype.Service;
 import org.springframework.util.DigestUtils;
+import top.panll.assist.dto.SpaceInfo;
 import top.panll.assist.utils.RedisUtil;
 import top.panll.assist.dto.MergeOrCutTaskInfo;
 import top.panll.assist.dto.UserSettings;
@@ -72,6 +73,43 @@ public class VideoFileService {
         }
     }
  
+    public SpaceInfo getSpaceInfo(){
+        File recordFile = new File(userSettings.getRecord());
+        SpaceInfo spaceInfo = new SpaceInfo();
+        spaceInfo.setFree(recordFile.getFreeSpace());
+        spaceInfo.setTotal(recordFile.getTotalSpace());
+        return spaceInfo;
+    }
+
+//    public String getPrintSize(long size) {
+//        // 如果字节数少于1024，则直接以B为单位，否则先除于1024，后3位因太少无意义
+//        if (size < 1024) {
+//            return String.valueOf(size) + "B";
+//        } else {
+//            size = size / 1024;
+//        }
+//        // 如果原字节数除于1024之后，少于1024，则可以直接以KB作为单位
+//        // 因为还没有到达要使用另一个单位的时候
+//        // 接下去以此类推
+//        if (size < 1024) {
+//            return String.valueOf(size) + "KB";
+//        } else {
+//            size = size / 1024;
+//        }
+//        if (size < 1024) {
+//            // 因为如果以MB为单位的话，要保留最后1位小数，
+//            // 因此，把此数乘以100之后再取余
+//            size = size * 100;
+//            return String.valueOf((size / 100)) + "."
+//                    + String.valueOf((size % 100)) + "MB";
+//        } else {
+//            // 否则如果要以GB为单位的，先除于1024再作同样的处理
+//            size = size * 100 / 1024;
+//            return String.valueOf((size / 100)) + "."
+//                    + String.valueOf((size % 100)) + "GB";
+//        }
+//    }
+
     public List<File> getStreamList(String app) {
         File appFile = new File(userSettings.getRecord() + File.separator + app);
         if (appFile != null) {