Go语言中读取XZ压缩文件：策略与实践

package main

import (
    "bytes"
    "fmt"
    "io"
    "log"
    "os"
    "os/exec"
)

// xzReader 函数接收一个io.Reader作为XZ压缩数据的来源，
// 返回一个io.ReadCloser，用于读取解压后的数据。
// 它通过执行外部"xz"命令来完成解压。
func xzReader(r io.Reader) io.ReadCloser {
    // 创建一个管道，用于连接外部命令的Stdout和Go程序的ReadCloser
    rpipe, wpipe := io.Pipe()

    // 准备执行xz命令
    // --decompress: 指定解压操作
    // --stdout: 将解压后的数据输出到标准输出
    cmd := exec.Command("xz", "--decompress", "--stdout")
    cmd.Stdin = r       // 将传入的io.Reader作为xz命令的Stdin
    cmd.Stdout = wpipe  // 将xz命令的Stdout连接到管道的写入端

    // 在一个新的goroutine中运行xz命令
    // 这样可以避免阻塞主goroutine，允许数据并行流动
    go func() {
        // 运行命令并等待其完成
        // 如果命令执行失败，将错误传递给管道的写入端并关闭
        err := cmd.Run()
        wpipe.CloseWithError(err) // 关闭管道的写入端，并传递可能的错误
    }()

    return rpipe // 返回管道的读取端，调用者可以从中读取解压后的数据
}

func main() {
    // 示例：模拟一个XZ压缩数据源
    // 在实际应用中，r可能是一个文件句柄、网络连接或其他io.Reader
    // 为了演示，我们先创建一个假的XZ文件内容（需要实际的XZ数据才能正确运行）
    // 注意：这里的"fake xz data"只是占位符，实际运行时需要有效的XZ数据
    // 你可以使用 `echo "Hello, XZ!" | xz > test.xz` 来创建一个测试文件
    // 然后将文件内容读入bytes.Reader

    // 为了简化演示，我们假设有一个名为 'test.xz' 的文件
    // 实际使用时，请确保 'test.xz' 是一个有效的XZ压缩文件
    xzFilePath := "test.xz"

    // 创建一个示例XZ文件（如果不存在）
    // 注意：这是一个简化的创建过程，实际XZ文件需要通过xz工具创建
    // 运行 `echo "Hello from XZ file!" | xz > test.xz` 在命令行创建
    if _, err := os.Stat(xzFilePath); os.IsNotExist(err) {
        fmt.Printf("Warning: '%s' not found. Please create it using 'echo \"Hello from XZ file!\" | xz > %s'\n", xzFilePath, xzFilePath)
        // 为了演示，这里创建一个简单的非XZ数据，但xzReader会失败
        // 如果你想测试成功的xzReader，请务必手动创建test.xz
        dummyContent := "This is not an XZ file. Please create 'test.xz' for a successful demo."
        err := os.WriteFile(xzFilePath, []byte(dummyContent), 0644)
        if err != nil {
            log.Fatalf("Failed to write dummy file: %v", err)
        }
    }


    // 打开XZ文件
    file, err := os.Open(xzFilePath)
    if err != nil {
        log.Fatalf("无法打开XZ文件: %v", err)
    }
    defer file.Close()

    // 使用xzReader函数获取解压后的数据流
    decompressedReader := xzReader(file)
    defer decompressedReader.Close() // 确保关闭解压后的读取器

    // 从解压后的数据流中读取内容
    decompressedData, err := io.ReadAll(decompressedReader)
    if err != nil {
        // 这里的错误可能是xz命令执行失败，或者管道读取错误
        // 例如，如果test.xz不是一个有效的XZ文件，xz命令会报错
        log.Fatalf("读取解压数据失败: %v", err)
    }

    fmt.Println("解压后的内容:")
    fmt.Println(string(decompressedData))

    // 另一个例子：直接从内存中的XZ数据解压
    // 假设我们有一些XZ压缩的字节数据
    // compressedBytes := []byte{...} // 实际的XZ压缩字节
    // bufReader := bytes.NewReader(compressedBytes)
    // decompressedReaderFromBuf := xzReader(bufReader)
    // defer decompressedReaderFromBuf.Close()
    // decompressedDataFromBuf, err := io.ReadAll(decompressedReaderFromBuf)
    // if err != nil {
    //  log.Fatalf("从内存解压失败: %v", err)
    // }
    // fmt.Println("从内存解压后的内容:", string(decompressedDataFromBuf))
}