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

package main
import (
	"database/sql"
	"fmt"
	_ "GitHub.com/Go-sql-driver/Mysql"
	"strings"
)

type SchemaMeta struct {
	Field   string
	Type    string
	Null    string
	Key     string
	Default interface{}
	Extra   string
}
func main() {
	dbString := "{YOUR_mysql_CONNECTioN}"
	db, _ := sql.Open("mysql", dbString)
	tables := getTables(db)
	for _, table := range tables {
		metas := getTableInfo(table, db)
		result := changeMetas(table, metas)
		fmt.Println(result)
	}
}
func getTables(db *sql.DB) []string {
	var tables []string
	res, _ := db.Query("SHOW TABLES")
	for res.Next() {
		var table string
		res.Scan(&table)
		tables = append(tables, table)
	}
	return tables
}
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
}
func changeMetas(tableName string, metas []SchemaMeta) string {
	var modelStr string
	for _, val := range metas {
		dataType := "interface{}"
		if val.Type[:3] == "int" {
			dataType = "int"
		} else if val.Type[:7] == "varchar" {
			dataType = "string"
		} else if val.Type[:7] == "tinyint" {
			dataType = "bool"
		} else if val.Type == "datetime" {
			dataType = "time.Time"
		}
		field := val.Field
		field = strings.ToUpper(field[:1]) + field[1:]
		modelStr += fmt.Sprintf("%s %s\n", field, dataType)
	}
	tableName = strings.ToUpper(tableName[:1]) + tableName[1:]
	return fmt.Sprintf("type %s struct {\n %s }", tableName, modelStr)
}

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

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
}