From fc16d6cf8107a925245d3bec6d4f374f5666a932 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=8B=A4=E5=BF=AB=E7=9A=84=E5=B0=8F=E6=99=B4=E5=90=8C?=
 =?UTF-8?q?=E5=AD=A6?= <941403820@qq.com>
Date: Sat, 17 Feb 2024 04:32:57 +0000
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=AD=97=E6=AE=B5=E4=B8=A2?=
 =?UTF-8?q?=E5=A4=B1=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: 勤快的小晴同学 <941403820@qq.com>
---
 apps/develop/gen/gen.go | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/apps/develop/gen/gen.go b/apps/develop/gen/gen.go
index 1f3f871..7d7346d 100644
--- a/apps/develop/gen/gen.go
+++ b/apps/develop/gen/gen.go
@@ -148,6 +148,8 @@ func (g *Generator) Generate() (entity.DevGenTable, error) {
 	data.FunctionAuthor = "panda"
 
 	var wg sync.WaitGroup
+	columnChan := make(chan entity.DevGenTableColumn, len(dbColumn)) // 创建带缓冲的通道
+	
 	for x := 0; x < len(dbColumn); x++ {
 		index := x
 		wg.Add(1)
@@ -239,7 +241,7 @@ func (g *Generator) Generate() (entity.DevGenTable, error) {
 			} else {
 				column.IsRequired = "0"
 				column.IsInsert = "1"
-				if strings.Index(column.ColumnName, "name") >= 0 || strings.Index(column.ColumnName, "status") >= 0 {
+				if strings.Contains(column.ColumnName, "name") || strings.Contains(column.ColumnName, "status") {
 					column.IsRequired = "1"
 				}
 			}
@@ -278,10 +280,17 @@ func (g *Generator) Generate() (entity.DevGenTable, error) {
 				column.HtmlType = "select"
 			}
 
-			data.Columns = append(data.Columns, column)
+			columnChan <- column // 将处理结果放入通道
 		}(&wg, index)
 	}
 	wg.Wait()
+
+	close(columnChan) // 关闭通道
+
+	for column := range columnChan {
+		data.Columns = append(data.Columns, column) // 将通道中的结果放入data.Columns
+	}
+
 	return data, nil
 }