regolancer/payment.go

104 lines
3.2 KiB
Go
Raw Normal View History

2022-08-20 14:11:45 +00:00
package main
import (
"context"
"fmt"
"log"
"os"
2022-08-20 14:11:45 +00:00
"time"
"github.com/lightningnetwork/lnd/lnrpc"
"github.com/lightningnetwork/lnd/lnrpc/routerrpc"
)
2022-08-20 17:02:25 +00:00
type ErrRetry struct {
amount int64
}
func (e ErrRetry) Error() string {
return fmt.Sprintf("retry payment with %d sats", e.amount)
}
var ErrProbeFailed = fmt.Errorf("probe failed")
func (r *regolancer) createInvoice(ctx context.Context, from, to uint64, amount int64) (*lnrpc.AddInvoiceResponse, error) {
return r.lnClient.AddInvoice(ctx, &lnrpc.Invoice{Value: amount,
2022-08-20 14:11:45 +00:00
Memo: fmt.Sprintf("Rebalance %d ⇒ %d", from, to),
Expiry: int64(time.Hour.Seconds() * 24)})
}
2022-08-22 20:01:46 +00:00
func (r *regolancer) pay(ctx context.Context, invoice *lnrpc.AddInvoiceResponse, amount int64,
minAmount int64, route *lnrpc.Route, probeSteps int) error {
2022-08-21 09:57:47 +00:00
fmt.Println()
defer fmt.Println()
2022-08-20 14:11:45 +00:00
lastHop := route.Hops[len(route.Hops)-1]
lastHop.MppRecord = &lnrpc.MPPRecord{
PaymentAddr: invoice.PaymentAddr,
TotalAmtMsat: amount * 1000,
}
result, err := r.routerClient.SendToRouteV2(ctx,
2022-08-20 14:11:45 +00:00
&routerrpc.SendToRouteRequest{
PaymentHash: invoice.RHash,
Route: route,
})
if err != nil {
return err
}
if result.Status == lnrpc.HTLCAttempt_FAILED {
nodeCtx, cancel := context.WithTimeout(ctx, time.Minute)
defer cancel()
node1, err := r.getNodeInfo(nodeCtx, route.Hops[result.Failure.FailureSourceIndex-1].PubKey)
2022-08-20 14:11:45 +00:00
node1name := ""
node2name := ""
if err != nil {
node1name = fmt.Sprintf("node%d", result.Failure.FailureSourceIndex-1)
} else {
node1name = node1.Node.Alias
}
node2, err := r.getNodeInfo(nodeCtx, route.Hops[result.Failure.FailureSourceIndex].PubKey)
2022-08-20 14:11:45 +00:00
if err != nil {
node2name = fmt.Sprintf("node%d", result.Failure.FailureSourceIndex)
} else {
node2name = node2.Node.Alias
}
2022-08-21 09:57:47 +00:00
log.Printf("%s %s ⇒ %s", faintWhiteColor(result.Failure.Code.String()),
2022-08-20 14:11:45 +00:00
cyanColor(node1name), cyanColor(node2name))
2022-08-25 16:57:56 +00:00
if probeSteps > 0 && int(result.Failure.FailureSourceIndex) == len(route.Hops)-2 &&
result.Failure.Code == lnrpc.Failure_TEMPORARY_CHANNEL_FAILURE {
2022-08-20 17:02:25 +00:00
fmt.Println("Probing route...")
2022-08-22 20:01:46 +00:00
min := int64(0)
start := amount / 2
if minAmount > 0 && minAmount < amount {
min = -minAmount - 1
start = minAmount
}
maxAmount, err := r.probeRoute(ctx, route, min, amount, start, probeSteps, params.EconRatio)
2022-08-20 17:02:25 +00:00
if err != nil {
return err
}
if maxAmount == 0 {
return ErrProbeFailed
}
2022-08-20 22:15:39 +00:00
return ErrRetry{amount: maxAmount}
2022-08-20 17:02:25 +00:00
}
2022-08-20 14:11:45 +00:00
return fmt.Errorf("error: %s @ %d", result.Failure.Code.String(), result.Failure.FailureSourceIndex)
} else {
2022-08-20 15:04:49 +00:00
log.Printf("Success! Paid %s in fees", hiWhiteColor(result.Route.TotalFeesMsat/1000))
if r.statFilename != "" {
_, err := os.Stat(r.statFilename)
f, ferr := os.OpenFile(r.statFilename, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
if ferr != nil {
log.Print(errColorF("Error saving rebalance stats to %s: %s", r.statFilename, ferr))
return nil
}
defer f.Close()
if os.IsNotExist(err) {
f.WriteString("timestamp,from_channel,to_channel,amount_msat,fees_msat\n")
}
f.Write([]byte(fmt.Sprintf("%d,%d,%d,%d,%d\n", time.Now().Unix(), route.Hops[0].ChanId,
lastHop.ChanId, route.TotalAmtMsat-route.TotalFeesMsat, route.TotalFeesMsat)))
}
2022-08-20 14:11:45 +00:00
return nil
}
}