From e9502e1ac8b824e9943be93a354f9be873ec1e2e Mon Sep 17 00:00:00 2001
From: huangzj <Huangzj@123>
Date: Mon, 25 May 2020 18:08:17 +0800
Subject: [PATCH] =?UTF-8?q?feat(ReverseEngine)=EF=BC=9A=E6=8F=90=E4=BA=A4?=
 =?UTF-8?q?=E5=B0=8F=E4=BC=98=E5=8C=96=EF=BC=8C=E5=87=86=E5=A4=87=E4=B8=8B?=
 =?UTF-8?q?=E7=8F=AD=EF=BC=8C=E5=9B=9E=E5=AE=B6=E5=BC=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 reverse/Engine.go |  6 ++++--
 util/DbUtil.go    | 22 ++++++++++++++++++++++
 2 files changed, 26 insertions(+), 2 deletions(-)

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
+}