diff --git a/cmd/commands/cmd_import_mission_control.go b/cmd/commands/cmd_import_mission_control.go index 1fb63c859..ad1626f07 100644 --- a/cmd/commands/cmd_import_mission_control.go +++ b/cmd/commands/cmd_import_mission_control.go @@ -31,6 +31,12 @@ var importMissionControlCommand = cli.Command{ Name: "force", Usage: "whether to force the history entry import", }, + &cli.StringFlag{ + Name: "mission_control_namespace", + Usage: "an optional mission control namespace to import " + + "into. If not specified, the default " + + "mission control namespace will be used.", + }, }, } @@ -94,7 +100,8 @@ func importMissionControl(ctx *cli.Context) error { Pairs: []*routerrpc.PairHistory{ importResult, }, - Force: ctx.IsSet("force"), + Force: ctx.IsSet("force"), + MissionControlNamespace: ctx.String("mission_control_namespace"), } rpcCtx := context.Background() diff --git a/cmd/commands/cmd_mission_control.go b/cmd/commands/cmd_mission_control.go index fe4acb25c..aae176d5c 100644 --- a/cmd/commands/cmd_mission_control.go +++ b/cmd/commands/cmd_mission_control.go @@ -273,7 +273,15 @@ var queryMissionControlCommand = cli.Command{ Name: "querymc", Category: "Mission Control", Usage: "Query the internal mission control state.", - Action: actionDecorator(queryMissionControl), + Flags: []cli.Flag{ + &cli.StringFlag{ + Name: "mission_control_namespace", + Usage: "an optional mission control namespace to query. " + + "If not specified, the default mission control " + + "namespace will be used.", + }, + }, + Action: actionDecorator(queryMissionControl), } func queryMissionControl(ctx *cli.Context) error { @@ -283,7 +291,9 @@ func queryMissionControl(ctx *cli.Context) error { client := routerrpc.NewRouterClient(conn) - req := &routerrpc.QueryMissionControlRequest{} + req := &routerrpc.QueryMissionControlRequest{ + MissionControlNamespace: ctx.String("mission_control_namespace"), + } snapshot, err := client.QueryMissionControl(ctxc, req) if err != nil { return err @@ -299,8 +309,16 @@ var queryProbCommand = cli.Command{ Category: "Mission Control", Usage: "Deprecated. Estimate a success probability.", ArgsUsage: "from-node to-node amt", - Action: actionDecorator(queryProb), - Hidden: true, + Flags: []cli.Flag{ + &cli.StringFlag{ + Name: "mission_control_namespace", + Usage: "an optional mission control namespace to query. " + + "If not specified, the default mission control " + + "namespace will be used.", + }, + }, + Action: actionDecorator(queryProb), + Hidden: true, } func queryProb(ctx *cli.Context) error { @@ -336,9 +354,10 @@ func queryProb(ctx *cli.Context) error { client := routerrpc.NewRouterClient(conn) req := &routerrpc.QueryProbabilityRequest{ - FromNode: fromNode[:], - ToNode: toNode[:], - AmtMsat: int64(amtMsat), + FromNode: fromNode[:], + ToNode: toNode[:], + AmtMsat: int64(amtMsat), + MissionControlNamespace: ctx.String("mission_control_namespace"), } response, err := client.QueryProbability(ctxc, req) @@ -355,7 +374,15 @@ var resetMissionControlCommand = cli.Command{ Name: "resetmc", Category: "Mission Control", Usage: "Reset internal mission control state.", - Action: actionDecorator(resetMissionControl), + Flags: []cli.Flag{ + &cli.StringFlag{ + Name: "mission_control_namespace", + Usage: "an optional mission control namespace to reset. " + + "If not specified, the default mission control " + + "namespace will be used.", + }, + }, + Action: actionDecorator(resetMissionControl), } func resetMissionControl(ctx *cli.Context) error { @@ -365,7 +392,9 @@ func resetMissionControl(ctx *cli.Context) error { client := routerrpc.NewRouterClient(conn) - req := &routerrpc.ResetMissionControlRequest{} + req := &routerrpc.ResetMissionControlRequest{ + MissionControlNamespace: ctx.String("mission_control_namespace"), + } _, err := client.ResetMissionControl(ctxc, req) return err