mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-16 01:44:03 +03:00
f0bef2b482
PR-URL: https://github.com/hasura/graphql-engine-mono/pull/2509 Co-authored-by: Aravind K P <8335904+scriptonist@users.noreply.github.com> GitOrigin-RevId: 03ee878072b58c6dc7f43af9d26d30a5594d9a50
130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
package commands
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"text/tabwriter"
|
|
|
|
"github.com/hasura/graphql-engine/cli/v2/internal/hasura"
|
|
"github.com/hasura/graphql-engine/cli/v2/internal/metadatautil"
|
|
|
|
"github.com/hasura/graphql-engine/cli/v2/util"
|
|
|
|
"github.com/hasura/graphql-engine/cli/v2"
|
|
"github.com/hasura/graphql-engine/cli/v2/migrate"
|
|
"github.com/pkg/errors"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func newMigrateStatusCmd(ec *cli.ExecutionContext) *cobra.Command {
|
|
opts := &MigrateStatusOptions{
|
|
EC: ec,
|
|
StatusOpts: make(StatusOptions),
|
|
}
|
|
migrateStatusCmd := &cobra.Command{
|
|
Use: "status",
|
|
Short: "Display current status of migrations on a database",
|
|
Example: ` # Use with admin secret:
|
|
hasura migrate status --admin-secret "<your-admin-secret>"
|
|
|
|
# Check status on a different server:
|
|
hasura migrate status --endpoint "<endpoint>"`,
|
|
SilenceUsage: true,
|
|
PreRunE: func(cmd *cobra.Command, args []string) error {
|
|
return validateConfigV3FlagsWithAll(cmd, ec)
|
|
},
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
if err := opts.Run(); err != nil {
|
|
return err
|
|
}
|
|
buf := printStatus(opts.StatusOpts)
|
|
fmt.Fprintf(ec.Stdout, "%s", buf)
|
|
return nil
|
|
},
|
|
}
|
|
return migrateStatusCmd
|
|
}
|
|
|
|
func (o *MigrateStatusOptions) Run() error {
|
|
o.EC.Spin("Fetching migration status...")
|
|
defer o.EC.Spinner.Stop()
|
|
if ec.AllDatabases {
|
|
sourcesAndKind, err := metadatautil.GetSourcesAndKind(o.EC.APIClient.V1Metadata.ExportMetadata)
|
|
if err != nil {
|
|
return fmt.Errorf("got error while getting the sources list : %v", err)
|
|
}
|
|
for _, source := range sourcesAndKind {
|
|
o.Source = cli.Source(source)
|
|
status, err := o.RunOnSource()
|
|
if err != nil {
|
|
return fmt.Errorf("error getting status for database %s: %v", o.Source.Name, err)
|
|
}
|
|
o.StatusOpts[o.Source] = status
|
|
}
|
|
return err
|
|
}
|
|
o.Source = ec.Source
|
|
status, err := o.RunOnSource()
|
|
if err != nil {
|
|
return fmt.Errorf("error getting status for database %s: %v", o.Source.Name, err)
|
|
}
|
|
o.StatusOpts[o.Source] = status
|
|
return err
|
|
}
|
|
|
|
type StatusOptions map[cli.Source]*migrate.Status
|
|
type MigrateStatusOptions struct {
|
|
EC *cli.ExecutionContext
|
|
Source cli.Source
|
|
StatusOpts StatusOptions
|
|
}
|
|
|
|
func (o *MigrateStatusOptions) RunOnSource() (*migrate.Status, error) {
|
|
if o.EC.Config.Version <= cli.V2 {
|
|
o.Source.Name = ""
|
|
o.Source.Kind = hasura.SourceKindPG
|
|
}
|
|
migrateDrv, err := migrate.NewMigrate(o.EC, true, o.Source.Name, o.Source.Kind)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
status, err := executeStatus(migrateDrv)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "cannot fetch migrate status")
|
|
}
|
|
return status, nil
|
|
}
|
|
|
|
func printStatus(statusOpts StatusOptions) *bytes.Buffer {
|
|
out := new(tabwriter.Writer)
|
|
buf := &bytes.Buffer{}
|
|
out.Init(buf, 0, 8, 2, ' ', 0)
|
|
w := util.NewPrefixWriter(out)
|
|
for source, status := range statusOpts {
|
|
if source.Name != "" {
|
|
w.Write(util.LEVEL_0, fmt.Sprintf("\nDatabase: %s\n", source.Name))
|
|
}
|
|
w.Write(util.LEVEL_0, "VERSION\tNAME\tSOURCE STATUS\tDATABASE STATUS\n")
|
|
for _, version := range status.Index {
|
|
w.Write(util.LEVEL_0, "%d\t%s\t%s\t%s\n",
|
|
version,
|
|
status.Migrations[version].Name,
|
|
convertBool(status.Migrations[version].IsPresent),
|
|
convertBool(status.Migrations[version].IsApplied),
|
|
)
|
|
}
|
|
}
|
|
out.Flush()
|
|
return buf
|
|
}
|
|
|
|
func convertBool(ok bool) string {
|
|
switch ok {
|
|
case true:
|
|
return "Present"
|
|
case false:
|
|
return "Not Present"
|
|
}
|
|
return ""
|
|
}
|