Go语言中读取XZ文件：方法与推荐实践

package main

import (
    "io"
    "log"
    "os"
    "os/exec"
    "strings" // 用于示例中的模拟数据
)

// xzReader 接收一个io.Reader作为XZ压缩数据的来源，
// 返回一个io.ReadCloser，用于读取解压后的数据。
// 此函数通过调用外部 'xz' 命令行工具实现解压。
func xzReader(r io.Reader) io.ReadCloser {
    // 创建一个管道，用于连接外部命令的Stdout和Go程序的ReadCloser。
    // rpipe是管道的读取端，wpipe是管道的写入端。
    rpipe, wpipe := io.Pipe()

    // 构建xz解压命令。
    // "xz": 命令行工具名称。
    // "--decompress": 指定执行解压操作。
    // "--stdout": 将解压后的数据输出到标准输出，而不是写入文件。
    cmd := exec.Command("xz", "--decompress", "--stdout")
    cmd.Stdin = r        // 将传入的io.Reader作为xz命令的标准输入。
    cmd.Stdout = wpipe   // 将xz命令的标准输出连接到管道的写入端。

    // 在一个新的goroutine中运行xz命令。
    // 这样做是为了避免阻塞主goroutine，同时允许数据流式传输：
    // 当Go程序从rpipe读取时，xz命令会同时将解压数据写入wpipe。
    go func() {
        // 运行命令并捕获可能发生的错误。
        err := cmd.Run()
        // 命令结束后，关闭管道的写入端，并传递可能发生的错误。
        // 这会通知管道的读取端 (rpipe)，数据已结束或发生错误。
        wpipe.CloseWithError(err)
    }()

    return rpipe // 返回管道的读取端，作为解压数据的io.ReadCloser。
}

func main() {
    log.Println("--- 演示通过模拟数据解压 ---")
    // 示例：创建一个模拟的XZ压缩数据。
    // 注意：这里的字符串 "模拟的XZ压缩数据..." 并不是一个有效的XZ格式数据。
    // 在实际应用中，`xzInput` 应该是一个真正的XZ压缩字节流。
    // 例如，你可以通过 `echo "Hello Go!" | xz -c > test.xz` 创建一个测试文件。
    mockCompressedData := "模拟的XZ压缩数据，这里需要真实的XZ数据才能运行"
    xzInput := strings.NewReader(mockCompressedData)

    // 获取解压后的Reader
    reader := xzReader(xzInput)
    defer reader.Close() // 确保关闭Reader，释放资源

    // 尝试读取解压后的数据
    decompressedBytes, err := io.ReadAll(reader)
    if err != nil {
        // 对于模拟数据，这里很可能会因为 'xz' 命令无法识别输入而报错。
        log.Printf("读取模拟解压数据失败（预期错误，因为输入不是有效XZ格式）: %v", err)
    } else {
        log.Printf("从模拟数据解压后的数据: %s", string(decompressedBytes))
    }

    log.Println("\n--- 演示通过创建并读取真实XZ文件解压 ---")
    // 为了更真实的演示，我们尝试创建一个临时的XZ文件并读取它。
    testFilePath := "hello.xz"
    // 创建一个临时的XZ文件用于演示。
    // `sh -c` 用于执行一个shell命令，方便管道操作。
    cmdCreateXZ := exec.Command("sh", "-c", `echo "Hello from XZ in Go!" | xz > `+testFilePath)
    if err := cmdCreateXZ.Run(); err != nil {
        log.Printf("无法创建测试XZ文件（可能xz未安装、权限问题或系统不支持）：%v", err)
        log.Println("请确保您的系统已安装 'xz' 命令行工具。")
    } else {
        log.Printf("成功创建测试文件: %s", testFilePath)
        // 确保在程序结束时清理这个临时文件
        defer func() {
            if err := os.Remove(testFilePath); err != nil {
                log.Printf("无法删除临时文件 %s: %v", testFilePath, err)
            } else {
                log.Printf("成功删除临时文件: %s", testFilePath)
            }
        }()

        // 打开刚刚创建的XZ文件
        file, err := os.Open(testFilePath)
        if err != nil {
            log.Fatalf("无法打开测试XZ文件: %v", err)
        }
        defer file.Close() // 确保关闭文件句柄

        // 使用xzReader函数解压真实文件
        readerFromRealFile := xzReader(file)
        defer readerFromRealFile.Close() // 确保关闭解压Reader

        // 读取解压后的数据
        realDecompressedBytes, err := io.ReadAll(readerFromRealFile)
        if err != nil {
            log.Fatalf("读取真实XZ文件解压数据失败: %v", err)
        }
        log.Printf("从真实XZ文件解压后的数据: %s", string(realDecompressedBytes))
    }
}