阿里云建网站教程,响应式网页模板下载,最好的外贸网站建设,互联网软件有哪些目录 Go 语言压缩文件处理1. 压缩文件#xff1a;Zip函数2. 解压文件#xff1a;UnZip 函数3. 小结 Go 语言压缩文件处理
在现代的应用开发中#xff0c;处理压缩文件#xff08;如 .zip 格式#xff09;是常见的需求。Go 语言提供了内置的 archive/zip 包来处理 .zip 文… 目录 Go 语言压缩文件处理1. 压缩文件Zip函数2. 解压文件UnZip 函数3. 小结 Go 语言压缩文件处理
在现代的应用开发中处理压缩文件如 .zip 格式是常见的需求。Go 语言提供了内置的 archive/zip 包来处理 .zip 文件的读写但有时我们需要封装一些常用操作使得代码更加简洁、易用。本文将介绍如何使用 Go 语言封装一个 ziputil 包来处理文件的压缩和解压操作。
1. 压缩文件Zip函数
在 Go 语言中压缩文件通常需要使用 archive/zip 包。我们将对文件夹或文件进行遍历创建一个新的 .zip 文件并将文件或文件夹逐个添加到压缩包中。
package ziputilimport (archive/zipgo-admin/app/brush/utilssyncioospath/filepathlog github.com/go-admin-team/go-admin-core/logger
)// Zip 将指定的文件夹或文件压缩为 .zip 文件
func Zip(source, zipFile string) error {// 创建一个新的 zip 文件zipFileWriter, err : os.Create(zipFile)if err ! nil {return err}defer func(zipFileWriter *os.File) {err : zipFileWriter.Close()if err ! nil {log.Errorf(关闭 zip 文件失败: %s, err)}}(zipFileWriter)// 创建 zip 写入器zipWriter : zip.NewWriter(zipFileWriter)defer func(zipWriter *zip.Writer) {err : zipWriter.Close()if err ! nil {log.Errorf(关闭 zip 写入器失败: %s, err)}}(zipWriter)// 获取源文件的绝对路径absSource, err : filepath.Abs(source)if err ! nil {return err}// 遍历文件夹并添加到 zip 文件中return filepath.Walk(absSource, func(path string, info os.FileInfo, err error) error {if err ! nil {return err}// 计算文件相对路径relPath, err : filepath.Rel(absSource, path)if err ! nil {return err}// 如果是目录则在 zip 文件中创建一个目录项if info.IsDir() {if relPath ! . {_, err : zipWriter.Create(relPath /)if err ! nil {return err}}return nil}// 否则将文件添加到 zip 文件return addFileToZip(zipWriter, path, relPath)})
}// addFileToZip 将单个文件添加到 zip 写入器
func addFileToZip(zipWriter *zip.Writer, file string, relPath string) error {f, err : os.Open(file)if err ! nil {return err}defer func(f *os.File) {err : f.Close()if err ! nil {log.Errorf(关闭文件失败: %s, err)}}(f)// 在 zip 文件中创建该文件writer, err : zipWriter.Create(relPath)if err ! nil {return err}// 将文件内容写入 zip_, err io.Copy(writer, f)if err ! nil {return err}return nil
}2. 解压文件UnZip 函数
解压 .zip 文件时我们需要将 .zip 文件中的每个文件提取到指定的目录中。UnZip 函数不仅能够提取文件还能够处理文件夹结构保证提取后的目录结构不丢失。
// UnZip 解压 zip 文件到目标目录
func UnZip(zipFile, destDir string) error {log.Debugf(解压文件: %s 到 %s, zipFile, destDir)r, err : zip.OpenReader(zipFile)if err ! nil {return err}defer func(r *zip.ReadCloser) {err : r.Close()if err ! nil {log.Errorf(关闭 zip 文件失败: %s, err)}}(r)log.Debugf(总共 %d 个文件, len(r.File))// 并发解压每个文件wg : sync.WaitGroup{}for _, f : range r.File {wg.Add(1)go func(rf *zip.File, w *sync.WaitGroup) {defer w.Done()if err : unzipFile(rf, destDir); err ! nil {log.Errorf(解压文件 [%s] 失败: %v, rf.Name, err)}}(f, wg)}wg.Wait()return nil
}// unzipFile 解压单个文件到目标目录
func unzipFile(f *zip.File, destDir string) error {// 将文件名转换为 UTF-8filename : utils.ConvertToUTF8([]byte(f.Name))filePath : filepath.Join(destDir, filename)// 创建文件夹if f.FileInfo().IsDir() {return os.MkdirAll(filePath, os.ModePerm)}// 创建文件的父目录if err : os.MkdirAll(filepath.Dir(filePath), os.ModePerm); err ! nil {log.Errorf(创建目录 [%s] 失败: %v, filepath.Dir(filePath), err)return err}// 打开文件file, err : f.Open()if err ! nil {log.Errorf(打开文件 [%s] 失败: %v, filePath, err)return err}defer func(file io.ReadCloser) {err : file.Close()if err ! nil {log.Errorf(关闭文件 [%s] 失败: %v, filePath, err)}}(file)// 创建文件outFile, err : os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())if err ! nil {log.Errorf(创建文件 [%s] 失败: %v, filePath, err)return err}defer func(outFile *os.File) {err : outFile.Close()if err ! nil {log.Errorf(关闭文件 [%s] 失败: %v, filePath, err)}}(outFile)// 将文件内容写入_, err io.Copy(outFile, file)if err ! nil {log.Errorf(复制文件 [%s] 失败: %v, filePath, err)return err}return nil
}3. 小结
通过 ziputil 包我们可以方便地进行文件和文件夹的压缩和解压操作。该包使用了 Go 内置的 archive/zip 包来处理 .zip 文件并通过 sync.WaitGroup 实现了解压过程的并发处理提高了解压效率。对于较大的压缩文件或包含大量文件的压缩包使用并发处理可以显著提升性能。
