use multi epoch (#192)

rpcpool · Nov 21, 2024 · 272e6b1 · 272e6b1
1 parent ef02393
commit 272e6b1
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/multiepoch-getSignaturesForAddress.go b/multiepoch-getSignaturesForAddress.go
@@ -43,7 +43,7 @@ func (ser *MultiEpoch) getGsfaReadersInEpochDescendingOrder() ([]*gsfa.GsfaReade
 }
 
 // getGsfaReadersInEpochDescendingOrder returns a list of gsfa readers in epoch order (from most recent to oldest).
-func (ser *MultiEpoch) getGsfaReadersInEpochDescendingOrderForSlotRange(ctx context.Context, startSlot, endSlot uint64) ([]*gsfa.GsfaReader, []uint64) {
+func (ser *MultiEpoch) getGsfaReadersInEpochDescendingOrderForSlotRange(ctx context.Context, startSlot, endSlot uint64) (*gsfa.GsfaReaderMultiepoch, []uint64) {
 	ser.mu.RLock()
 	defer ser.mu.RUnlock()
 
@@ -72,7 +72,15 @@ func (ser *MultiEpoch) getGsfaReadersInEpochDescendingOrderForSlotRange(ctx cont
 			epochNums = append(epochNums, epoch.Epoch())
 		}
 	}
-	return gsfaReaders, epochNums
+
+	gsfaReaderMultiEpoch, err := gsfa.NewGsfaReaderMultiepoch(gsfaReaders)
+	if err != nil {
+		klog.Errorf("failed to construct gsfaReaderMultiEpoch: %w", err)
+		return nil, nil
+	}
+
+	return gsfaReaderMultiEpoch, epochNums
+
 }
 
 func countTransactions(v gsfa.EpochToTransactionObjects) int {