Go groot 怎样把ROOT对象写入新文件

Go groot 写入 ROOT 对象的核心是：先用 groot.Create 创建文件，再用返回的 File.Put 按键名保存对象，最后检查并处理 Close 错误。下面用一个 rbase.ObjString 做最小例子，关闭后再读回确认对象仍在文件中。

官方地址：https://pkg.go.dev/go-hep.org/x/hep/groot

不要把“创建文件”和“写入对象”混成一个调用：Create 负责得到可写文件句柄，Put 负责建立对象键并保存内容，文件关闭才是写入生命周期的收尾。

安装依赖并准备一个 ROOT 对象

groot 是 Go-HEP 提供的纯 Go ROOT 文件读写包。文章示例选择 rbase.NewObjString，因为它不需要先定义复杂的树结构，足以说明“对象名—对象值—ROOT 文件”的关系。

# 初始化模块并添加 groot 依赖
go mod init example.com/groot-write
go get go-hep.org/x/hep/groot

如果项目已经有 go.mod，只执行第二条命令即可。注意导入路径中的 go-hep.org/x/hep/groot/rbase 用来构造可写入的 ROOT 基础对象。

用 groot.Create 创建文件并写入 ROOT 对象

下面的写法把字符串对象放到 notes 这个键下。键名是后续用 Get 取对象时使用的索引，不是 Go 变量名；同一个文件里应给不同对象使用清晰且稳定的名称。

package main

import (
    "log"

    "go-hep.org/x/hep/groot"
    "go-hep.org/x/hep/groot/rbase"
)

func main() {
    const filename = "notes.root"

    // Create 负责创建新的 ROOT 文件，并返回可写句柄。
    file, err := groot.Create(filename)
    if err != nil {
        log.Fatalf("创建 ROOT 文件失败: %v", err)
    }

    // Put 使用稳定键名保存一个 ROOT 对象。
    obj := rbase.NewObjString("由 Go groot 写入")
    if err := file.Put("notes", obj); err != nil {
        _ = file.Close() // 写入失败时也释放文件资源。
        log.Fatalf("写入 ROOT 对象失败: %v", err)
    }

    // Close 负责完成文件收尾；关闭错误也必须向调用方报告。
    if err := file.Close(); err != nil {
        log.Fatalf("关闭 ROOT 文件失败: %v", err)
    }
}

这里没有使用 defer file.Close()，是为了让关闭错误可以直接被检查。若业务代码更适合延迟关闭，也可以先注册清理逻辑，但不要忽略最终的 Close 返回值。

关闭后重新打开，按键名读取对象

写入代码结束后，用 groot.Open 打开文件，再用同一个键名调用 Get。读取结果是 ROOT 的对象接口，需要根据你写入的实际类型做类型断言。

// readback.go：打开刚写好的文件，并按键名取回对象。
package main

import (
    "fmt"
    "log"

    "go-hep.org/x/hep/groot"
    "go-hep.org/x/hep/groot/root"
)

func main() {
    // Open 以只读方式打开已有 ROOT 文件。
    file, err := groot.Open("notes.root")
    if err != nil {
        log.Fatalf("打开 ROOT 文件失败: %v", err)
    }
    defer file.Close() // 读取结束后释放只读句柄。

    // Get 的参数必须与写入时的键名一致。
    value, err := file.Get("notes")
    if err != nil {
        log.Fatalf("读取对象失败: %v", err)
    }

    // 只把返回值断言为正文写入时使用的 ROOT 对象类型。
    text, ok := value.(root.ObjString)
    if !ok {
        log.Fatalf("对象类型不是 root.ObjString，而是 %T", value)
    }
    fmt.Println(text)
}

压缩、命名和关闭时最容易踩的坑

对象键不是文件名。notes 是文件内部的对象键，notes.root 才是操作系统中的文件名。读回时要保持键名完全一致。

不要在 Put 成功后跳过 Close。ROOT 文件有文件头、记录和索引信息，关闭阶段是完成文件生命周期的重要一步。遇到关闭错误时，应把它当成发布文件失败处理，而不是只看 Put 是否返回 nil。

压缩是创建选项，不是对象写入 API。如果需要调整压缩策略，应根据当前 groot 文档选择对应的 riofs.FileOption，再传给 groot.Create；不要把压缩参数塞进 Put。

相关问题

Go groot 能不能直接写入复杂结构？

可以，但对象必须符合 groot 支持的 ROOT 对象模型。复杂树、数组或自定义对象应先确认对应包的接口与 streamer 支持，再决定是写入对象还是使用树结构。

为什么 Get 找不到刚写入的对象？

优先检查键名是否一致、写入是否真的返回成功，以及文件是否在读取前完成关闭。若对象类型断言失败，再检查读取端使用的 ROOT 类型是否与写入端匹配。