From 552111517e86f085fe7655f72b696be34d37e3a0 Mon Sep 17 00:00:00 2001 From: Gnanakeethan Balasubramaniam Date: Sun, 16 Jul 2017 08:13:08 +0530 Subject: [PATCH] Revert "Update: complementary fix for removing calling ddlSpec on migration" The odds of getting this perfectly up is not good. This reverts commit 3673659178f5925f103e62e25fd3f67ae9e13f09. --- generate/g_migration.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/generate/g_migration.go b/generate/g_migration.go index 4014a2e..adb13d0 100644 --- a/generate/g_migration.go +++ b/generate/g_migration.go @@ -203,10 +203,12 @@ func GenerateMigration(mname, upsql, downsql, curpath string) { fpath := path.Join(migrationFilePath, fmt.Sprintf("%s_%s.go", today, mname)) if f, err := os.OpenFile(fpath, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666); err == nil { defer utils.CloseFile(f) + ddlSpec := "" spec := "" up := "" down := "" if DDL != "" { + ddlSpec = "m.ddlSpec()" switch strings.Title(DDL.String()) { case "Create": spec = strings.Replace(DDLSpecCreate, "{{StructName}}", utils.CamelCase(mname)+"_"+today, -1) @@ -222,6 +224,7 @@ func GenerateMigration(mname, upsql, downsql, curpath string) { } header := strings.Replace(MigrationHeader, "{{StructName}}", utils.CamelCase(mname)+"_"+today, -1) + header = strings.Replace(header, "{{ddlSpec}}", ddlSpec, -1) header = strings.Replace(header, "{{CurrTime}}", today, -1) f.WriteString(header + spec + up + down) // Run 'gofmt' on the generated source code @@ -247,6 +250,7 @@ const ( func init() { m := &{{StructName}}{} m.Created = "{{CurrTime}}" + {{ddlSpec}} migration.Register("{{StructName}}", m) } `