aljaz/lndhub.go
1
1
Fork 0
mirror of https://github.com/getAlby/lndhub.go.git synced 2026-01-07 15:05:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix republish script

Browse Source
This commit is contained in:
kiwiidb
2023-09-22 12:54:04 +02:00
parent 15f5f18f1b
commit f1a8442b84
2 changed files with 7 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
cmd/invoice-republishing/main.go
View File

@@ -11,15 +11,11 @@ import (
"github.com/getAlby/lndhub.go/lib"
"github.com/getAlby/lndhub.go/lib/service"
"github.com/getAlby/lndhub.go/rabbitmq"
"github.com/getsentry/sentry-go"
"github.com/joho/godotenv"
"github.com/kelseyhightower/envconfig"
"github.com/sirupsen/logrus"
)
// WARNING!
// Code has been refactored and not tested in it's currenc configuration yet
// Test before use
func main() {
c := &service.Config{}
@@ -60,6 +56,8 @@ func main() {
if err != nil {
logger.Fatal(err)
}
//small hack to get the script to work decently
defaultClient := rabbitmqClient.(*rabbitmq.DefaultClient)
// close the connection gently at the end of the runtime
defer rabbitmqClient.Close()
@@ -78,18 +76,6 @@ func main() {
InvoicePubSub: service.NewPubsub(),
}
ctx := context.Background()
go func() {
err = svc.RabbitMQClient.StartPublishInvoices(ctx,
svc.SubscribeIncomingOutgoingInvoices,
svc.EncodeInvoiceWithUserLogin,
)
if err != nil {
svc.Logger.Error(err)
sentry.CaptureException(err)
}
svc.Logger.Info("Rabbit invoice publisher done")
}()
dryRun := os.Getenv("DRY_RUN") == "true"
errCount := 0
for _, inv := range result {
@@ -97,10 +83,9 @@ func main() {
if dryRun {
continue
}
svc.InvoicePubSub.Publish(inv.Type, inv)
err = defaultClient.PublishToLndhubExchange(ctx, inv, svc.EncodeInvoiceWithUserLogin)
if err != nil {
errCount += 1
logger.Error(err)
logrus.WithError(err).Error("errror publishing to lndhub exchange")
}
}
logger.Infof("Published %d invoices, # errors %d", len(result), errCount)

6
rabbitmq/rabbitmq.go
View File

@@ -377,7 +377,7 @@ func (client *DefaultClient) StartPublishInvoices(ctx context.Context, invoicesS
case <-ctx.Done():
return context.Canceled
case incomingInvoice := <-in:
err = client.publishToLndhubExchange(ctx, incomingInvoice, payloadFunc)
err = client.PublishToLndhubExchange(ctx, incomingInvoice, payloadFunc)
if err != nil {
captureErr(client.logger, err, log.JSON{
@@ -387,7 +387,7 @@ func (client *DefaultClient) StartPublishInvoices(ctx context.Context, invoicesS
})
}
case outgoing := <-out:
err = client.publishToLndhubExchange(ctx, outgoing, payloadFunc)
err = client.PublishToLndhubExchange(ctx, outgoing, payloadFunc)
if err != nil {
captureErr(client.logger, err, log.JSON{
@@ -400,7 +400,7 @@ func (client *DefaultClient) StartPublishInvoices(ctx context.Context, invoicesS
}
}
func (client *DefaultClient) publishToLndhubExchange(ctx context.Context, invoice models.Invoice, payloadFunc EncodeOutgoingInvoiceFunc) error {
func (client *DefaultClient) PublishToLndhubExchange(ctx context.Context, invoice models.Invoice, payloadFunc EncodeOutgoingInvoiceFunc) error {
payload := bufPool.Get().(*bytes.Buffer)
err := payloadFunc(ctx, payload, invoice)
if err != nil {