lncli: add mission control import command

This commit is contained in:
carla 2021-03-18 10:46:47 +02:00
parent d0e7164c07
commit cb977622c6
No known key found for this signature in database
GPG Key ID: 4CA7FE54A6213C91
2 changed files with 95 additions and 0 deletions

@ -0,0 +1,94 @@
package main
import (
"context"
"errors"
"fmt"
"strconv"
"github.com/lightningnetwork/lnd/lnrpc/routerrpc"
"github.com/lightningnetwork/lnd/routing/route"
"github.com/urfave/cli"
)
const argsStr = "[source node] [dest node] [unix ts seconds] [amount in msat]"
var importMissionControlCommand = cli.Command{
Name: "importmc",
Category: "Payments",
Usage: "Import a result to the internal mission control state.",
ArgsUsage: fmt.Sprintf("importmc %v", argsStr),
Action: actionDecorator(importMissionControl),
Flags: []cli.Flag{
cli.BoolFlag{
Name: "failure",
Usage: "whether the routing history entry was a failure",
},
},
}
func importMissionControl(ctx *cli.Context) error {
conn := getClientConn(ctx, false)
defer conn.Close()
if ctx.NArg() != 4 {
return fmt.Errorf("please provide args: %v", argsStr)
}
args := ctx.Args()
sourceNode, err := route.NewVertexFromStr(args[0])
if err != nil {
return fmt.Errorf("please provide valid source node: %v", err)
}
destNode, err := route.NewVertexFromStr(args[1])
if err != nil {
return fmt.Errorf("please provide valid dest node: %v", err)
}
ts, err := strconv.ParseInt(args[2], 10, 64)
if err != nil {
return fmt.Errorf("please provide unix timestamp "+
"in seconds: %v", err)
}
if ts <= 0 {
return errors.New("please provide positive timestamp")
}
amt, err := strconv.ParseInt(args[3], 10, 64)
if err != nil {
return fmt.Errorf("please provide amount in msat: %v", err)
}
if amt <= 0 {
return errors.New("amount must be >0")
}
client := routerrpc.NewRouterClient(conn)
importResult := &routerrpc.PairHistory{
NodeFrom: sourceNode[:],
NodeTo: destNode[:],
History: &routerrpc.PairData{},
}
if ctx.IsSet("failure") {
importResult.History.FailAmtMsat = amt
importResult.History.FailTime = ts
} else {
importResult.History.SuccessAmtMsat = amt
importResult.History.SuccessTime = ts
}
req := &routerrpc.XImportMissionControlRequest{
Pairs: []*routerrpc.PairHistory{
importResult,
},
}
rpcCtx := context.Background()
_, err = client.XImportMissionControl(rpcCtx, req)
return err
}

@ -6,6 +6,7 @@ import "github.com/urfave/cli"
func routerCommands() []cli.Command {
return []cli.Command{
queryMissionControlCommand,
importMissionControlCommand,
queryProbCommand,
resetMissionControlCommand,
buildRouteCommand,