如何扩展go-yaml以支持自定义标签

package main

import (
    "fmt"
    "gopkg.in/yaml.v2"
    "reflect"
)

type CustomStruct struct {
    Value string `yaml:"custom_tag"`
}

func (c *CustomStruct) UnmarshalYAML(unmarshal func(interface{}) error) error {
    // 定义一个临时结构体，用于解析yaml数据
    type tempStruct struct {
        Value string `yaml:"custom_tag"`
    }

    // 解析yaml数据到临时结构体
    temp := tempStruct{}
    if err := unmarshal(&temp); err != nil {
        return err
    }

    // 将临时结构体的值赋给自定义结构体
    c.Value = temp.Value

    return nil
}

func (c *CustomStruct) MarshalYAML() (interface{}, error) {
    // 定义一个临时结构体，用于转换为yaml数据
    type tempStruct struct {
        Value string `yaml:"custom_tag"`
    }

    // 将自定义结构体的值赋给临时结构体
    temp := tempStruct{
        Value: c.Value,
    }

    return temp, nil
}

func main() {
    yamlData := `
custom_tag: test value
`

    custom := CustomStruct{}
    err := yaml.Unmarshal([]byte(yamlData), &custom)
    if err != nil {
        fmt.Println("Unmarshal error:", err)
        return
    }

    fmt.Println("Value:", custom.Value)

    data, err := yaml.Marshal(&custom)
    if err != nil {
        fmt.Println("Marshal error:", err)
        return
    }

    fmt.Println("YAML data:", string(data))
}