diff --git a/backend/script/gorm_gen/generate.go b/backend/script/gorm_gen/generate.go index 1fbc6244d..85a98965f 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,17 @@ 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) + + 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() +}