mirror of
https://github.com/aljazceru/lspd.git
synced 2025-12-20 23:34:22 +01:00
cli: add revenue command
This commit is contained in:
Jesse de Wit
@@ -5,6 +5,8 @@ import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/breez/lspd/lightning"
|
||||
"github.com/btcsuite/btcd/wire"
|
||||
"github.com/jackc/pgx/v5/pgxpool"
|
||||
)
|
||||
|
||||
@@ -164,3 +166,204 @@ func (s *RevenueCliStore) sanityCheck(ctx context.Context, startNs, endNs uint64
|
||||
return nil
|
||||
}
|
||||
|
||||
type RevenueForward struct {
|
||||
Identifier string
|
||||
Nodeid []byte
|
||||
PeeridIn []byte
|
||||
PeeridOut []byte
|
||||
AmtMsatIn uint64
|
||||
AmtMsatOut uint64
|
||||
ResolvedTime uint64
|
||||
ChannelPointIn wire.OutPoint
|
||||
ChannelPointOut wire.OutPoint
|
||||
SendToken *string
|
||||
ReceiveToken *string
|
||||
OpenChannelHtlc *OpenChannelHtlc
|
||||
}
|
||||
|
||||
type OpenChannelHtlc struct {
|
||||
Nodeid []byte
|
||||
Peerid []byte
|
||||
ForwardAmountMsat uint64
|
||||
OriginalAmountMsat uint64
|
||||
IncomingAmountMsat uint64
|
||||
ChannelPoint *wire.OutPoint
|
||||
}
|
||||
|
||||
func (s *RevenueCliStore) GetOpenChannelHtlcs(
|
||||
ctx context.Context,
|
||||
startNs uint64,
|
||||
endNs uint64,
|
||||
) ([]*OpenChannelHtlc, error) {
|
||||
// filter htlcs used for channel opens. Only include the ones that may have been actually settled.
|
||||
openChannelHtlcs, err := s.pool.Query(ctx, `
|
||||
SELECT htlc.nodeid
|
||||
, htlc.peerid
|
||||
, htlc.forward_amt_msat
|
||||
, htlc.original_amt_msat
|
||||
, htlc.incoming_amt_msat
|
||||
, htlc.funding_tx_id
|
||||
, htlc.funding_tx_outnum
|
||||
FROM open_channel_htlcs htlc
|
||||
INNER JOIN (
|
||||
SELECT DISTINCT h.nodeid, h.peerid, c.funding_tx_id, c.funding_tx_outnum
|
||||
FROM forwarding_history h
|
||||
INNER JOIN channels c
|
||||
ON h.nodeid = c.nodeid AND (h.chanid_out = c.confirmed_scid OR h.chanid_out = c.alias_scid)
|
||||
WHERE h.resolved_time >= $1 AND h.resolved_time < $2
|
||||
) a
|
||||
ON htlc.nodeid = a.nodeid
|
||||
AND htlc.peerid = a.peerid
|
||||
AND htlc.funding_tx_id = a.funding_tx_id
|
||||
AND htlc.funding_tx_outnum = a.funding_tx_outnum
|
||||
ORDER BY htlc.nodeid, htlc.peerid, htlc.funding_tx_id, htlc.funding_tx_outnum
|
||||
`, startNs, endNs)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to query open channel htlcs: %w", err)
|
||||
}
|
||||
defer openChannelHtlcs.Close()
|
||||
|
||||
var result []*OpenChannelHtlc
|
||||
for openChannelHtlcs.Next() {
|
||||
var nodeid []byte
|
||||
var peerid []byte
|
||||
var forward_amt_msat int64
|
||||
var original_amt_msat int64
|
||||
var incoming_amt_msat int64
|
||||
var funding_tx_id []byte
|
||||
var funding_tx_outnum uint32
|
||||
err = openChannelHtlcs.Scan(
|
||||
&nodeid,
|
||||
&peerid,
|
||||
&forward_amt_msat,
|
||||
&original_amt_msat,
|
||||
&incoming_amt_msat,
|
||||
&funding_tx_id,
|
||||
&funding_tx_outnum,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to scan open channel htlc: %w", err)
|
||||
}
|
||||
|
||||
cp, err := lightning.NewOutPoint(funding_tx_id, funding_tx_outnum)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("invalid funding outpoint: %w", err)
|
||||
}
|
||||
|
||||
result = append(result, &OpenChannelHtlc{
|
||||
Nodeid: nodeid,
|
||||
Peerid: peerid,
|
||||
ForwardAmountMsat: uint64(forward_amt_msat),
|
||||
OriginalAmountMsat: uint64(original_amt_msat),
|
||||
IncomingAmountMsat: uint64(incoming_amt_msat),
|
||||
ChannelPoint: cp,
|
||||
})
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
|
||||
// Gets all settled forwards in the defined time range. Ordered by nodeid, peerid_in, amt_msat_in, resolved_time
|
||||
func (s *RevenueCliStore) GetForwards(
|
||||
ctx context.Context,
|
||||
startNs uint64,
|
||||
endNs uint64,
|
||||
) ([]*RevenueForward, error) {
|
||||
err := s.sanityCheck(ctx, startNs, endNs)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
ctxc, cancel := context.WithCancel(ctx)
|
||||
defer cancel()
|
||||
|
||||
// Select all forwards, and include information about the channel and token used
|
||||
ourForwards, err := s.pool.Query(ctxc, tokenChannelsCte+`
|
||||
SELECT h.identifier
|
||||
, h.nodeid
|
||||
, h.peerid_in
|
||||
, h.peerid_out
|
||||
, h.amt_msat_in
|
||||
, h.amt_msat_out
|
||||
, h.resolved_time
|
||||
, c_in.funding_tx_id AS funding_tx_id_in
|
||||
, c_in.funding_tx_outnum AS funding_tx_outnum_in
|
||||
, c_out.funding_tx_id AS funding_tx_id_out
|
||||
, c_out.funding_tx_outnum AS funding_tx_outnum_out
|
||||
, tc_in.token AS send_token
|
||||
, tc_out.token AS receive_token
|
||||
FROM forwarding_history h
|
||||
INNER JOIN channels c_in
|
||||
ON h.nodeid = c_in.nodeid AND (h.chanid_in = c_in.confirmed_scid OR h.chanid_in = c_in.alias_scid)
|
||||
INNER JOIN channels c_out
|
||||
ON h.nodeid = c_out.nodeid AND (h.chanid_out = c_out.confirmed_scid OR h.chanid_out = c_out.alias_scid)
|
||||
LEFT JOIN token_channels tc_in
|
||||
ON c_in.nodeid = tc_in.nodeid AND c_in.funding_tx_id = tc_in.funding_tx_id AND c_in.funding_tx_outnum = tc_in.funding_tx_outnum
|
||||
LEFT JOIN token_channels tc_out
|
||||
ON c_out.nodeid = tc_out.nodeid AND c_out.funding_tx_id = tc_out.funding_tx_id AND c_out.funding_tx_outnum = tc_out.funding_tx_outnum
|
||||
WHERE h.resolved_time >= $1 AND h.resolved_time < $2
|
||||
ORDER BY h.nodeid, h.peerid_in, h.amt_msat_in, h.resolved_time
|
||||
`, startNs, endNs)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to query our forwards: %w", err)
|
||||
}
|
||||
|
||||
var forwards []*RevenueForward
|
||||
for ourForwards.Next() {
|
||||
var identifier string
|
||||
var nodeid []byte
|
||||
var peerid_in []byte
|
||||
var peerid_out []byte
|
||||
var amt_msat_in int64
|
||||
var amt_msat_out int64
|
||||
var resolved_time int64
|
||||
var funding_tx_id_in []byte
|
||||
var funding_tx_outnum_in uint32
|
||||
var funding_tx_id_out []byte
|
||||
var funding_tx_outnum_out uint32
|
||||
var send_token *string
|
||||
var receive_token *string
|
||||
err = ourForwards.Scan(
|
||||
&identifier,
|
||||
&nodeid,
|
||||
&peerid_in,
|
||||
&peerid_out,
|
||||
&amt_msat_in,
|
||||
&amt_msat_out,
|
||||
&resolved_time,
|
||||
&funding_tx_id_in,
|
||||
&funding_tx_outnum_in,
|
||||
&funding_tx_id_out,
|
||||
&funding_tx_outnum_out,
|
||||
&send_token,
|
||||
&receive_token,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to scan our forward: %w", err)
|
||||
}
|
||||
|
||||
cpIn, err := lightning.NewOutPoint(funding_tx_id_in, funding_tx_outnum_in)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("invalid funding outpoint: %w", err)
|
||||
}
|
||||
cpOut, err := lightning.NewOutPoint(funding_tx_id_out, funding_tx_outnum_out)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("invalid funding outpoint: %w", err)
|
||||
}
|
||||
forwards = append(forwards, &RevenueForward{
|
||||
Identifier: identifier,
|
||||
Nodeid: nodeid,
|
||||
PeeridIn: peerid_in,
|
||||
PeeridOut: peerid_out,
|
||||
AmtMsatIn: uint64(amt_msat_in),
|
||||
AmtMsatOut: uint64(amt_msat_out),
|
||||
ResolvedTime: uint64(resolved_time),
|
||||
ChannelPointIn: *cpIn,
|
||||
ChannelPointOut: *cpOut,
|
||||
SendToken: send_token,
|
||||
ReceiveToken: receive_token,
|
||||
})
|
||||
}
|
||||
|
||||
return forwards, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user