如何在YAML文件中正确解析rune类型参数

问题原因 #

常用的Go YAML解析库（如gopkg.in/yaml.v3）默认没有内置rune类型的转换逻辑：YAML中';'会被识别为长度为1的字符串，无法直接映射到Go中作为int32别名的rune类型。另外你给出的结构体tag写法存在语法错误，yaml: "CsvDelimiter"中冒号和引号之间不能有空格，正确写法为yaml:"CsvDelimiter"，tag格式错误也会导致字段无法正常赋值。

解决方案 #

有两种常用的落地方式：

方案1：自定义rune类型实现YAML反序列化接口 #

自定义封装rune类型，实现yaml.Unmarshaler接口，自定义字符串到rune的转换逻辑：

import "fmt"

// 自定义Rune类型，支持YAML反序列化
type Rune rune

func (r *Rune) UnmarshalYAML(unmarshal func(interface{}) error) error {
    var s string
    if err := unmarshal(&s); err != nil {
        return err
    }
    runeSlice := []rune(s)
    if len(runeSlice) != 1 {
        return fmt.Errorf("invalid rune value: expect single character, got %d characters", len(runeSlice))
    }
    *r = Rune(runeSlice[0])
    return nil
}

修改原结构体的字段类型为自定义Rune：

type Params struct {
  // 其他字段
  CsvDelimiter Rune `yaml:"CsvDelimiter"`
  // 其他字段
}

使用时直接强转即可得到原生rune值：rune(params.CsvDelimiter)

方案2：先用string类型接收，解析后手动转换 #

如果不想引入自定义类型，可以先将配置值读入string字段，解析完成后手动转换：

type Params struct {
  // 其他字段
  CsvDelimiterStr string `yaml:"CsvDelimiter"`
  CsvDelimiter    rune
  // 其他字段
}

YAML解析完成后添加转换逻辑：

// 假设已完成YAML解析到params变量
runeSlice := []rune(params.CsvDelimiterStr)
if len(runeSlice) != 1 {
    // 处理非法配置的错误逻辑
}
params.CsvDelimiter = runeSlice[0]

内容的提问来源于stack exchange，提问作者Tier