public/PandaX
1
0
Fork 0
mirror of https://gitee.com/XM-GO/PandaX.git synced 2026-04-23 02:48:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化生成逻辑

Browse Source 
Signed-off-by: lixxxww <941403820@qq.com>
This commit is contained in:
lixxxww
2024-01-22 05:26:33 +00:00
committed by Gitee
parent 043327d9b4
commit 1f661627c2
2 changed files with 26 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
apps/develop/api/table.go
View File

@@ -99,7 +99,10 @@ func (g *GenTableApi) Insert(rc *restfulx.ReqCtx) {
wg.Add(1)
go func(table string) {
defer wg.Done()
genTable, err := gen.ToolsGenTableColumn.GenTableInit(table)
var tg = gen.Generator{
TableName: table,
}
genTable, err := tg.Generate()
if err != nil {
biz.ErrIsNil(err, "创建表结构")
}

40
apps/develop/gen/gen.go
View File

@@ -22,6 +22,8 @@ import (
var ToolsGenTableColumn = toolsGenTableColumn{}
type toolsGenTableColumn struct{}
var (
ColumnTypeStr = []string{"char", "varchar", "narchar", "varchar2", "tinytext", "text", "mediumtext", "longtext"}
ColumnTypeTime = []string{"datetime", "time", "date", "timestamp", "timestamptz"}
@@ -31,8 +33,6 @@ var (
ColumnNameNotQuery = []string{"create_by", "update_by", "create_time", "update_time", "delete_time", "remark"}
)
type toolsGenTableColumn struct{}
func (s *toolsGenTableColumn) GetDbType(columnType string) string {
if strings.Index(columnType, "(") > 0 {
return columnType[0:strings.Index(columnType, "(")]
@@ -119,10 +119,14 @@ func (s *toolsGenTableColumn) CheckSexColumn(columnName string) bool {
return false
}
func (s *toolsGenTableColumn) GenTableInit(tableName string) (entity.DevGenTable, error) {
type Generator struct {
TableName string
}
func (g *Generator) Generate() (entity.DevGenTable, error) {
var data entity.DevGenTable
data.TableName = tableName
tableNameList := strings.Split(tableName, "_")
data.TableName = g.TableName
tableNameList := strings.Split(g.TableName, "_")
for i := 0; i < len(tableNameList); i++ {
strStart := string([]byte(tableNameList[i])[:1])
strEnd := string([]byte(tableNameList[i])[1:])
@@ -134,9 +138,9 @@ func (s *toolsGenTableColumn) GenTableInit(tableName string) (entity.DevGenTable
}
data.PackageName = "system"
data.TplCategory = "crud"
data.ModuleName = strings.Replace(tableName, "_", "-", -1)
data.ModuleName = strings.Replace(g.TableName, "_", "-", -1)
dbColumn, err := services.DevTableColumnModelDao.FindDbTableColumnList(tableName)
dbColumn, err := services.DevTableColumnModelDao.FindDbTableColumnList(g.TableName)
if err != nil {
return data, err
}
@@ -173,18 +177,18 @@ func (s *toolsGenTableColumn) GenTableInit(tableName string) (entity.DevGenTable
}
dataType := strings.ToLower(column.ColumnType)
if s.IsStringObject(dataType) {
if ToolsGenTableColumn.IsStringObject(dataType) {
column.GoType = "string"
columnLength := s.GetColumnLength(column.ColumnType)
columnLength := ToolsGenTableColumn.GetColumnLength(column.ColumnType)
if columnLength >= 500 {
column.HtmlType = "textarea"
} else {
column.HtmlType = "input"
}
} else if s.IsTimeObject(dataType) {
} else if ToolsGenTableColumn.IsTimeObject(dataType) {
column.GoType = "Time"
column.HtmlType = "datetime"
} else if s.IsNumberObject(dataType) {
} else if ToolsGenTableColumn.IsNumberObject(dataType) {
column.HtmlType = "input"
t := ""
if global.Conf.Server.DbType == "postgresql" {
@@ -229,7 +233,7 @@ func (s *toolsGenTableColumn) GenTableInit(tableName string) (entity.DevGenTable
}
}
if s.IsNotEdit(column.ColumnName) {
if ToolsGenTableColumn.IsNotEdit(column.ColumnName) {
column.IsRequired = "0"
column.IsInsert = "0"
} else {
@@ -240,7 +244,7 @@ func (s *toolsGenTableColumn) GenTableInit(tableName string) (entity.DevGenTable
}
}
if s.IsNotEdit(column.ColumnName) {
if ToolsGenTableColumn.IsNotEdit(column.ColumnName) {
column.IsEdit = "0"
} else {
if column.IsPk == "1" {
@@ -250,27 +254,27 @@ func (s *toolsGenTableColumn) GenTableInit(tableName string) (entity.DevGenTable
}
}
if s.IsNotList(column.ColumnName) {
if ToolsGenTableColumn.IsNotList(column.ColumnName) {
column.IsList = "0"
} else {
column.IsList = "1"
}
if s.IsNotQuery(column.ColumnName) {
if ToolsGenTableColumn.IsNotQuery(column.ColumnName) {
column.IsQuery = "0"
} else {
column.IsQuery = "1"
}
if s.CheckNameColumn(column.ColumnName) {
if ToolsGenTableColumn.CheckNameColumn(column.ColumnName) {
column.QueryType = "LIKE"
} else {
column.QueryType = "EQ"
}
if s.CheckStatusColumn(column.ColumnName) {
if ToolsGenTableColumn.CheckStatusColumn(column.ColumnName) {
column.HtmlType = "radio"
} else if s.CheckTypeColumn(column.ColumnName) || s.CheckSexColumn(column.ColumnName) {
} else if ToolsGenTableColumn.CheckTypeColumn(column.ColumnName) || ToolsGenTableColumn.CheckSexColumn(column.ColumnName) {
column.HtmlType = "select"
}