go-unarr: Golang 多格式压缩文件解压库

go-unarr 是一个基于 C 库 unarr 的 Go 语言绑定，支持解压 RAR、TAR、ZIP 和 7z 等多种压缩格式。它提供了简单易用的 API 来处理压缩文件。

安装

首先需要安装 go-unarr:

go get github.com/gen2brain/go-unarr

基本用法

1. 列出压缩文件内容

package main

import (
	"fmt"
	"github.com/gen2brain/go-unarr"
)

func main() {
	a, err := unarr.NewArchive("example.zip")
	if err != nil {
		panic(err)
	}
	defer a.Close()

	list, err := a.List()
	if err != nil {
		panic(err)
	}

	for _, file := range list {
		fmt.Println(file)
	}
}

2. 解压单个文件

package main

import (
	"fmt"
	"io"
	"os"
	"github.com/gen2brain/go-unarr"
)

func main() {
	a, err := unarr.NewArchive("example.zip")
	if err != nil {
		panic(err)
	}
	defer a.Close()

	// 解压特定文件
	err = a.EntryFor("path/to/file.txt")
	if err != nil {
		panic(err)
	}

	// 创建目标文件
	out, err := os.Create("output.txt")
	if err != nil {
		panic(err)
	}
	defer out.Close()

	// 复制文件内容
	_, err = io.Copy(out, a)
	if err != nil {
		panic(err)
	}

	fmt.Println("File extracted successfully")
}

3. 解压全部文件

package main

import (
	"fmt"
	"os"
	"path/filepath"
	"github.com/gen2brain/go-unarr"
)

func main() {
	a, err := unarr.NewArchive("example.zip")
	if err != nil {
		panic(err)
	}
	defer a.Close()

	list, err := a.List()
	if err != nil {
		panic(err)
	}

	// 创建输出目录
	err = os.MkdirAll("output", 0755)
	if err != nil {
		panic(err)
	}

	for _, file := range list {
		err = a.EntryFor(file)
		if err != nil {
			fmt.Printf("Failed to extract %s: %v\n", file, err)
			continue
		}

		path := filepath.Join("output", file)
		err = os.MkdirAll(filepath.Dir(path), 0755)
		if err != nil {
			fmt.Printf("Failed to create directory for %s: %v\n", file, err)
			continue
		}

		out, err := os.Create(path)
		if err != nil {
			fmt.Printf("Failed to create %s: %v\n", file, err)
			continue
		}

		_, err = out.ReadFrom(a)
		out.Close()
		if err != nil {
			fmt.Printf("Failed to extract %s: %v\n", file, err)
			continue
		}

		fmt.Printf("Extracted: %s\n", file)
	}
}

高级功能

1. 密码保护的压缩文件

package main

import (
	"fmt"
	"github.com/gen2brain/go-unarr"
)

func main() {
	a, err := unarr.NewArchive("protected.zip")
	if err != nil {
		panic(err)
	}
	defer a.Close()

	// 设置密码
	a.SetPassword("secret")

	// 验证密码是否正确
	valid, err := a.EntryFor("file.txt")
	if err != nil || !valid {
		fmt.Println("Invalid password or file not found")
		return
	}

	fmt.Println("Password correct, file can be extracted")
}

2. 处理大文件

package main

import (
	"fmt"
	"io"
	"os"
	"github.com/gen2brain/go-unarr"
)

func extractLargeFile(a *unarr.Archive, filename, outputPath string) error {
	err := a.EntryFor(filename)
	if err != nil {
		return err
	}

	out, err := os.Create(outputPath)
	if err != nil {
		return err
	}
	defer out.Close()

	// 使用缓冲区处理大文件
	buf := make([]byte, 32*1024) // 32KB 缓冲区
	_, err = io.CopyBuffer(out, a, buf)
	return err
}

func main() {
	a, err := unarr.NewArchive("largefile.rar")
	if err != nil {
		panic(err)
	}
	defer a.Close()

	err = extractLargeFile(a, "bigfile.bin", "bigfile.bin")
	if err != nil {
		panic(err)
	}

	fmt.Println("Large file extracted successfully")
}

注意事项

1. go-unarr 只支持解压，不支持创建压缩文件
2. 对于加密文件，需要先调用 SetPassword() 方法
3. 处理大文件时，建议使用缓冲区
4. 某些 RAR 版本可能不完全支持

性能考虑

• 对于 ZIP 文件，原生 Go 的 archive/zip 包可能性能更好
• go-unarr 的优势在于支持多种格式的统一接口
• 对于大批量小文件，可以预先调用 List() 然后批量处理

go-unarr 是一个简单实用的多格式解压库，特别适合需要处理多种压缩格式的项目。