Skip to content

Commit 9610758

Browse files
committed
Added validator to generate.timeout
1 parent 9c3b0f1 commit 9610758

File tree

1 file changed

+23
-2
lines changed

1 file changed

+23
-2
lines changed

src/main/java/io/confluent/kafka/connect/datagen/DatagenConnectorConfig.java

Lines changed: 23 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -93,8 +93,17 @@ public static ConfigDef conf() {
9393
Importance.HIGH,
9494
QUICKSTART_DOC
9595
)
96-
.define(RANDOM_SEED_CONF, Type.LONG, null, Importance.LOW, RANDOM_SEED_DOC)
97-
.define(GENERATE_TIMEOUT_CONF, Type.LONG, null, Importance.LOW, GENERATE_TIMEOUT_DOC);
96+
.define(RANDOM_SEED_CONF,
97+
Type.LONG,
98+
null,
99+
Importance.LOW,
100+
RANDOM_SEED_DOC)
101+
.define(GENERATE_TIMEOUT_CONF,
102+
Type.LONG,
103+
null,
104+
new GenerateTimeoutValidator(),
105+
Importance.LOW,
106+
GENERATE_TIMEOUT_DOC);
98107
}
99108

100109
public String getKafkaTopic() {
@@ -202,5 +211,17 @@ public void ensureValid(String name, Object value) {
202211
ConfigUtils.getSchemaFromSchemaFileName((String) value);
203212
}
204213
}
214+
215+
private static class GenerateTimeoutValidator implements Validator {
216+
217+
@Override
218+
public void ensureValid(String name, Object value) {
219+
Long longValue = (Long) value;
220+
if (longValue > 0 && longValue <= 60000L) {
221+
return;
222+
}
223+
throw new ConfigException(name + " must be in the range [1, 60,000] ms");
224+
}
225+
}
205226
}
206227

0 commit comments

Comments
 (0)