我要评论go 语言压缩文件处理
在现代的应用开发中，处理压缩文件（如 .zip 格式）是常见的需求。go 语言提供了内置的 archive/zip 包来处理 .zip 文件的读写，但有时我们需要封装一些常用操作，使得代码更加简洁、易用。本文将介绍如何使用 go 语言封装一个 ziputil 包，来处理文件的压缩和解压操作。
1. 压缩文件：zip函数
在 go 语言中，压缩文件通常需要使用 archive/zip 包。我们将对文件夹或文件进行遍历，创建一个新的 .zip 文件，并将文件或文件夹逐个添加到压缩包中。
package ziputil
import (
"archive/zip"
"go-admin/app/brush/utils"
"sync"
"io"
"os"
"path/filepath"
log "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-8
filename := 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 实现了解压过程的并发处理，提高了解压效率。对于较大的压缩文件或包含大量文件的压缩包，使用并发处理可以显著提升性能。
到此这篇关于基于go语言实现压缩文件处理的文章就介绍到这了,更多相关go压缩文件内容请搜索代码网以前的文章或继续浏览下面的相关文章希望大家以后多多支持代码网！
发表评论