From 200c5af771dbf99367594032cefba49d73263dfe Mon Sep 17 00:00:00 2001 From: yanghaiquan Date: Wed, 13 Aug 2025 17:31:30 +0800 Subject: [PATCH 1/2] MOD: add generateForFoundation to gorm_gen Change-Id: Ie835a76edb803997f984d0bab7aeeb528071a493 --- backend/script/gorm_gen/generate.go | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) diff --git a/backend/script/gorm_gen/generate.go b/backend/script/gorm_gen/generate.go index 1fbc6244d..7cb0ee266 100644 --- a/backend/script/gorm_gen/generate.go +++ b/backend/script/gorm_gen/generate.go @@ -20,6 +20,7 @@ func main() { generateForEvaluationEvaluator(db) generateForEvaluationExpt(db) generateForObservability(db) + generateForFoundation(db) } func initDB() *gorm.DB { @@ -202,3 +203,30 @@ func generateForObservability(db *gorm.DB) { g.ApplyBasic(observabilityView) g.Execute() } + +func generateForFoundation(db *gorm.DB) { + path := "modules/foundation/infra/repo/mysql/gorm_gen" + g := gen.NewGenerator(getGenerateConfig(path)) + g.UseDB(db) + tables := []string{ + "space", + "space_user", + "user", + "api_key", + } + + var models []any + titleCaser := cases.Title(language.English) + for _, tn := range tables { + parts := strings.Split(tn, "_") + for i := range parts { + if len(parts[i]) > 0 { + parts[i] = titleCaser.String(parts[i]) + } + } + name := strings.Join(parts, "") + models = append(models, g.GenerateModelAs(tn, name)) + } + g.ApplyBasic(models...) + g.Execute() +} From 937ba6873093f3e58d61a37566c412ccf8963770 Mon Sep 17 00:00:00 2001 From: yanghaiquan Date: Wed, 13 Aug 2025 18:51:53 +0800 Subject: [PATCH 2/2] MOD: to upper APIKey Change-Id: I5fa9fa16b3f35c69ad8023a858b2ee64af6dcab3 --- backend/script/gorm_gen/generate.go | 25 ++++++------------------- 1 file changed, 6 insertions(+), 19 deletions(-) diff --git a/backend/script/gorm_gen/generate.go b/backend/script/gorm_gen/generate.go index 7cb0ee266..85a98965f 100644 --- a/backend/script/gorm_gen/generate.go +++ b/backend/script/gorm_gen/generate.go @@ -208,25 +208,12 @@ func generateForFoundation(db *gorm.DB) { path := "modules/foundation/infra/repo/mysql/gorm_gen" g := gen.NewGenerator(getGenerateConfig(path)) g.UseDB(db) - tables := []string{ - "space", - "space_user", - "user", - "api_key", - } - var models []any - titleCaser := cases.Title(language.English) - for _, tn := range tables { - parts := strings.Split(tn, "_") - for i := range parts { - if len(parts[i]) > 0 { - parts[i] = titleCaser.String(parts[i]) - } - } - name := strings.Join(parts, "") - models = append(models, g.GenerateModelAs(tn, name)) - } - g.ApplyBasic(models...) + spaceModel := g.GenerateModelAs("space", "Space") + spaceUserModel := g.GenerateModelAs("space_user", "SpaceUser") + userModel := g.GenerateModelAs("user", "User") + apiKeyModel := g.GenerateModelAs("api_key", "APIKey") + + g.ApplyBasic(spaceModel, spaceUserModel, userModel, apiKeyModel) g.Execute() }