golang查询数据库时如果有很多列且每次查询列数不固定，scan方法应该怎么写？

Question

如果有张表有102列，后边100列列名有规律，如t1~t100，如果每次要查询很多列，且每次列数不固定，sql语句是好拼接的，但是rows.scan方法应该怎么写？或者有没有其他更新好方法？

Answer 1

可以使用反射动态解析查询结果，并将解析后的数据存储到一个结构体中，这样可以避免手动编写大量的 scan 代码。

首先定义一个结构体，结构体的字段需要与表中的列名相同或者通过 struct tag 指定与表中列名对应的字段名，例如：

type MyRow struct {
    Col1 string `db:"col1"`
    Col2 int    `db:"col2"`
    Col3 string `db:"t1"`
    Col4 string `db:"t2"`
    // ...
    Col100 string `db:"t100"`
}

在查询时，先构造一个 SELECT 子句，例如：

cols := []string{"col1", "col2", "t1", "t2", /* ... */, "t100"}
selectClause := "SELECT " + strings.Join(cols, ", ") + " FROM my_table WHERE ..."

然后执行查询，将查询结果存储到一个 []MyRow 类型的变量中：

rows, err := db.Query(selectClause)
if err != nil {
    // handle error
}

defer rows.Close()

var myRows []MyRow

for rows.Next() {
    var r MyRow
    if err := rows.Scan(&r.Col1, &r.Col2, &r.Col3, &r.Col4, /* ... */, &r.Col100); err != nil {
        // handle error
    }
    myRows = append(myRows, r)
}

if err := rows.Err(); err != nil {
    // handle error
}

上述代码中，rows.Scan 中的参数可以通过反射动态生成，例如：

var r MyRow
rv := reflect.ValueOf(&r).Elem()
cols := make([]interface{}, rv.NumField())
for i := 0; i < rv.NumField(); i++ {
    cols[i] = rv.Field(i).Addr().Interface()
}
if err := rows.Scan(cols...); err != nil {
    // handle error
}

这样可以更方便地扩展结构体中的字段，而不需要修改大量的 scan 代码。