专业的编程技术博客社区

网站首页 > 博客文章 正文

【多线程】效率提升四倍,python & java 文件夹列表zip压缩

baijin 2024-08-31 16:17:46 博客文章 3 ℃ 0 评论

直接show code,话不多说

1、python

import os
import zipfile
import concurrent.futures

def zip_folder(folder_path):
    # 获取文件夹名称
    folder_name = os.path.basename(folder_path)
    
    # 创建压缩文件名
    zip_filename = folder_name + '.zip'
    
    # 创建压缩文件
    with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zipf:
        # 遍历文件夹下的所有文件和子文件夹
        for root, dirs, files in os.walk(folder_path):
            # 压缩当前文件夹内所有的文件
            for file in files:
                file_path = os.path.join(root, file)
                print(file_path)
                zipf.write(file_path, os.path.relpath(file_path, folder_path))

def main():
    # 指定文件夹路径
    target_folder = 'c:/py-test'
    
    # 获取文件夹下的所有子文件夹路径
    subfolders = [os.path.join(target_folder, name) for name in os.listdir(target_folder) if os.path.isdir(os.path.join(target_folder, name))]

    # 使用多线程压缩文件夹
    with concurrent.futures.ThreadPoolExecutor() as executor:
        executor.map(zip_folder, subfolders)

if __name__ == '__main__':
    main()

2、java ExecutorService

import cn.hutool.core.date.StopWatch;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.ZipUtil;
import lombok.extern.slf4j.Slf4j;
import java.io.File;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;

@Slf4j
public class FileDealTask implements Callable<Integer> {

    private String filePath;
    private String suffix;

    public FileDealTask(String filePath,String suffix){
        this.filePath = filePath;
        this.suffix = suffix;
    }

    @Override
    public Integer call() throws Exception {
        StopWatch stopWatch = new StopWatch();
        try {
            stopWatch.start();
            String zipPath = filePath + suffix;
            ZipUtil.zip(filePath,zipPath);
            stopWatch.stop();
            log.info("{} zip file cost time : {}",filePath,stopWatch.getTotalTimeSeconds());
        } catch (Exception e) {
            stopWatch.stop();
            log.info("{} zip file cost time : {}",filePath,stopWatch.getTotalTimeSeconds());
            return 0;
        }
        return 1;
    }

    public static void main(String[] args) throws Exception {
        File rootPath = new File("c:/test");
        File[] rootSecondDirs = rootPath.listFiles();
        List<File> unzipFileDirs = new ArrayList<>();

        for (File targetZipDir : Objects.requireNonNull(rootSecondDirs)) {
            if(targetZipDir.getAbsolutePath().contains(".zip")){
                FileUtil.del(targetZipDir);
            } else {
                unzipFileDirs.add(targetZipDir);
            }
        }

        StopWatch stopWatch = new StopWatch();
        stopWatch.start();

        ExecutorService executorService = Executors.newFixedThreadPool(10);

        List<FileDealTask> tasks = new ArrayList<>();


        for (File targetZipDir : Objects.requireNonNull(unzipFileDirs)) {
            tasks.add(new FileDealTask(targetZipDir.getAbsolutePath(),".zip"));
        }

        List<Future<Integer>> results = executorService.invokeAll(tasks);

        Integer total = 0;

        for (Future<Integer> result : results) {
            total += result.get();
        }

        log.info("预处理的文件个数: {},已处理的文件个数: {}",unzipFileDirs.size(),total);
        executorService.shutdown();

        stopWatch.stop();
        log.info("zip file total time : {} seconds",stopWatch.getTotalTimeSeconds());

        log.info("------------------");
        handleSingleThreadZipFile(unzipFileDirs,"-single.zip");
    }

    static void handleSingleThreadZipFile(List<File> unzipFileDirs,String suffix){
        StopWatch mainStopWatch = new StopWatch();
        mainStopWatch.start();
        unzipFileDirs.forEach(zipFileDir -> {
            StopWatch stopWatch = new StopWatch();
            String filePath = zipFileDir.getAbsolutePath();
            try {
                stopWatch.start();
                String zipPath = filePath + suffix;
                ZipUtil.zip(filePath,zipPath);
                stopWatch.stop();
                log.info("[single] {} zip file cost time : {}",filePath,stopWatch.getTotalTimeSeconds());
            } catch (Exception e) {
                stopWatch.stop();
                log.info("[single] {} zip file cost time : {}",filePath,stopWatch.getTotalTimeSeconds());
            }
        });
        mainStopWatch.stop();
        log.info("[single] total zip file total time : {} seconds",mainStopWatch.getTotalTimeSeconds());
    }
}
15:52:27.077 [pool-1-thread-9] INFO FileDealTask - c:\test\mqttServices zip file cost time : 0.1024325
15:52:32.028 [pool-1-thread-1] INFO FileDealTask - c:\test\business-general zip file cost time : 5.0669379
15:52:32.803 [pool-1-thread-5] INFO FileDealTask - c:\test\foundation-data-analyse zip file cost time : 5.8413171
15:52:32.843 [pool-1-thread-2] INFO FileDealTask - c:\test\data-calculate zip file cost time : 5.88223
15:52:32.880 [pool-1-thread-10] INFO FileDealTask - c:\test\rain zip file cost time : 5.9186832
15:52:32.905 [pool-1-thread-9] INFO FileDealTask - c:\test\rain-calculate zip file cost time : 5.8177634
15:52:33.891 [pool-1-thread-4] INFO FileDealTask - c:\test\forecast zip file cost time : 6.9297484
15:52:34.191 [pool-1-thread-6] INFO FileDealTask - c:\test\foundation-map-config zip file cost time : 7.2297071
15:52:34.872 [pool-1-thread-8] INFO FileDealTask - c:\test\monitor zip file cost time : 7.9113504
15:52:36.186 [pool-1-thread-1] INFO FileDealTask - c:\test\ScadaApiServer zip file cost time : 4.1588949
15:52:37.313 [pool-1-thread-10] INFO FileDealTask - c:\test\security-auth zip file cost time : 4.4325229
15:52:37.496 [pool-1-thread-2] INFO FileDealTask - c:\test\security-admin zip file cost time : 4.6532674
15:52:37.810 [pool-1-thread-9] INFO FileDealTask - c:\test\security-gateway zip file cost time : 4.9047188
15:52:38.481 [pool-1-thread-3] INFO FileDealTask - c:\test\event zip file cost time : 11.5199858
15:52:39.351 [pool-1-thread-7] INFO FileDealTask - c:\test\hs-message zip file cost time : 12.3901957
15:52:42.060 [pool-1-thread-5] INFO FileDealTask - c:\test\scenario zip file cost time : 9.2558464
15:52:42.060 [main] INFO FileDealTask - 预处理的文件个数: 16,已处理的文件个数: 16
15:52:42.061 [main] INFO FileDealTask - zip file total time : 15.1030676 seconds
15:52:42.061 [main] INFO FileDealTask - ------------------
15:52:44.274 [main] INFO FileDealTask - [single] c:\test\business-general zip file cost time : 2.2091358
15:52:46.867 [main] INFO FileDealTask - [single] c:\test\data-calculate zip file cost time : 2.592884
15:52:52.871 [main] INFO FileDealTask - [single] c:\test\event zip file cost time : 6.0036009
15:52:55.884 [main] INFO FileDealTask - [single] c:\test\forecast zip file cost time : 3.0128931
15:52:58.408 [main] INFO FileDealTask - [single] c:\test\foundation-data-analyse zip file cost time : 2.5239698
15:53:01.566 [main] INFO FileDealTask - [single] c:\test\foundation-map-config zip file cost time : 3.1574848
15:53:08.375 [main] INFO FileDealTask - [single] c:\test\hs-message zip file cost time : 6.80953
15:53:12.125 [main] INFO FileDealTask - [single] c:\test\monitor zip file cost time : 3.7499176
15:53:12.128 [main] INFO FileDealTask - [single] c:\test\mqttServices zip file cost time : 0.0030089
15:53:14.811 [main] INFO FileDealTask - [single] c:\test\rain zip file cost time : 2.6828985
15:53:17.853 [main] INFO FileDealTask - [single] c:\test\rain-calculate zip file cost time : 3.0420701
15:53:20.651 [main] INFO FileDealTask - [single] c:\test\ScadaApiServer zip file cost time : 2.7966223
15:53:30.241 [main] INFO FileDealTask - [single] c:\test\scenario zip file cost time : 9.5898768
15:53:33.647 [main] INFO FileDealTask - [single] c:\test\security-admin zip file cost time : 3.406758
15:53:37.601 [main] INFO FileDealTask - [single] c:\test\security-auth zip file cost time : 3.9528478
15:53:42.291 [main] INFO FileDealTask - [single] c:\test\security-gateway zip file cost time : 4.6902837
15:53:42.291 [main] INFO FileDealTask - [single] total zip file total time : 60.2297293 seconds


效率提升四倍 非常明显

Tags:

本文暂时没有评论,来添加一个吧(●'◡'●)

欢迎 发表评论:

最近发表
标签列表