stream perp/spot markets configs.

sava-software · Oct 12, 2024 · b2363fb · b2363fb
1 parent f21e50c
commit b2363fb
Show file tree

Hide file tree

Showing 4 changed files with 38 additions and 0 deletions.
diff --git a/programs/src/main/java/software/sava/anchor/programs/drift/PerpMarkets.java b/programs/src/main/java/software/sava/anchor/programs/drift/PerpMarkets.java
@@ -1,7 +1,9 @@
 package software.sava.anchor.programs.drift;
 
 import java.util.Arrays;
+import java.util.Collection;
 import java.util.EnumMap;
+import java.util.stream.Stream;
 
 public interface PerpMarkets {
 
@@ -22,4 +24,8 @@ static PerpMarkets createRecord(final PerpMarketConfig[] marketConfigs) {
   PerpMarketConfig forProduct(final DriftProduct product);
 
   int numMarkets();
+
+  Stream<PerpMarketConfig> streamMarkets();
+
+  Collection<PerpMarketConfig> markets();
 }
diff --git a/programs/src/main/java/software/sava/anchor/programs/drift/PerpMarketsRecord.java b/programs/src/main/java/software/sava/anchor/programs/drift/PerpMarketsRecord.java
@@ -1,6 +1,9 @@
 package software.sava.anchor.programs.drift;
 
+import java.util.Arrays;
+import java.util.Collection;
 import java.util.Map;
+import java.util.stream.Stream;
 
 public record PerpMarketsRecord(PerpMarketConfig[] marketConfigs,
                                 Map<DriftProduct, PerpMarketConfig> byProduct) implements PerpMarkets {
@@ -19,4 +22,14 @@ public PerpMarketConfig forProduct(final DriftProduct product) {
   public int numMarkets() {
     return marketConfigs.length;
   }
+
+  @Override
+  public Stream<PerpMarketConfig> streamMarkets() {
+    return Arrays.stream(marketConfigs);
+  }
+
+  @Override
+  public Collection<PerpMarketConfig> markets() {
+    return byProduct.values();
+  }
 }
diff --git a/programs/src/main/java/software/sava/anchor/programs/drift/SpotMarkets.java b/programs/src/main/java/software/sava/anchor/programs/drift/SpotMarkets.java
@@ -1,7 +1,9 @@
 package software.sava.anchor.programs.drift;
 
 import java.util.Arrays;
+import java.util.Collection;
 import java.util.EnumMap;
+import java.util.stream.Stream;
 
 public interface SpotMarkets {
 
@@ -16,4 +18,8 @@ static SpotMarkets createRecord(final SpotMarketConfig[] marketConfigs) {
   SpotMarketConfig forAsset(final DriftAsset asset);
 
   int numMarkets();
+
+  Stream<SpotMarketConfig> streamMarkets();
+
+  Collection<SpotMarketConfig> markets();
 }
diff --git a/programs/src/main/java/software/sava/anchor/programs/drift/SpotMarketsRecord.java b/programs/src/main/java/software/sava/anchor/programs/drift/SpotMarketsRecord.java
@@ -1,6 +1,9 @@
 package software.sava.anchor.programs.drift;
 
+import java.util.Arrays;
+import java.util.Collection;
 import java.util.Map;
+import java.util.stream.Stream;
 
 record SpotMarketsRecord(SpotMarketConfig[] marketConfigs,
                          Map<DriftAsset, SpotMarketConfig> byAsset) implements SpotMarkets {
@@ -19,4 +22,14 @@ public SpotMarketConfig forAsset(final DriftAsset asset) {
   public int numMarkets() {
     return marketConfigs.length;
   }
+
+  @Override
+  public Stream<SpotMarketConfig> streamMarkets() {
+    return Arrays.stream(marketConfigs);
+  }
+
+  @Override
+  public Collection<SpotMarketConfig> markets() {
+    return byAsset.values();
+  }
 }