diff --git a/pkg/cmd/migrate.go b/pkg/cmd/migrate.go index 6e5c3d3865..27712f5cf3 100644 --- a/pkg/cmd/migrate.go +++ b/pkg/cmd/migrate.go @@ -7,6 +7,7 @@ import ( "github.com/fatih/color" "github.com/jzelinskie/cobrautil/v2" + "github.com/jzelinskie/cobrautil/v2/cobraotel" "github.com/spf13/cobra" crdbmigrations "github.com/authzed/spicedb/internal/datastore/crdb/migrations" @@ -30,6 +31,9 @@ func RegisterMigrateFlags(cmd *cobra.Command) { cmd.Flags().String("datastore-mysql-table-prefix", "", "prefix to add to the name of all mysql database tables") cmd.Flags().Uint64("migration-backfill-batch-size", 1000, "number of items to migrate per iteration of a datastore backfill") cmd.Flags().Duration("migration-timeout", 1*time.Hour, "defines a timeout for the execution of the migration, set to 1 hour by default") + + otel := cobraotel.New(cmd.Use) + otel.RegisterFlags(cmd.Flags()) termination.RegisterFlags(cmd.Flags()) runtime.RegisterFlags(cmd.Flags()) }