Skip to content

Commit 644e243

Browse files
adityausathefmbenhassine
authored andcommitted
Make kafkaTemplate protected in KafkaItemWriter
Issue #3802
1 parent 72b019b commit 644e243

File tree

2 files changed

+14
-1
lines changed

2 files changed

+14
-1
lines changed

spring-batch-infrastructure/src/main/java/org/springframework/batch/item/kafka/KafkaItemWriter.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
*/
3434
public class KafkaItemWriter<K, T> extends KeyValueItemWriter<K, T> {
3535

36-
private KafkaTemplate<K, T> kafkaTemplate;
36+
protected KafkaTemplate<K, T> kafkaTemplate;
3737

3838
@Override
3939
protected void writeKeyValue(K key, T value) {

spring-batch-infrastructure/src/test/java/org/springframework/batch/item/kafka/KafkaItemWriterTests.java

+13
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,19 @@ public void testBasicDelete() throws Exception {
103103
verify(this.kafkaTemplate).sendDefault(items.get(1), null);
104104
}
105105

106+
@Test
107+
public void testKafkaTemplateCanBeReferencedFromSubclass() {
108+
KafkaItemWriter<String, String> kafkaItemWriter = new KafkaItemWriter<String, String>() {
109+
@Override
110+
protected void writeKeyValue(String key, String value) {
111+
this.kafkaTemplate.sendDefault(key, value);
112+
}
113+
};
114+
kafkaItemWriter.setKafkaTemplate(kafkaTemplate);
115+
kafkaItemWriter.writeKeyValue("k", "v");
116+
verify(kafkaTemplate).sendDefault("k", "v");
117+
}
118+
106119
static class KafkaItemKeyMapper implements Converter<String, String> {
107120

108121
@Override

0 commit comments

Comments
 (0)