如何使用反射和 GORM 实现动态创建和修改数据库表？

动态生成数据库表并修改字段

在构建cms管理系统时，我们需要一个功能，可以在无需人工干预的情况下动态创建和修改数据表。因为orm通常使用提前定义字段结构的方式来映射数据库表，对于动态需求来说会遇到一定障碍。

为了解决这个问题，我们可以利用以下原理：

先锋多用户商城系统

修改自网上仿乐购商城，新增功能：1、数据库在线备份与导入功能，可以随时备份数据库，数据受损可以导入数据库，确保数据安全；2、增加组合商品概念，可以用于组配商品销售（比如外套有蓝色和红色，鞋子有40码和41码等），买一送一、组合销售（比如上衣＋围巾＋长裙做为一个套装商品）和加价购买等销售方式；3、按照商品重量和送货距离实时计算精确运费，并可在订单中予以显示，使运费金额实现实时动态准确显示、清晰明了；

下载

• 反射structof：动态创建自定义结构体。
• gorm.automigrate：根据结构体自动创建或修改数据库表。

具体实现：

首先，我们需要了解如何根据数据库中的字段信息动态创建结构体。以下代码示例演示了如何从数据库表中获取字段信息并将其映射到反射结构体：

type Table struct {
    TableName    string
    SelectFields string
    FieldNames   []string
    FieldTypes   []string
    FieldDefault []string
    FieldIsNull  []bool
    DataType     reflect.Type
}

// newTable 从数据库获取字段信息并创建反射结构体
func newTable(db *sql.DB, object interface{}, name string) Table {
    // 获取字段信息
    rows, err := db.Query("SELECT COLUMN_NAME, data_type,column_default,is_nullable FROM information_schema.COLUMNS WHERE TABLE_NAME = '" + name + "' ORDER BY ordinal_position;")
    if err != nil {
        panic(err)
    }
    tb := Table{TableName: name}

    var (
        field, typ, colDefault, isNull string
    )

    // 遍历字段信息并构造反射结构体
    for rows.Next() {
        rows.Scan(&field, &typ, &colDefault, &isNull)
        tb.FieldNames = append(tb.FieldNames, field)
        tb.FieldTypes = append(tb.FieldTypes, tableTypes[typ])
        tb.FieldDefault = append(tb.FieldDefault, colDefault)
        tb.FieldIsNull = append(tb.FieldIsNull, isNull == "YES")
    }

    // 组装查询字段字符串
    tb.SelectFields = "`" + strings.Join(tb.FieldNames, "`,`") + "`"

    // 根据字段类型映射到反射类型
    iValue := reflect.ValueOf(object)
    for iValue.Kind() == reflect.Ptr || iValue.Kind() == reflect.Interface {
        iValue = iValue.Elem()
    }
    if iValue.Kind() == reflect.Struct || iValue.Kind() == reflect.Map {
        tb.DataType = iValue.Type()
    } else {
        // 定义新的反射结构体字段，包含字段名、类型和标签
        var sf []reflect.StructField
        for i := 0; i < len(tb.FieldNames); i++ {
            name := strings.ToLower(tb.FieldNames[i])
            sf = append(sf, reflect.StructField{