Skip to content

Commit a144cfe

Browse files
committed
cmd/litcli: refactor client conn
As a preparation to add more litd specific RPCs, we refactor the client connection to use the generic interface instead of a specific type.
1 parent f8f60c2 commit a144cfe

File tree

2 files changed

+8
-7
lines changed

2 files changed

+8
-7
lines changed

cmd/litcli/main.go

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,6 @@ import (
88
"strings"
99

1010
terminal "github.com/lightninglabs/lightning-terminal"
11-
"github.com/lightninglabs/lightning-terminal/litrpc"
1211
"github.com/lightninglabs/lndclient"
1312
"github.com/lightninglabs/protobuf-hex-display/jsonpb"
1413
"github.com/lightninglabs/protobuf-hex-display/proto"
@@ -97,7 +96,7 @@ func fatal(err error) {
9796
os.Exit(1)
9897
}
9998

100-
func getClient(ctx *cli.Context) (litrpc.SessionsClient, func(), error) {
99+
func connectClient(ctx *cli.Context) (grpc.ClientConnInterface, func(), error) {
101100
rpcServer := ctx.GlobalString("rpcserver")
102101
tlsCertPath, macPath, err := extractPathArgs(ctx)
103102
if err != nil {
@@ -109,8 +108,7 @@ func getClient(ctx *cli.Context) (litrpc.SessionsClient, func(), error) {
109108
}
110109
cleanup := func() { _ = conn.Close() }
111110

112-
sessionsClient := litrpc.NewSessionsClient(conn)
113-
return sessionsClient, cleanup, nil
111+
return conn, cleanup, nil
114112
}
115113

116114
func getClientConn(address, tlsCertPath, macaroonPath string) (*grpc.ClientConn,

cmd/litcli/sessions.go

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -69,11 +69,12 @@ var addSessionCommand = cli.Command{
6969
}
7070

7171
func addSession(ctx *cli.Context) error {
72-
client, cleanup, err := getClient(ctx)
72+
clientConn, cleanup, err := connectClient(ctx)
7373
if err != nil {
7474
return err
7575
}
7676
defer cleanup()
77+
client := litrpc.NewSessionsClient(clientConn)
7778

7879
label := ctx.String("label")
7980
if label == "" {
@@ -192,11 +193,12 @@ var sessionStateMap = map[litrpc.SessionState]sessionFilter{
192193

193194
func listSessions(filter sessionFilter) func(ctx *cli.Context) error {
194195
return func(ctx *cli.Context) error {
195-
client, cleanup, err := getClient(ctx)
196+
clientConn, cleanup, err := connectClient(ctx)
196197
if err != nil {
197198
return err
198199
}
199200
defer cleanup()
201+
client := litrpc.NewSessionsClient(clientConn)
200202

201203
ctxb := context.Background()
202204
resp, err := client.ListSessions(
@@ -240,11 +242,12 @@ var revokeSessionCommand = cli.Command{
240242
}
241243

242244
func revokeSession(ctx *cli.Context) error {
243-
client, cleanup, err := getClient(ctx)
245+
clientConn, cleanup, err := connectClient(ctx)
244246
if err != nil {
245247
return err
246248
}
247249
defer cleanup()
250+
client := litrpc.NewSessionsClient(clientConn)
248251

249252
pubkey, err := hex.DecodeString(ctx.String("localpubkey"))
250253
if err != nil {

0 commit comments

Comments
 (0)