如何在GORM中实现自定义字符串枚举类型到数据库文本列的映射

你遇到的报错是因为GORM默认无法识别你自定义的files.MIMEType类型与数据库varchar字段的映射关系，GORM本身就支持通过实现标准库的driver.Valuer和sql.Scanner接口来完成自定义类型的读写转换，具体实现方案如下：

给MIMEType类型添加接口实现方法即可，示例代码：

package files

import (
    "database/sql/driver"
    "errors"
)

type MIMEType string

const (
    MIMETypeJPG MIMEType = "image/jpeg"
    MIMETypePDF MIMEType = "application/pdf"
    MIMETypePNG MIMEType = "image/png"
    MIMETypeTXT MIMEType = "text/plain"
)

// Value 实现 driver.Valuer 接口，写入数据库时自动调用
func (m MIMEType) Value() (driver.Value, error) {
    return string(m), nil
}

// Scan 实现 sql.Scanner 接口，从数据库读取数据时自动调用
func (m *MIMEType) Scan(value interface{}) error {
    strVal, ok := value.(string)
    if !ok {
        return errors.New("failed to scan MIMEType: value is not string")
    }
    *m = MIMEType(strVal)

    // 可选：增加枚举合法性校验，避免非法值写入结构体
    // switch *m {
    // case MIMETypeJPG, MIMETypePDF, MIMETypePNG, MIMETypeTXT:
    // default:
    //     return errors.New("invalid MIME type: " + strVal)
    // }
    return nil
}

添加完上述方法后不需要修改现有Document结构体定义，GORM会自动完成类型转换：写入时将MIMEType转成字符串存入content_type字段，查询时自动把数据库的字符串转成MIMEType类型赋值给结构体属性，Create、查询等操作都可以正常运行。

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