Skip to content

Commit 4467453

Browse files
committed
Support arbitrary key-value argument in stream creator
1 parent 6dd42a0 commit 4467453

File tree

3 files changed

+20
-1
lines changed

3 files changed

+20
-1
lines changed

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

+9
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,15 @@ public interface StreamCreator {
102102
*/
103103
StreamCreator initialMemberCount(int initialMemberCount);
104104

105+
/**
106+
* Set an argument for the stream creation.
107+
*
108+
* @param key argument key
109+
* @param value argument value
110+
* @return this creator instance
111+
*/
112+
StreamCreator argument(String key, String value);
113+
105114
/**
106115
* Configure the super stream to create.
107116
*

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

+5-1
Original file line numberDiff line numberDiff line change
@@ -2753,7 +2753,11 @@ public StreamParametersBuilder initialMemberCount(int initialMemberCount) {
27532753
}
27542754

27552755
public StreamParametersBuilder put(String key, String value) {
2756-
parameters.put(key, value);
2756+
if (value == null) {
2757+
parameters.remove(key);
2758+
} else {
2759+
parameters.put(key, value);
2760+
}
27572761
return this;
27582762
}
27592763

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

+6
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,12 @@ public StreamCreator initialMemberCount(int initialMemberCount) {
9292
return this;
9393
}
9494

95+
@Override
96+
public StreamCreator argument(String key, String value) {
97+
streamParametersBuilder.put(key, value);
98+
return this;
99+
}
100+
95101
@Override
96102
public SuperStreamConfiguration superStream() {
97103
if (this.superStreamConfiguration == null) {

0 commit comments

Comments
 (0)