RetailCRM-Mirror/mg-transport-core
1
0
Fork 0
mirror of https://github.com/retailcrm/mg-transport-core.git synced 2025-02-16 12:53:13 +03:00
Code Issues Projects Releases Wiki Activity

Update migration generator to v2 migrations

Browse Source 
* update migration generator to v2 migrations
* use %w verb
This commit is contained in:
Pavel 2022-05-13 13:31:21 +03:00 committed by GitHub
parent 27aee62487
commit 8c5a614c93
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 23 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
core/db/migration_generator.go
View File

@ -5,21 +5,21 @@ import (
"os"
"path"
"strconv"
"strings"
"text/template"
"time"
)
var migrationTemplate = `package $package
var migrationTemplate = `package {{.Package}}
import (
"github.com/jinzhu/gorm"
"github.com/retailcrm/mg-transport-core/core"
"github.com/retailcrm/mg-transport-core/v2/core/db"
"gopkg.in/gormigrate.v1"
)
func init() {
core.Migrations().Add(&gormigrate.Migration{
ID: "$version",
db.Migrations().Add(&gormigrate.Migration{
ID: "{{.Version}}",
Migrate: func(db *gorm.DB) error {
// Write your migration code here...
},
@ -30,6 +30,12 @@ func init() {
}
`
// MigrationData contains base variables for the new migration.
type MigrationData struct {
Package string
Version string
}
// NewMigrationCommand struct.
type NewMigrationCommand struct {
Directory string `short:"d" long:"directory" default:"./migrations" description:"Directory where migration will be created"` // nolint:lll
@ -46,30 +52,30 @@ func (x *NewMigrationCommand) FileExists(filename string) bool {
// Execute migration generator command.
func (x *NewMigrationCommand) Execute(args []string) error {
version := strconv.FormatInt(time.Now().Unix(), 10)
tpl, err := template.New("migration").Parse(migrationTemplate)
if err != nil {
return fmt.Errorf("fatal: cannot parse base migration template: %w", err)
}
directory := path.Clean(x.Directory)
packageName := "migrations"
migrationData := MigrationData{
Package: "migrations",
Version: strconv.FormatInt(time.Now().Unix(), 10),
}
if _, err := os.Stat(directory); os.IsNotExist(err) {
return fmt.Errorf("err: specified directory doesn't exist")
}
if base := path.Base(directory); base != "/" && base != "." {
packageName = base
migrationData.Package = base
}
filePath := path.Join(directory, version+"_app.go")
filePath := path.Join(directory, migrationData.Version+"_app.go")
if x.FileExists(filePath) {
return fmt.Errorf("\"%s\" already exists or it's a directory", filePath)
}
migrationData := strings.Replace(
strings.Replace(migrationTemplate, "$version", version, 1),
"$package",
packageName,
1,
)
file, err := os.Create(filePath)
if err != nil {
return err
@ -77,7 +83,7 @@ func (x *NewMigrationCommand) Execute(args []string) error {
defer file.Close()
if _, err := file.WriteString(migrationData); err != nil {
if err := tpl.Execute(file, migrationData); err != nil {
return err
}