如何在Go中无需自定义类型实现数据库单字符的Scanner接口？

我尝试用Go结构体从SQLite数据库读取数据，数据库里部分属性以单字符形式存储，因此在Go中用byte（即uint8）类型来对应：

type Foo struct {
    Id   string
    Name string
    Mode byte
}

但执行扫描代码时：

var foo Foo
if err := rows.Scan(&foo.Id, &foo.Name, &foo.Mode); err != nil {
    log.Error().Msgf("Error: %v", err)
    return err
}

出现错误：

sql: Scan error on column index 2, name "mode": converting driver.Value type string ("D") to a uint8: invalid syntax

原因是Go sql包默认不支持直接将字符串类型的单字符扫描到byte类型中。我目前通过自定义uchar类型实现Scanner接口解决了问题：

type uchar byte

type Foo struct {
    Id   string
    Name string
    Mode uchar
}

func (d *uchar) Scan(value any) error {
    log.Debug().Msgf("Scan() given type is %T value is %v", value, value)
    switch v := value.(type) {
    case string:
        if v == "" {
            *d = ' '
        } else {
            *d = uchar(v[0])
        }
    }
    return nil
}

但由于无法为Go内置byte类型添加方法，必须引入自定义类型，因此想咨询：是否存在无需额外自定义类型即可读取并映射数据库单字符的更优方案？

无需自定义类型的解决方案 #

有两种常见的替代方案可以避免自定义类型：

1. 临时变量中转扫描

先将字段扫描到临时string变量，再转换为byte：

var foo Foo
var modeStr string
if err := rows.Scan(&foo.Id, &foo.Name, &modeStr); err != nil {
    log.Error().Msgf("Error: %v", err)
    return err
}
// 处理空字符串的情况
if modeStr != "" {
    foo.Mode = modeStr[0]
} else {
    foo.Mode = ' ' // 可替换为你需要的默认值
}

这种方式简单直接，无需额外定义类型，适合字段数量不多的场景。

2. 使用SQL查询转换

在SQL查询时直接将单字符转换为字节值，比如SQLite中可以用unicode()函数获取字符的ASCII值：

SELECT id, name, unicode(mode) AS mode FROM foo_table;

之后在Go中直接扫描到byte类型：

var foo Foo
if err := rows.Scan(&foo.Id, &foo.Name, &foo.Mode); err != nil {
    log.Error().Msgf("Error: %v", err)
    return err
}

注意：如果数据库中mode为空字符串，unicode()会返回NULL，扫描时需处理该情况——可以将Mode定义为*byte，或者在查询时用COALESCE(unicode(mode), 32)（32为空格的ASCII值）设置默认值。

方案对比 #

• 临时变量方案：代码直观，无需修改SQL，适合小范围使用；缺点是若存在多个单字符字段，会增加重复代码。
• SQL转换方案：Go代码更简洁，无需额外处理逻辑；缺点是依赖数据库函数，迁移到其他数据库时可能需要调整SQL。

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