如何使用 GORM 获取数据表所有字段

如何使用 GORM 获取数据表所有字段

在使用 GORM 进行数据库操作时，有时我们需要获取数据表的所有字段，以便进行后续的处理。本文将介绍如何使用 GORM 获取数据表的所有字段。

1. 获取数据表的所有字段

要获取数据表的所有字段，我们可以使用 GORM 的 Model 方法和反射机制。具体步骤如下：

type User struct {
    ID        uint
    Name      string
    Age       int
    CreatedAt time.Time
    UpdatedAt time.Time
}

func main() {
    db, err := gorm.Open("mysql", "user:password@tcp(127.0.0.1:3306)/db_name?charset=utf8mb4&parseTime=True&loc=Local")
    if err != nil {
        panic(err)
    }
    defer db.Close()

    var columns []string
    modelType := reflect.TypeOf(User{})
    for i := 0; i < modelType.NumField(); i++ {
        columns = append(columns, SnakeString(modelType.Field(i).Name))
    }
    fmt.Println(columns)
}

// SnakeString 驼峰转蛇形 XxYy to xx_yy , XxYY to xx_y_y
func SnakeString(s string) string {
	data := make([]byte, 0, len(s)*2)
	j := false
	num := len(s)
	for i := 0; i < num; i++ {
		d := s[i]
		// or通过ASCII码进行大小写的转化
		// 65-90（A-Z），97-122（a-z）
		//判断如果字母为大写的A-Z就在前面拼接一个_
		if i > 0 && d >= 'A' && d <= 'Z' && j {
			data = append(data, '_')
		}
		if d != '_' {
			j = true
		}
		data = append(data, d)
	}
	//ToLower把大写字母统一转小写
	return strings.ToLower(string(data[:]))
}

在上面的代码中，我们定义了一个 User 结构体，它包含了数据表中的所有字段。然后，我们使用 reflect.TypeOf(User{}) 方法获取 User 对应的类型，再使用 NumField 方法获取字段数量，最后使用 Field 方法获取每个字段的名称，并将其存储在 columns 切片中。

2. 结果输出

运行上面的代码，我们可以得到以下输出：

[id name age created_at updated_at]

可以看到，输出结果包含了数据表中的所有字段名。

3. 总结

本文介绍了如何使用 GORM 获取数据表的所有字段。通过使用 Model 方法和反射机制，我们可以轻松地获取数据表的所有字段名，方便后续的处理。