如何在Go语言中通过结构体生成数据库列名常量？

1. 手动定义常量（简单靠谱） #

直接给结构体对应的数据库列名定义常量，和结构体同步维护就行——虽然要手动写，但完全可控，没有额外运行时开销：

type myThingy struct {
    Col1 string `db:"col1"` // 列名和字段名不一致时，用tag明确标注
    Col2 int    `db:"col2"`
}

// 对应数据库列名的常量
const (
    Col1 = "col1"
    Col2 = "col2"
)

func queryThingy(col1Val string) (*myThingy, error) {
    var thing myThingy
    err := db.QueryRowf("SELECT * FROM thingy WHERE %s = ?", Col1, col1Val).Scan(&thing.Col1, &thing.Col2)
    return &thing, err
}

2. 用反射动态提取（适合批量场景） #

要是想从结构体自动获取列名，用Go的reflect包就能实现，但要注意运行时开销和安全风险：

import (
    "reflect"
    "strings"
    "fmt"
)

// 从结构体提取列名：优先取db tag，无tag则转小写字段名
func getColumnNames(t interface{}) map[string]string {
    cols := make(map[string]string)
    val := reflect.ValueOf(t).Elem()
    typ := val.Type()

    for i := 0; i < typ.NumField(); i++ {
        field := typ.Field(i)
        colName := field.Tag.Get("db")
        if colName == "" {
            colName = strings.ToLower(field.Name)
        }
        cols[field.Name] = colName
    }
    return cols
}

// 使用示例
func queryThingy(fieldName string, colVal interface{}) (*myThingy, error) {
    cols := getColumnNames(&myThingy{})
    colName, ok := cols[fieldName]
    if !ok {
        return nil, fmt.Errorf("无效字段名: %s", fieldName)
    }

    // 必须加白名单校验！防止SQL注入
    allowedCols := map[string]bool{Col1: true, Col2: true}
    if !allowedCols[colName] {
        return nil, fmt.Errorf("不允许的列名: %s", colName)
    }

    var thing myThingy
    err := db.QueryRowf(fmt.Sprintf("SELECT * FROM thingy WHERE %s = ?", colName), colVal).Scan(&thing.Col1, &thing.Col2)
    return &thing, err
}

⚠️ 重点提醒：反射方式一定要做列名白名单校验，绝对不能直接把外部传入的字段名拼进SQL，否则会有严重的SQL注入风险。

3. 代码生成（自动化最佳选择） #

如果项目中有大量结构体需要处理，写个小工具自动生成列名常量才是最优解——省得手动维护容易出错：

• 自定义生成器：用反射遍历结构体字段，读取db tag后自动输出包含列名常量的go文件。
• 现成工具：参考stringer的思路，或者直接用GORM这类ORM框架，它本身会自动解析结构体tag对应列名，不用额外处理。

比如自定义生成器的核心逻辑（伪代码）：

import (
    "strings"
    "fmt"
    "reflect"
)

func generateColConstants(typ reflect.Type) string {
    var buf strings.Builder
    buf.WriteString("package yourpkg\n\nconst (\n")
    for i := 0; i < typ.NumField(); i++ {
        field := typ.Field(i)
        colName := field.Tag.Get("db")
        if colName == "" {
            colName = strings.ToLower(field.Name)
        }
        buf.WriteString(fmt.Sprintf("    %s = \"%s\"\n", field.Name, colName))
    }
    buf.WriteString(")\n")
    return buf.String()
}

每次修改结构体后，运行生成器就能自动更新常量，省心又准确。

必注意的关键点 #

• SQL注入防护：无论用哪种方式，都不能直接将外部输入的列名拼入SQL语句，必须做合法性校验。
• 字段-列名映射：当数据库列名和Go字段名命名规则不一致时，一定要用db:"xxx"这类tag明确标注，避免自动转换出错。

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