diff --git a/reverse/Engine.go b/reverse/Engine.go index 90030a4..dc5669f 100644 --- a/reverse/Engine.go +++ b/reverse/Engine.go @@ -8,6 +8,7 @@ import ( "ReverseEngine/util" "bufio" "database/sql" + "fmt" _ "github.com/go-sql-driver/mysql" "gopkg.in/ini.v1" "log" @@ -146,8 +147,9 @@ func dbReverse(path string, tableName string, dbName string, db *sql.DB) { if !result { return } - cms := util.FindColumnMessage(dbName, tableName, db) //查询数据库表字段信息 - dErr := db.Close() //关闭数据库连接 + cms := util.FindColumnMessage(dbName, tableName, db) //查询数据库表字段信息 + tableComment := util.FindTableComment(dbName, tableName, db) //查询数据库表字段信息 + dErr := db.Close() //关闭数据库连接 checkError(dErr, "数据库关闭失败:") if nil == cms { log.Println("没有查到数据库字段信息略过") diff --git a/util/DbUtil.go b/util/DbUtil.go index 79713b8..4e05f7e 100644 --- a/util/DbUtil.go +++ b/util/DbUtil.go @@ -3,6 +3,7 @@ package util import ( "ReverseEngine/entity" "database/sql" + "fmt" "strings" ) @@ -30,3 +31,24 @@ func FindColumnMessage(dbName string, tableName string, db *sql.DB) []entity.Col return cms } + +func FindTableComment(dbName string, tableName string, db *sql.DB) string { + + sql := fmt.Sprintf("select table_comment from information_schema.tables where table_schema = '%s' and table_name ='%s'", dbName, tableName) + row, err := db.Query(sql) + if err != nil { + panic("数据库查询出错:" + err.Error()) + } + fmt.Println(row) + var s string + for row.Next() { + err := row.Scan(&s) + if err != nil { + panic("字段赋值错误了" + err.Error()) + return "" + } + break + } + + return s +}