Go语言中如何用结构体多类型值填充sqlmock.AddRow可变参数

当然可以实现！我给你分享两种实用方案，帮你摆脱手动拼接sqlmock参数的繁琐工作，完美实现你想要的结构体自动映射效果。

方案一：反射自动映射（灵活通用） #

利用Go的反射机制，我们可以自动遍历结构体的字段，提取对应的数据库列名和值。这种方案的好处是结构体字段更新时，不需要同步修改映射代码，非常适合字段较多或经常变动的场景。

首先给你的Account结构体添加结构体标签（用来映射数据库列名，避免字段名和列名不一致），然后实现两个方法：

import (
    "database/sql/driver"
    "reflect"
    "time"
)

type Account struct {
    ID                     string    `db:"id"`
    UpdatedAt              time.Time `db:"updated_at"`
    CreatedAt              time.Time `db:"created_at"`
    Meta                   string    `db:"meta"`
    AccountID              string    `db:"account_id"`
    Currency               string    `db:"currency"`
    Nickname               string    `db:"nickname"`
    SchemeName             string    `db:"scheme_name"`
    Identification         string    `db:"identification"`
    Name                   string    `db:"name"`
    IdentificationSecondary string   `db:"identification_secondary"`
    ServicerSchemeName     string    `db:"servicer_scheme_name"`
    ServicerIdentification string    `db:"servicer_identification"`
    InstitutionID          string    `db:"institution_id"`
    ClientID               string    `db:"client_id"`
}

// GetFields 提取数据库列名切片
func (a *Account) GetFields() []string {
    fields := make([]string, 0, reflect.TypeOf(*a).NumField())
    t := reflect.TypeOf(*a)
    
    for i := 0; i < t.NumField(); i++ {
        field := t.Field(i)
        // 优先使用结构体标签中的列名，没有则用字段名
        colName := field.Tag.Get("db")
        if colName == "" {
            colName = field.Name
        }
        fields = append(fields, colName)
    }
    return fields
}

// GetValues 提取对应driver.Value类型的值切片
func (a *Account) GetValues() []driver.Value {
    values := make([]driver.Value, 0, reflect.ValueOf(*a).NumField())
    v := reflect.ValueOf(*a)
    
    for i := 0; i < v.NumField(); i++ {
        fieldVal := v.Field(i).Interface()
        // 转换为driver.Value，Go基本类型（string、time.Time等）都默认支持
        dv, ok := fieldVal.(driver.Value)
        if !ok {
            // 处理不支持的自定义类型，这里可以根据需求返回错误或panic
            panic("unsupported type for driver.Value: " + reflect.TypeOf(fieldVal).Name())
        }
        values = append(values, dv)
    }
    return values
}

使用的时候只需要这样写（注意AddRow需要用...展开切片）：

account := Account{
    ID:                     "ABCD12436364",
    UpdatedAt:              time.Now(),
    CreatedAt:              time.Now(),
    Meta:                   "{}",
    AccountID:              "12345678",
    Currency:               "GBP",
    Nickname:               "Test",
    SchemeName:             "Scheme",
    Identification:         "12345676534263",
    Name:                   "New account",
    IdentificationSecondary: "12345",
    ServicerSchemeName:     "scheme",
    ServicerIdentification: "test id",
    InstitutionID:          "ABCD123456",
    ClientID:               "ZXCVB12436",
}

// 生成Mock行
rows := sqlmock.NewRows(account.GetFields()).AddRow(account.GetValues()...)

方案二：手动实现（性能可控） #

如果你的结构体字段固定，或者担心反射的性能开销，可以选择手动实现GetFields和GetValues方法。这种方案性能更好，可控性强，但结构体字段变动时需要同步修改方法代码。

func (a *Account) GetFields() []string {
    return []string{
        "id",
        "updated_at",
        "created_at",
        "meta",
        "account_id",
        "currency",
        "nickname",
        "scheme_name",
        "identification",
        "name",
        "identification_secondary",
        "servicer_scheme_name",
        "servicer_identification",
        "institution_id",
        "client_id",
    }
}

func (a *Account) GetValues() []driver.Value {
    return []driver.Value{
        a.ID,
        a.UpdatedAt,
        a.CreatedAt,
        a.Meta,
        a.AccountID,
        a.Currency,
        a.Nickname,
        a.SchemeName,
        a.Identification,
        a.Name,
        a.IdentificationSecondary,
        a.ServicerSchemeName,
        a.ServicerIdentification,
        a.InstitutionID,
        a.ClientID,
    }
}

使用方式和方案一完全一致。

注意事项 #

• 结构体字段必须首字母大写（导出字段），否则反射无法访问到字段值。
• 确保结构体字段的类型实现了driver.Value接口：Go的基本类型（string、int、time.Time等）都默认支持，如果有自定义类型，需要自己实现driver.Value的Value() (driver.Value, error)方法。
• 结构体标签db:"column_name"可以让你灵活映射结构体字段到数据库列名，避免字段名和列名不一致的问题。

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