Jelajahi Sumber

增加列举oss文件大小

dsx 2 tahun lalu
induk
melakukan
5c1cd8bf23
1 mengubah file dengan 21 tambahan dan 6 penghapusan
  1. 21 6
      src/main/java/com/fdkankan/job/job/CountSizeOssHandler.java

+ 21 - 6
src/main/java/com/fdkankan/job/job/CountSizeOssHandler.java

@@ -26,9 +26,7 @@ import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Component;
 
 import java.io.File;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.List;
+import java.util.*;
 
 /**
  * <p>
@@ -57,12 +55,29 @@ public class CountSizeOssHandler {
             dirList = Arrays.asList(dirs.split(","));
         }
 
-        ObjectListing objectListing = ossClient.listObjects("4dkankan", "scene/");
+        Map<String, Long> map = new HashMap<>();
+        ObjectListing objectListing = ossClient.listObjects("4dkankan", dirs);
         for (OSSObjectSummary objectSummary : objectListing.getObjectSummaries()) {
-            log.info(objectSummary.getKey());
-            log.info(String.valueOf(objectSummary.getSize()));//输出每个object文件的大小
+            String key = objectSummary.getKey();
+            String[] split = key.split("/");
+            key = split[1];
+            Long size = map.get(key);
+            if(Objects.isNull(size)){
+                size = 0L;
+            }
+            size += objectSummary.getSize();
+            map.put(key, size);
         }
 
+        Set<String> strings = map.keySet();
+        Long total = 0L;
+        for (String key : strings) {
+            total += map.get(key);
+            log.info(key + ":" + map.get(key));
+        }
+
+        log.info("总大小:{}", total);
+
         XxlJobHelper.log("countSizeOssHandler end.....");
     }