mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-15 17:31:56 +03:00
980c65dbe2
### Description This PR introduces three new features: - Support for a new migrations folder structure. - Add `squash` command in preview. - ~List of migrations on the Console and ability to squash them from console.~ #### New migrations folder structure Starting with this commit, Hasura CLI supports a new directory structure for migrations folder and defaults to that for all new migrations created. Each migration will get a new directory with the name format `timestamp_name` and inside the directory, there will be four files: ```bash └── migrations ├── 1572237730898_squashed │ ├── up.sql │ ├── up.yaml │ ├── down.yaml │ └── down.sql ``` Existing files old migration format `timestamp_name.up|down.yaml|sql` will continue to work alongside new migration files. #### Squash command Lots of users have expressed their interest in squashing migrations (see #2724 and #2254) and some even built [their own tools](https://github.com/domasx2/hasura-squasher) to do squash. In this PR, we take a systematic approach to squash migrations. A new command called `migrate squash` is introduced. Note that this command is in **PREVIEW** and the correctness of squashed migration is not guaranteed (especially for down migrations). From our tests, **it works for most use cases**, but we have found some issues with squashing all the down migrations, partly because the console doesn't generate down migrations for all actions. Hence, until we add an extensive test suite for squashing, we'll keep the command in preview. We recommend you to confirm the correctness yourself by diffing the SQL and Metadata before and after applying the squashed migrations (we're also thinking about embedding some checks into the command itself). ```bash $ hasura migrate squash --help (PREVIEW) Squash multiple migrations leading upto the latest one into a single migration file Usage: hasura migrate squash [flags] Examples: # NOTE: This command is in PREVIEW, correctness is not guaranteed and the usage may change. # squash all migrations from version 1572238297262 to the latest one: hasura migrate squash --from 1572238297262 Flags: --from uint start squashing form this version --name string name for the new squashed migration (default "squashed") --delete-source delete the source files after squashing without any confirmation ``` ### Affected components <!-- Remove non-affected components from the list --> - CLI ### Related Issues <!-- Please make sure you have an issue associated with this Pull Request --> <!-- And then add `(close #<issue-no>)` to the pull request title --> <!-- Add the issue number below (e.g. #234) --> Close #2724, Close #2254, ### Solution and Design <!-- How is this issue solved/fixed? What is the design? --> <!-- It's better if we elaborate --> For the squash command, a state machine is implemented to track changes to Hasura metadata. After applying each action on the metadata state, a list of incremental changes is created. ### Steps to test and verify 1. Open console via cli and create some migrations. 2. Run `hasura migrate squash --from <version>` ### Limitations, known bugs & workarounds <!-- Limitations of the PR, known bugs and suggested workarounds --> <!-- Feel free to delete these comment lines --> - The `squash` command is in preview - Support for squashing from the console is WIP - Support for squashing migrations that are not committed yet is planned. - Un-tracking or dropping a table will cause inconsistent squashed down migration since console doesn't generate correct down migration. - If cascade setting is set to `true` on any of the metadata action, generated migration may be wrong
144 lines
4.1 KiB
Go
144 lines
4.1 KiB
Go
package commands
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
"text/tabwriter"
|
|
|
|
"github.com/hasura/graphql-engine/cli"
|
|
"github.com/hasura/graphql-engine/cli/util"
|
|
"github.com/pkg/errors"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
|
|
mig "github.com/hasura/graphql-engine/cli/migrate/cmd"
|
|
)
|
|
|
|
func newMigrateSquashCmd(ec *cli.ExecutionContext) *cobra.Command {
|
|
v := viper.New()
|
|
opts := &migrateSquashOptions{
|
|
EC: ec,
|
|
}
|
|
migrateSquashCmd := &cobra.Command{
|
|
Use: "squash",
|
|
Short: "(PREVIEW) Squash multiple migrations into a single one",
|
|
Long: "(PREVIEW) Squash multiple migrations leading upto the latest one into a single migration file",
|
|
Example: ` # NOTE: This command is in PREVIEW, correctness is not guaranteed and the usage may change.
|
|
|
|
# squash all migrations from version 123 to the latest one:
|
|
hasura migrate squash --from 123`,
|
|
SilenceUsage: true,
|
|
PreRunE: func(cmd *cobra.Command, args []string) error {
|
|
ec.Viper = v
|
|
return ec.Validate()
|
|
},
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
opts.newVersion = getTime()
|
|
return opts.run()
|
|
},
|
|
}
|
|
|
|
f := migrateSquashCmd.Flags()
|
|
f.Uint64Var(&opts.from, "from", 0, "start squashing form this version")
|
|
f.StringVar(&opts.name, "name", "squashed", "name for the new squashed migration")
|
|
f.BoolVar(&opts.deleteSource, "delete-source", false, "delete the source files after squashing without any confirmation")
|
|
|
|
f.String("endpoint", "", "http(s) endpoint for Hasura GraphQL Engine")
|
|
f.String("admin-secret", "", "admin secret for Hasura GraphQL Engine")
|
|
f.String("access-key", "", "access key for Hasura GraphQL Engine")
|
|
f.MarkDeprecated("access-key", "use --admin-secret instead")
|
|
|
|
// need to create a new viper because https://github.com/spf13/viper/issues/233
|
|
v.BindPFlag("endpoint", f.Lookup("endpoint"))
|
|
v.BindPFlag("admin_secret", f.Lookup("admin-secret"))
|
|
v.BindPFlag("access_key", f.Lookup("access-key"))
|
|
|
|
return migrateSquashCmd
|
|
}
|
|
|
|
type migrateSquashOptions struct {
|
|
EC *cli.ExecutionContext
|
|
|
|
from uint64
|
|
name string
|
|
newVersion int64
|
|
|
|
deleteSource bool
|
|
}
|
|
|
|
func (o *migrateSquashOptions) run() error {
|
|
o.EC.Logger.Warnln("This command is currently experimental and hence in preview, correctness of squashed migration is not guaranteed!")
|
|
o.EC.Spin(fmt.Sprintf("Squashing migrations from %d to latest...", o.from))
|
|
defer o.EC.Spinner.Stop()
|
|
migrateDrv, err := newMigrate(o.EC.MigrationDir, o.EC.ServerConfig.ParsedEndpoint, o.EC.ServerConfig.AdminSecret, o.EC.Logger, o.EC.Version, true)
|
|
if err != nil {
|
|
return errors.Wrap(err, "unable to initialize migrations driver")
|
|
}
|
|
|
|
versions, err := mig.SquashCmd(migrateDrv, o.from, o.newVersion, o.name, o.EC.MigrationDir)
|
|
o.EC.Spinner.Stop()
|
|
if err != nil {
|
|
return errors.Wrap(err, "unable to squash migrations")
|
|
}
|
|
|
|
// squashed migration is generated
|
|
// TODO: capture keyboard interrupt and offer to delete the squashed migration
|
|
|
|
o.EC.Logger.Infof("Created '%d_%s' after squashing '%d' till '%d'", o.newVersion, o.name, versions[0], versions[len(versions)-1])
|
|
|
|
if !o.deleteSource {
|
|
ok := ask2confirmDeleteMigrations(versions, o.EC.Logger)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
}
|
|
|
|
for _, v := range versions {
|
|
delOptions := mig.CreateOptions{
|
|
Version: strconv.FormatInt(v, 10),
|
|
Directory: o.EC.MigrationDir,
|
|
}
|
|
err = delOptions.Delete()
|
|
if err != nil {
|
|
return errors.Wrap(err, "unable to delete source file")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func ask2confirmDeleteMigrations(versions []int64, log *logrus.Logger) bool {
|
|
var s string
|
|
|
|
log.Infof("The following migrations are squashed into a new one:")
|
|
|
|
out := new(tabwriter.Writer)
|
|
buf := &bytes.Buffer{}
|
|
out.Init(buf, 0, 8, 2, ' ', 0)
|
|
w := util.NewPrefixWriter(out)
|
|
for _, version := range versions {
|
|
w.Write(util.LEVEL_0, "%d\n",
|
|
version,
|
|
)
|
|
}
|
|
_ = out.Flush()
|
|
fmt.Println(buf.String())
|
|
log.Infof("Do you want to delete these migration source files? (y/N)")
|
|
|
|
_, err := fmt.Scan(&s)
|
|
if err != nil {
|
|
log.Error("unable to take user input, skipping deleting files")
|
|
return false
|
|
}
|
|
|
|
s = strings.TrimSpace(s)
|
|
s = strings.ToLower(s)
|
|
|
|
if s == "y" || s == "yes" {
|
|
return true
|
|
}
|
|
return false
|
|
}
|