百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 编程字典 > 正文

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

toyiye 2024-07-09 22:51 9 浏览 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


效率提升四倍 非常明显

相关推荐

为何越来越多的编程语言使用JSON(为什么编程)

JSON是JavascriptObjectNotation的缩写,意思是Javascript对象表示法,是一种易于人类阅读和对编程友好的文本数据传递方法,是JavaScript语言规范定义的一个子...

何时在数据库中使用 JSON(数据库用json格式存储)

在本文中,您将了解何时应考虑将JSON数据类型添加到表中以及何时应避免使用它们。每天?分享?最新?软件?开发?,Devops,敏捷?,测试?以及?项目?管理?最新?,最热门?的?文章?,每天?花?...

MySQL 从零开始:05 数据类型(mysql数据类型有哪些,并举例)

前面的讲解中已经接触到了表的创建,表的创建是对字段的声明,比如:上述语句声明了字段的名称、类型、所占空间、默认值和是否可以为空等信息。其中的int、varchar、char和decimal都...

JSON对象花样进阶(json格式对象)

一、引言在现代Web开发中,JSON(JavaScriptObjectNotation)已经成为数据交换的标准格式。无论是从前端向后端发送数据,还是从后端接收数据,JSON都是不可或缺的一部分。...

深入理解 JSON 和 Form-data(json和formdata提交区别)

在讨论现代网络开发与API设计的语境下,理解客户端和服务器间如何有效且可靠地交换数据变得尤为关键。这里,特别值得关注的是两种主流数据格式:...

JSON 语法(json 语法 priority)

JSON语法是JavaScript语法的子集。JSON语法规则JSON语法是JavaScript对象表示法语法的子集。数据在名称/值对中数据由逗号分隔花括号保存对象方括号保存数组JS...

JSON语法详解(json的语法规则)

JSON语法规则JSON语法是JavaScript对象表示法语法的子集。数据在名称/值对中数据由逗号分隔大括号保存对象中括号保存数组注意:json的key是字符串,且必须是双引号,不能是单引号...

MySQL JSON数据类型操作(mysql的json)

概述mysql自5.7.8版本开始,就支持了json结构的数据存储和查询,这表明了mysql也在不断的学习和增加nosql数据库的有点。但mysql毕竟是关系型数据库,在处理json这种非结构化的数据...

JSON的数据模式(json数据格式示例)

像XML模式一样,JSON数据格式也有Schema,这是一个基于JSON格式的规范。JSON模式也以JSON格式编写。它用于验证JSON数据。JSON模式示例以下代码显示了基本的JSON模式。{"...

前端学习——JSON格式详解(后端json格式)

JSON(JavaScriptObjectNotation)是一种轻量级的数据交换格式。易于人阅读和编写。同时也易于机器解析和生成。它基于JavaScriptProgrammingLa...

什么是 JSON:详解 JSON 及其优势(什么叫json)

现在程序员还有谁不知道JSON吗?无论对于前端还是后端,JSON都是一种常见的数据格式。那么JSON到底是什么呢?JSON的定义...

PostgreSQL JSON 类型:处理结构化数据

PostgreSQL提供JSON类型,以存储结构化数据。JSON是一种开放的数据格式,可用于存储各种类型的值。什么是JSON类型?JSON类型表示JSON(JavaScriptO...

JavaScript:JSON、三种包装类(javascript 包)

JOSN:我们希望可以将一个对象在不同的语言中进行传递,以达到通信的目的,最佳方式就是将一个对象转换为字符串的形式JSON(JavaScriptObjectNotation)-JS的对象表示法...

Python数据分析 只要1分钟 教你玩转JSON 全程干货

Json简介:Json,全名JavaScriptObjectNotation,JSON(JavaScriptObjectNotation(记号、标记))是一种轻量级的数据交换格式。它基于J...

比较一下JSON与XML两种数据格式?(json和xml哪个好)

JSON(JavaScriptObjectNotation)和XML(eXtensibleMarkupLanguage)是在日常开发中比较常用的两种数据格式,它们主要的作用就是用来进行数据的传...

取消回复欢迎 发表评论:

请填写验证码