fix: 仅主键才生成InsertGetXX的辅助函数

go-home-admin · Jun 16, 2023 · 081a0cf · 081a0cf
1 parent 329a19b
commit 081a0cf
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/console/commands/orm/mysql.go b/console/commands/orm/mysql.go
@@ -134,10 +134,12 @@ func genFieldFunc(table string, columns []tableColumn) string {
 				goType = "*" + goType
 			}
 			// if 主键, 生成In, > <
-			str += "\nfunc (orm *Orm" + TableName + ") InsertGet" + column.ColumnName + "(row *" + TableName + ") " + goType + " {" +
-				"\n\torm.db.Create(row)" +
-				"\n\treturn row." + column.ColumnName +
-				"\n}"
+			if column.COLUMN_KEY == "PRI" {
+				str += "\nfunc (orm *Orm" + TableName + ") InsertGet" + column.ColumnName + "(row *" + TableName + ") " + goType + " {" +
+					"\n\torm.db.Create(row)" +
+					"\n\treturn row." + column.ColumnName +
+					"\n}"
+			}
 
 			str += "\nfunc (orm *Orm" + TableName + ") Where" + column.ColumnName + "In(val []" + column.GoType + ") *Orm" + TableName + " {" +
 				"\n\torm.db.Where(\"`" + column.COLUMN_NAME + "` IN ?\", val)" +