Merge pull request #637 from rabbitmq/key-value-argument-in-stream-creation

Support arbitrary key-value argument in stream creator

Author: acogoluegnes

src/main/java/com/rabbitmq/stream/StreamCreator.java

@@ -102,6 +102,15 @@ public interface StreamCreator {
    */
   StreamCreator initialMemberCount(int initialMemberCount);
 
+  /**
+   * Set an argument for the stream creation.
+   *
+   * @param key argument key
+   * @param value argument value
+   * @return this creator instance
+   */
+  StreamCreator argument(String key, String value);
+
   /**
    * Configure the super stream to create.
    *

src/main/java/com/rabbitmq/stream/impl/Client.java

@@ -2753,7 +2753,11 @@ public StreamParametersBuilder initialMemberCount(int initialMemberCount) {
     }
 
     public StreamParametersBuilder put(String key, String value) {
-      parameters.put(key, value);
+      if (value == null) {
+        parameters.remove(key);
+      } else {
+        parameters.put(key, value);
+      }
       return this;
     }
 

src/main/java/com/rabbitmq/stream/impl/StreamStreamCreator.java

@@ -92,6 +92,12 @@ public StreamCreator initialMemberCount(int initialMemberCount) {
     return this;
   }
 
+  @Override
+  public StreamCreator argument(String key, String value) {
+    streamParametersBuilder.put(key, value);
+    return this;
+  }
+
   @Override
   public SuperStreamConfiguration superStream() {
     if (this.superStreamConfiguration == null) {