部分判断优化

Signed-off-by: 勤快的小晴同学 <941403820@qq.com>
This commit is contained in:
勤快的小晴同学
2024-03-11 01:50:13 +00:00
committed by Gitee
parent bb0ec5a25a
commit 0d3bae0001
6 changed files with 17 additions and 47 deletions

View File

@@ -85,36 +85,33 @@ func (s *toolsGenTableColumn) IsNotQuery(name string) bool {
func (s *toolsGenTableColumn) CheckNameColumn(columnName string) bool {
if len(columnName) >= 4 {
if tmp := columnName[len(columnName)-4:]; tmp == "name" {
return true
}
tmp := columnName[len(columnName)-4:]
return tmp == "name"
}
return false
}
func (s *toolsGenTableColumn) CheckStatusColumn(columnName string) bool {
if len(columnName) >= 6 {
if tmp := columnName[len(columnName)-6:]; tmp == "status" {
return true
}
tmp := columnName[len(columnName)-6:]
return tmp == "status"
}
return false
}
func (s *toolsGenTableColumn) CheckTypeColumn(columnName string) bool {
if len(columnName) >= 4 {
if tmp := columnName[len(columnName)-4:]; tmp == "type" {
return true
}
tmp := columnName[len(columnName)-4:]
return tmp == "type"
}
return false
}
func (s *toolsGenTableColumn) CheckSexColumn(columnName string) bool {
if len(columnName) >= 3 {
if tmp := columnName[len(columnName)-3:]; tmp == "sex" {
return true
}
tmp := columnName[len(columnName)-4:]
return tmp == "sex"
}
return false
}
@@ -149,7 +146,7 @@ func (g *Generator) Generate() (entity.DevGenTable, error) {
var wg sync.WaitGroup
columnChan := make(chan entity.DevGenTableColumn, len(dbColumn)) // 创建带缓冲的通道
for x := 0; x < len(dbColumn); x++ {
index := x
wg.Add(1)