aljaz/electronwall
1
1
Fork 0
mirror of https://github.com/callebtc/electronwall.git synced 2025-12-18 15:44:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

clean

Browse Source
This commit is contained in:
callebtc
2022-07-08 12:38:18 +02:00
parent 5a00611984
commit f55b101151
2 changed files with 7 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
htlcInterceptor.go
View File

@@ -11,15 +11,16 @@ import (
) )
func dispatchHTLCAcceptor(ctx context.Context, conn *grpc.ClientConn, client lnrpc.LightningClient) { func dispatchHTLCAcceptor(ctx context.Context, conn *grpc.ClientConn, client lnrpc.LightningClient) {
// htlc event subscriber, reports on incoming htlc events
router := routerrpc.NewRouterClient(conn) router := routerrpc.NewRouterClient(conn)
// htlc event subscriber, reports on incoming htlc events
stream, err := router.SubscribeHtlcEvents(ctx, &routerrpc.SubscribeHtlcEventsRequest{}) stream, err := router.SubscribeHtlcEvents(ctx, &routerrpc.SubscribeHtlcEventsRequest{})
if err != nil { if err != nil {
return return
} }
go func() { go func() {
err := processHtlcEvents(stream) err := logHtlcEvents(stream)
if err != nil { if err != nil {
log.Error("htlc events error", log.Error("htlc events error",
"err", err) "err", err)
@@ -33,7 +34,7 @@ func dispatchHTLCAcceptor(ctx context.Context, conn *grpc.ClientConn, client lnr
} }
go func() { go func() {
err := processInterceptor(interceptor) err := interceptHtlcEvents(interceptor)
if err != nil { if err != nil {
log.Error("interceptor error", log.Error("interceptor error",
"err", err) "err", err)
@@ -43,13 +44,14 @@ func dispatchHTLCAcceptor(ctx context.Context, conn *grpc.ClientConn, client lnr
log.Info("Listening for incoming HTLCs") log.Info("Listening for incoming HTLCs")
} }
func processHtlcEvents(stream routerrpc.Router_SubscribeHtlcEventsClient) error { func logHtlcEvents(stream routerrpc.Router_SubscribeHtlcEventsClient) error {
for { for {
event, err := stream.Recv() event, err := stream.Recv()
if err != nil { if err != nil {
return err return err
} }
// we only care about HTLC forward events
if event.EventType != routerrpc.HtlcEvent_FORWARD { if event.EventType != routerrpc.HtlcEvent_FORWARD {
continue continue
} }
@@ -71,7 +73,7 @@ func processHtlcEvents(stream routerrpc.Router_SubscribeHtlcEventsClient) error
} }
} }
func processInterceptor(interceptor routerrpc.Router_HtlcInterceptorClient) error { func interceptHtlcEvents(interceptor routerrpc.Router_HtlcInterceptorClient) error {
for { for {
event, err := interceptor.Recv() event, err := interceptor.Recv()
if err != nil { if err != nil {

31
main.go
View File

@@ -76,37 +76,6 @@ func main() {
// htlc acceptor // htlc acceptor
go dispatchHTLCAcceptor(ctx, conn, client) go dispatchHTLCAcceptor(ctx, conn, client)
// // htlc event subscriber, reports on incoming htlc events
// router := routerrpc.NewRouterClient(conn)
// stream, err := router.SubscribeHtlcEvents(ctx, &routerrpc.SubscribeHtlcEventsRequest{})
// if err != nil {
// return
// }
// go func() {
// err := processHtlcEvents(stream)
// if err != nil {
// log.Error("htlc events error",
// "err", err)
// }
// }()
// // interceptor, decide whether to accept or reject
// interceptor, err := router.HtlcInterceptor(ctx)
// if err != nil {
// return
// }
// log.Info("HTLC Interceptor registered")
// go func() {
// err := processInterceptor(interceptor)
// if err != nil {
// log.Error("interceptor error",
// "err", err)
// }
// }()
wg.Wait() wg.Wait()
} }