feat: support sqlite_fields_comment

server/utility/db/db.go

@@ -0,0 +1,54 @@
+package db
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/gogf/gf/v2/database/gdb"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+// 获取数据库表字段及注释
+// usage:
+//
+// fields, err := db.GetFieldsWithComment(ctx, in.Table, in.Name)
+// if err != nil {
+// 	return
+// }
+// for _, v := range fields {}
+func GetFieldsWithComment(ctx context.Context, tableName, dbTag string) (fields map[string]*gdb.TableField, err error) {
+	db := g.DB(dbTag)
+	fields, err = db.TableFields(ctx, tableName) // 使用 goframe 框架本身已完美支持 mysql 获取表字段及注释
+	dbConf := db.GetConfig()
+	switch dbConf.Type {
+	case "sqlite":
+		fields, err = fixSqliteFieldsComment(ctx, tableName, db, fields)
+	}
+	return
+}
+
+type TableComment struct {
+	Name    string `json:"name"`
+	Comment string `json:"comment"`
+}
+
+// 获取数据库表字段及注释
+func GetTablesWithComment(ctx context.Context, dbTag string) (tables []*TableComment, err error) {
+	db := g.DB(dbTag)
+	dbConf := db.GetConfig()
+	switch dbConf.Type {
+	case "mysql":
+		sql := "SELECT TABLE_NAME as name, TABLE_COMMENT as comment FROM information_schema.`TABLES` WHERE TABLE_SCHEMA = '%s'"
+		if err = db.Ctx(ctx).Raw(fmt.Sprintf(sql, dbConf.Name)).Scan(&tables); err != nil {
+			return
+		}
+	case "sqlite":
+		var tableNames []string
+		tableNames, err = db.Tables(ctx)
+		if err != nil {
+			return
+		}
+		tables, err = transSqliteTablesComment(ctx, tableNames, db)
+	}
+	return
+}