Go实现简单的数据库表转结构体

简介: 设计好了数据库表之后最烦的就是又要在代码中建一层实体层然后一个个创建对应表的结构体。关键那么多项目每次都是需要创建一份,所以就使用 go 实现一个简单的将数据库的表转化为结构体。

前言

设计好了数据库表之后最烦的就是又要在代码中建一层实体层然后一个个创建对应表的结构体。关键那么多项目每次都是需要创建一份,所以就使用 go 实现一个简单的将数据库的表转化为结构体。

基本的设计思路也是很简单:

  • 连接数据库
  • 获取数据库中所有的表
  • 获取表的所有的元数据
  • 将元数据转换为字符串,该字符串就是一个 struct的内容并输出结果

设计过程

经常使用 MySQL 的时候,可以通过 SQL 语句查询相关的信息,比如查看数据库的状态、数据库的数据表、数据库服务的状态等等语句。

show databases 查看所有的数据库
show tables [from dbName] 查看数据库的所有表数据
show columns from tableName [from dbName] 查看数据库表的所有元信息
show status 查看数据服务的状态

所以就可以使用以上的 SQL 语句进行相操作,比如获取所有的表,获取相关表的元数据

func getTableInfo(tableName string, db *sql.DB) (metas []SchemaMeta) {
    list, _ := db.Query(fmt.Sprintf("show columns from %s", tableName))

    for list.Next() {
        var data SchemaMeta
        err := list.Scan(&data.Field, &data.Type, &data.Null, &data.Key, &data.Default, &data.Extra)
        if err != nil {
            fmt.Println(err.Error())
        }
        metas = append(metas, data)
    }
    return metas
}

表的元数据获取完之后就可以进行封装成一个字符串。最主要的功能有两个:

  • db 数据类型转 go 数据类型
  • 首字母大小写,

数据类型的转换简单处理的话可以使用枚举的方式将 db 数据类型和 go 的数据类型建立一个map 进行一一对应,当然也可以使用模糊匹配的方式。我这里直接使用截取字符串值去匹配。

因要设置为对象类型,必须设置为公有的,所以结构体名称和字段名称都必须是大写。直接使用截取第一位然后进行大写转换再重新拼接出去。比如 field = strings.ToUpper(field[:1]) + field[1:]。这部分的数据大小写还需要根据自己的需求进行调整,比如有写字段是这样:“user_code”那么转换成字段应该是“UserCode”。

转换完之后就可以输出,一个简单的工具就完成了。

迭代升级

  • 支持多种数据库:现目前使用的是 MySQL 数据库进行开发,而市面上其实有很多种数据,所以这部分可以继续扩展支持多种数据库。
  • 数据类型转换:现只支持自己所使用的相关数据类型
  • 如果有前端搭配使用可以涉及按需生成
目录
相关文章
|
3月前
|
SQL 关系型数据库 MySQL
go如何使用SQLX操作MySQL数据库?
sqlx是Go语言中一款流行的第三方数据库操作包,它扩展了Go标准库`database/sql`的功能,极大地简化了数据库操作流程并提供了丰富的数据库交互方法。
|
16天前
|
存储 Rust Go
Go nil 空结构体 空接口有什么区别?
本文介绍了Go语言中的`nil`、空结构体和空接口的区别。`nil`是预定义的零值变量,适用于指针、管道等类型;空结构体大小为0,多个空结构体实例指向同一地址;空接口由`_type`和`data`字段组成,仅当两者均为`nil`时,空接口才为`nil`。
Go nil 空结构体 空接口有什么区别?
|
11天前
|
SQL 关系型数据库 MySQL
go语言数据库中mysql驱动安装
【11月更文挑战第2天】
26 4
|
12天前
|
SQL 关系型数据库 MySQL
go语言中安装数据库驱动
【11月更文挑战第1天】
34 5
|
13天前
|
SQL 关系型数据库 MySQL
go语言中数据库操作
【10月更文挑战第22天】
27 4
|
12天前
|
关系型数据库 MySQL 数据库连接
go语言中打开数据库连接
【11月更文挑战第1天】
26 2
|
1月前
|
SQL 关系型数据库 MySQL
Go语言项目高效对接SQL数据库:实践技巧与方法
在Go语言项目中,与SQL数据库进行对接是一项基础且重要的任务
55 11
|
2月前
|
Go
Go to Learn Go之结构体
Go to Learn Go之结构体
37 5
|
2月前
|
存储 Shell Go
Go语言结构体和元组全面解析
Go语言结构体和元组全面解析
|
2月前
|
存储 Go
Go: struct 结构体类型和指针【学习笔记记录】
本文是Go语言中struct结构体类型和指针的学习笔记,包括结构体的定义、成员访问、使用匿名字段,以及指针变量的声明使用、指针数组定义使用和函数传参修改值的方法。