add close

momo199059 · Dec 20, 2021 · 89af537 · 89af537
1 parent b637058
commit 89af537
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/core/src/main/java/com/bytedance/android/aabresguard/bundle/AppBundleAnalyzer.java b/core/src/main/java/com/bytedance/android/aabresguard/bundle/AppBundleAnalyzer.java
@@ -18,6 +18,7 @@ public class AppBundleAnalyzer {
 
     private static final Logger logger = Logger.getLogger(AppBundleAnalyzer.class.getName());
     private final Path bundlePath;
+    private ZipFile bundleZip;
 
     public AppBundleAnalyzer(Path bundlePath) {
         checkFileExistsAndReadable(bundlePath);
@@ -26,10 +27,17 @@ public AppBundleAnalyzer(Path bundlePath) {
 
     public AppBundle analyze() throws IOException {
         TimeClock timeClock = new TimeClock();
-        ZipFile bundleZip = new ZipFile(bundlePath.toFile());
+        bundleZip = new ZipFile(bundlePath.toFile());
         AppBundle appBundle = AppBundle.buildFromZip(bundleZip);
-//        bundleZip.close();
         System.out.println(String.format("analyze bundle file done, const %s", timeClock.getCoast()));
         return appBundle;
     }
+
+    public void closeFile() throws IOException {
+        if(bundleZip != null){
+            bundleZip.close();
+            bundleZip = null;
+            System.out.println(String.format("bundle zip close"));
+        }
+    }
 }
diff --git a/core/src/main/java/com/bytedance/android/aabresguard/commands/ObfuscateBundleCommand.java b/core/src/main/java/com/bytedance/android/aabresguard/commands/ObfuscateBundleCommand.java
@@ -172,7 +172,8 @@ public static ObfuscateBundleCommand fromFlags(ParsedFlags flags) throws Documen
     public Path execute() throws IOException, InterruptedException {
         TimeClock timeClock = new TimeClock();
 
-        AppBundle appBundle = new AppBundleAnalyzer(getBundlePath()).analyze();
+        AppBundleAnalyzer analyzer = new AppBundleAnalyzer(getBundlePath());
+        AppBundle appBundle = analyzer.analyze();
         // filter file
         if (getFilterFile().isPresent() && getFilterFile().get()) {
             Set<String> fileFilterRules = new HashSet<>();
@@ -243,6 +244,7 @@ storeFile, getStorePassword().get(), getKeyAlias().get(), getKeyPassword().get()
                 getNetFileSizeDescription(rawSize),
                 getNetFileSizeDescription(filteredSize)
         ));
+        analyzer.closeFile();
         return getOutputPath();
     }