highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-12-08 20:03:10 +01:00
Code Issues Packages Projects Releases Wiki Activity

lncli: cancel RPC context on OS interrupts

Browse Source
This commit is contained in:
whythat
2020-07-18 18:46:16 +03:00
committed by Elle Mouton
parent fa4155c126
commit 54c93b1b86
19 changed files with 211 additions and 181 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
cmd/lncli/wtclient.go
View File

@@ -1,7 +1,6 @@
package main
import (
"context"
"encoding/hex"
"errors"
"fmt"
@@ -51,6 +50,8 @@ var addTowerCommand = cli.Command{
}
func addTower(ctx *cli.Context) error {
ctxc := getContext()
// Display the command's help message if the number of arguments/flags
// is not what we expect.
if ctx.NArg() != 1 || ctx.NumFlags() > 0 {
@@ -74,7 +75,7 @@ func addTower(ctx *cli.Context) error {
Pubkey: pubKey,
Address: address,
}
resp, err := client.AddTower(context.Background(), req)
resp, err := client.AddTower(ctxc, req)
if err != nil {
return err
}
@@ -96,6 +97,8 @@ var removeTowerCommand = cli.Command{
}
func removeTower(ctx *cli.Context) error {
ctxc := getContext()
// Display the command's help message if the number of arguments/flags
// is not what we expect.
if ctx.NArg() != 1 || ctx.NumFlags() > 0 {
@@ -130,7 +133,7 @@ func removeTower(ctx *cli.Context) error {
Pubkey: pubKey,
Address: address,
}
resp, err := client.RemoveTower(context.Background(), req)
resp, err := client.RemoveTower(ctxc, req)
if err != nil {
return err
}
@@ -153,6 +156,8 @@ var listTowersCommand = cli.Command{
}
func listTowers(ctx *cli.Context) error {
ctxc := getContext()
// Display the command's help message if the number of arguments/flags
// is not what we expect.
if ctx.NArg() > 0 || ctx.NumFlags() > 1 {
@@ -165,7 +170,7 @@ func listTowers(ctx *cli.Context) error {
req := &wtclientrpc.ListTowersRequest{
IncludeSessions: ctx.Bool("include_sessions"),
}
resp, err := client.ListTowers(context.Background(), req)
resp, err := client.ListTowers(ctxc, req)
if err != nil {
return err
}
@@ -190,6 +195,8 @@ var getTowerCommand = cli.Command{
}
func getTower(ctx *cli.Context) error {
ctxc := getContext()
// Display the command's help message if the number of arguments/flags
// is not what we expect.
if ctx.NArg() != 1 || ctx.NumFlags() > 1 {
@@ -211,7 +218,7 @@ func getTower(ctx *cli.Context) error {
Pubkey: pubKey,
IncludeSessions: ctx.Bool("include_sessions"),
}
resp, err := client.GetTowerInfo(context.Background(), req)
resp, err := client.GetTowerInfo(ctxc, req)
if err != nil {
return err
}
@@ -227,6 +234,8 @@ var statsCommand = cli.Command{
}
func stats(ctx *cli.Context) error {
ctxc := getContext()
// Display the command's help message if the number of arguments/flags
// is not what we expect.
if ctx.NArg() > 0 || ctx.NumFlags() > 0 {
@@ -237,7 +246,7 @@ func stats(ctx *cli.Context) error {
defer cleanUp()
req := &wtclientrpc.StatsRequest{}
resp, err := client.Stats(context.Background(), req)
resp, err := client.Stats(ctxc, req)
if err != nil {
return err
}
@@ -264,6 +273,8 @@ var policyCommand = cli.Command{
}
func policy(ctx *cli.Context) error {
ctxc := getContext()
// Display the command's help message if the number of arguments/flags
// is not what we expect.
if ctx.NArg() > 0 || ctx.NumFlags() > 1 {
@@ -288,7 +299,7 @@ func policy(ctx *cli.Context) error {
req := &wtclientrpc.PolicyRequest{
PolicyType: policyType,
}
resp, err := client.Policy(context.Background(), req)
resp, err := client.Policy(ctxc, req)
if err != nil {
return err
}