直接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
效率提升四倍 非常明显