Groovy创建Zip文件后删除原文件的最优实现方案咨询

我需要在待压缩文件的同一目录下生成对应的Zip压缩包，完成后删除原文件。自己尝试写了一段Groovy代码，但因为有Python开发背景，对Groovy/Java了解有限，想请教这个需求的最优实现方案，以及实现过程中需要注意哪些异常情况。

我尝试编写的代码 #

import java.util.zip.*

class ZipUtility {

    def createZipFile(fileToZip, cleanUp=false){
        String zipFileName = fileToZip.toString() + ".zip"
        ZipOutputStream zipFile = new ZipOutputStream(new FileOutputStream(zipFileName))
        if (fileToZip.isFile()) {
            zipFile.putNextEntry(new ZipEntry(fileToZip.name))
            def buffer = new byte[fileToZip.size()]
            fileToZip.withInputStream {
                zipFile.write(buffer, 0, it.read(buffer))
            }
            zipFile.closeEntry()
        }
        zipFile.close()

        if (cleanUp) {
            // this deletes the file test.txt
            boolean fileSuccessfullyDeleted =  new File(fileToZip).delete()
        }
        return zipFile // Not sure if this is the correct approach. Any Exceptions to look for?
  }
}

优化后的实现方案 #

针对原代码的不足，优化后的版本解决了资源泄漏、大文件内存占用、异常回滚等问题：

import java.util.zip.*

class ZipUtility {

    boolean createZipFile(File fileToZip, boolean cleanUp = false) {
        // 校验输入文件合法性
        if (!fileToZip.exists() || !fileToZip.isFile()) {
            println "输入不是有效文件"
            return false
        }

        String zipFileName = "${fileToZip.absolutePath}.zip"
        File zipFile = new File(zipFileName)
        
        // 避免覆盖已存在的压缩包（可根据需求调整为覆盖逻辑）
        if (zipFile.exists()) {
            println "目标压缩包已存在"
            return false
        }

        // 使用try-with-resources自动关闭流，避免资源泄漏
        try (ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(zipFile));
             FileInputStream fis = new FileInputStream(fileToZip)) {

            ZipEntry entry = new ZipEntry(fileToZip.name)
            zos.putNextEntry(entry)

            // 用固定大小缓冲区，避免大文件占用过多内存
            byte[] buffer = new byte[4096]
            int bytesRead
            while ((bytesRead = fis.read(buffer)) != -1) {
                zos.write(buffer, 0, bytesRead)
            }
            zos.closeEntry()
            zos.finish() // 确保压缩包写入完成

            // 只有压缩成功才执行原文件删除
            if (cleanUp) {
                if (!fileToZip.delete()) {
                    println "原文件删除失败"
                    // 回滚：删除不完整的压缩包，避免残留
                    zipFile.delete()
                    return false
                }
            }
            return true
        } catch (IOException e) {
            // 异常时删除已生成的无效压缩包
            zipFile.delete()
            println "压缩过程出错: ${e.message}"
            return false
        }
    }
}

需要注意的异常与问题 #

• IO异常：原文件不存在、无读写权限、磁盘空间不足、文件被其他进程占用等场景，都会抛出IOException，必须捕获处理。
• 大文件内存溢出：原代码用文件大小作为缓冲区大小，若文件是GB级别，会直接耗尽内存，改用固定大小缓冲区循环读写即可避免。
• 压缩包覆盖风险：目标位置已有同名压缩包时，直接写入会覆盖原有文件，需根据需求选择终止操作或覆盖逻辑。
• 原文件删除失败：压缩成功但原文件因锁定、权限不足无法删除时，要回滚删除生成的压缩包，避免出现数据混乱。
• 资源泄漏：原代码未正确处理流的关闭，一旦中间抛出异常，流资源会泄漏，使用try-with-resources可自动关闭实现AutoCloseable的资源。
• 参数合法性：必须校验输入的fileToZip是否为有效文件，避免传入目录或不存在的路径。

内容的提问来源于stack exchange，提问作者iDev