Skip to content

Commit a9bc1e9

Browse files
author
jruaux
committed
fix: take into account the maxTasks config parameter.
Fixes #27
1 parent 5a5042a commit a9bc1e9

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

core/redis-kafka-connect/src/main/java/com/redis/kafka/connect/RedisSinkConnector.java

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,9 @@
1515
*/
1616
package com.redis.kafka.connect;
1717

18+
import java.util.Collections;
1819
import java.util.List;
1920
import java.util.Map;
20-
import java.util.stream.Collectors;
21-
import java.util.stream.IntStream;
2221

2322
import org.apache.kafka.common.config.ConfigDef;
2423
import org.apache.kafka.connect.connector.Task;
@@ -44,7 +43,7 @@ public Class<? extends Task> taskClass() {
4443

4544
@Override
4645
public List<Map<String, String>> taskConfigs(int maxTasks) {
47-
return IntStream.range(0, maxTasks).mapToObj(i -> props).collect(Collectors.toList());
46+
return Collections.nCopies(maxTasks, props);
4847
}
4948

5049
@Override

core/redis-kafka-connect/src/test/java/com/redis/kafka/connect/SinkConnectorTest.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,9 @@ void testTaskConfigs() {
2828
HashMap<String, String> props = new HashMap<>();
2929
props.put("field1", "value1");
3030
connector.start(props);
31-
Assertions.assertEquals(props, connector.taskConfigs(123).get(0));
31+
int maxTasks = 123;
32+
Assertions.assertEquals(props, connector.taskConfigs(maxTasks).get(0));
33+
Assertions.assertEquals(maxTasks, connector.taskConfigs(maxTasks).size());
3234
}
3335

3436
@Test

0 commit comments

Comments
 (0)