Golang sql行。用于所有通用类型的所有字段的scan函数

Question

我想从 sql 软件包中使用scan（）函数来执行一个可能（或不）返回多行的选择语句，并在我的函数中返回这些结果。

我是Golang Generics 的新手，并且对如何实现这一目标感到困惑。 通常，我们会在scan函数上使用*SQL.Rows，并提供对我们的所有预期“结果类型”字段的引用，我们想将行读入，例如：

var alb Album
rows.Scan(&alb.ID, &alb.Title, &alb.Artist,
            &alb.Price, &alb.Quantity)

专辑是一个结构类型，其中显示了这五个字段。

现在，为了不为我拥有的每个SQL表编写类似的函数n次，我想改用 generic type r 。 r是通用接口类型结果，我将此类型定义为n个不同的结构之一：

type Result interface {
    StructA | StructB | StructC
}

func ExecSelect[R Result](conn *sql.DB, cmd Command, template R) []R

我现在如何编写lows.scan（...）在所有字段上应用扫描操作我对R的混凝土类型的结构？例如，我想拥有lows.scan（＆amp; res.field1，＆amp; res.field2，...） res属于R类型R，并且扫描应接收我当前混凝土的所有字段类型R。我实际上需要提供一个“模板”作为R的混凝土类型的参数，以便在运行时清楚哪个结构现在相关？

请在考虑仿制药的任何错误上纠正我。

Answer 1

这是仿制药的糟糕用例。

函数的参数 sql.rows.sql.rows.scan 应该是扫描目的地，即您的结构字段，结果集中的每一列之一，在通用功能主体中您 > r 类型参数。

即使您这样做了，结果中的结构约束可能具有不同的字段...？那么，您如何设想与不同的字段一起使用的通用代码？

您可以使用提供任意结构扫描的软件包来完成想要的工作.structscan“ rel =” nofollow noreferrer“> structscan ，但这使用引擎盖下的反射将struct字段映射到sql.rows.scan.scan.scan.scan.scan参数，参数，参数，参数，参数，参数>因此，您对仿制药根本没有任何好处。

如果有的话，您会使情况变得更糟，因为现在您拥有使用类型参数的其他性能开销。

Answer 2

另一个答案显示了如何创建

对于每种类型，您需要一个函数来创建指向新值和功能的指针，以尊重指针。让我们声明所需功能的接口：

type FieldTyper interface {
    // Return pointer to new value.
    New() (ptr any)
    // Dereference pointer.
    Deref(ptr any) (val any)
}

创建该接口的通用实现：

type FieldType[T any] struct{}

func (v FieldType[T]) New() any {
    return new(T)
}

func (v FieldType[T]) Deref(p any) any {
    return *p.(*T)
}

创建列名称为fieldTypers：

var fieldTypes = map[string]FieldTyper{
    "id":   FieldType[Type_int]{},
    "name": FieldType[Type_string]{},
}

使用字段打字器设置扫描ARGS并尊重这些args并添加到映射。

func SetupScanArgs(columnNames []string, fieldTypes map[string]FieldTyper) []any {
    args := make([]any, len(columnNames))
    for i, n := range columnNames {
        args[i] = fieldTypes[n].New()
    }
    return args
}

func ArgsToValueMap(columnNames []string, fieldTypes map[string]FieldTyper, args []any) map[string]any {
    result := make(map[string]any)
    for i, n := range columnNames {
        result[n] = fieldTypes[n].Deref(args[i])
    }
    return result
}

这样的扫描：

args := SetupScanArgs(columnNames, fieldTypes)
if err := rows.Scan(args...); err != nil {
    return err
}
m := ArgsToValueMap(columnNames, fieldTypes, args)

Answer 3

使用反射软件包：

func scan(rows *sql.Rows, types map[string]reflect.Type) (map[string]any, error) {
    names, _ := rows.Columns()
    ptrs := make([]any, len(names))
    vals := make([]reflect.Value, len(names))
    for i, n := range names {
        vals[i] = reflect.New(types[n])
        ptrs[i] = vals[i].Interface()
    }
    if err := rows.Scan(ptrs...); err != nil {
        return nil, err
    }
    result := make(map[string]any)
    for i, v := range vals {
        result[names[i]] = v.Elem().Interface()
    }
    return result, nil
}

其中类型就是这样：

types = map[string]reflect.Type{
   "id": reflect.TypeOf(Type_int(0)),
   "name": reflect.TypeOf(Type_string("")),
}