[nats] Add maxBytes to config

so one can prevent NATS running out of memory incase the storage type is set to "Memory"
openmessaging · Feb 22, 2023 · bd56d9c · bd56d9c
1 parent 1ff73a0
commit bd56d9c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 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
@@ -94,6 +94,7 @@ public CompletableFuture<Void> createTopic(String topic, int partitions) {
                                     .name(topic)
                                     .subjects(topic)
                                     .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
@@ -22,4 +22,7 @@ public class NatsConfig {
     public int replicationFactor;
 
     public StorageType storageType = StorageType.File;
+
+    // -1 is unlimited
+    public int maxBytes = -1;
 }