From 92c50b66b4b6ff303fdb175890a78a89d33888bd Mon Sep 17 00:00:00 2001
From: Christian Lohr <christian@volume.finance>
Date: Mon, 2 Oct 2023 21:52:08 +0200
Subject: [PATCH] fix: filter gravity batches queried from paloma

---
 chain/paloma/client.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/chain/paloma/client.go b/chain/paloma/client.go
index 46396181..62587c33 100644
--- a/chain/paloma/client.go
+++ b/chain/paloma/client.go
@@ -480,7 +480,14 @@ func gravityQueryLastUnsignedBatch(ctx context.Context, grpcClient grpc.ClientCo
 		return nil, err
 	}
 
-	return batches.Batch, nil
+	filtered := make([]gravity.OutgoingTxBatch, 0, len(batches.Batch))
+	for _, v := range batches.Batch {
+		if v.GetChainReferenceID() == chainReferenceID {
+			filtered = append(filtered, v)
+		}
+	}
+
+	return filtered, nil
 }
 
 func (c Client) GravityConfirmBatches(ctx context.Context, signatures ...chain.SignedGravityOutgoingTxBatch) error {