[NATS] Extend to support in-memory stream (#370)

* [nats] Add storage type to config so one can test with file or memory stream. * [nats] Add maxBytes to config so one can prevent NATS running out of memory incase the storage type is set to "Memory"
openmessaging · Aug 28, 2023 · b3a9044 · b3a9044
1 parent 1c681e9
commit b3a9044
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/driver-nats/src/main/java/io/openmessaging/benchmark/driver/nats/NatsBenchmarkDriver.java b/driver-nats/src/main/java/io/openmessaging/benchmark/driver/nats/NatsBenchmarkDriver.java
@@ -28,7 +28,6 @@
 import io.nats.client.Options;
 import io.nats.client.PushSubscribeOptions;
 import io.nats.client.api.ConsumerConfiguration;
-import io.nats.client.api.StorageType;
 import io.nats.client.api.StreamConfiguration;
 import io.nats.client.api.StreamInfo;
 import io.nats.client.support.JsonUtils;
@@ -94,7 +93,8 @@ public CompletableFuture<Void> createTopic(String topic, int partitions) {
                             StreamConfiguration.builder()
                                     .name(topic)
                                     .subjects(topic)
-                                    .storageType(StorageType.File)
+                                    .storageType(config.storageType)
+                                    .maxBytes(config.maxBytes)
                                     .replicas(config.replicationFactor)
                                     .build());
             log.info("Created stream {} -- {}", topic, JsonUtils.getFormatted(streamInfo));

diff --git a/driver-nats/src/main/java/io/openmessaging/benchmark/driver/nats/NatsConfig.java b/driver-nats/src/main/java/io/openmessaging/benchmark/driver/nats/NatsConfig.java
@@ -13,8 +13,16 @@
  */
 package io.openmessaging.benchmark.driver.nats;
 
+
+import io.nats.client.api.StorageType;
+
 public class NatsConfig {
     public String natsHostUrl;
 
     public int replicationFactor;
+
+    public StorageType storageType = StorageType.File;
+
+    // -1 is unlimited
+    public int maxBytes = -1;
 }