Golang如何实现简单的文件压缩与解压_Golang文件压缩解压项目实战

func compressFiles(filename string, files []string) error {
    outFile, err := os.Create(filename)
    if err != nil {
        return err
    }
    defer outFile.Close()

    zipWriter := zip.NewWriter(outFile)
    defer zipWriter.Close()

    for _, filePath := range files {
        if err := addFileToZip(zipWriter, filePath); err != nil {
            return err
        }
    }
    return nil
}

func addFileToZip(zipWriter *zip.Writer, filename string) error {
    file, err := os.Open(filename)
    if err != nil {
        return err
    }
    defer file.Close()

    info, err := file.Stat()
    if err != nil {
        return err
    }

    header, err := zip.FileInfoHeader(info, "")
    if err != nil {
        return err
    }
    // 使用Deflate压缩算法，可减小体积
    header.Method = zip.Deflate

    writer, err := zipWriter.CreateHeader(header)
    if err != nil {
        return err
    }

    _, err = io.Copy(writer, file)
    return err
}

compressFiles("output.zip", []string{"file1.txt", "file2.txt"})

func decompressZip(zipPath, targetDir string) error {
    reader, err := zip.OpenReader(zipPath)
    if err != nil {
        return err
    }
    defer reader.Close()

    for _, file := range reader.File {
        filePath := filepath.Join(targetDir, file.Name)

        if file.FileInfo().IsDir() {
            os.MkdirAll(filePath, os.ModePerm)
            continue
        }

        if err := os.MkdirAll(filepath.Dir(filePath), os.ModePerm); err != nil {
            return err
        }

        outFile, err := os.Create(filePath)
        if err != nil {
            return err
        }

        rc, err := file.Open()
        if err != nil {
            outFile.Close()
            return err
        }

        _, err = io.Copy(outFile, rc)
        rc.Close()
        outFile.Close()

        if err != nil {
            return err
        }
    }
    return nil
}

decompressZip("output.zip", "./extracted/")