feat: add partitions to PRINT TOPIC output

confluentinc · Nov 19, 2020 · a77316a · a77316a
1 parent 976c852
commit a77316a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/...-app/src/main/java/io/confluent/ksql/rest/server/resources/streaming/RecordFormatter.java b/...-app/src/main/java/io/confluent/ksql/rest/server/resources/streaming/RecordFormatter.java
@@ -157,7 +157,8 @@ private List<String> formatRecords(final Iterable<ConsumerRecord<Bytes, Bytes>>
   private String formatRecord(final ConsumerRecord<Bytes, Bytes> record) {
     return "rowtime: " + formatRowTime(record.timestamp())
         + ", " + "key: " + keyDeserializers.format(record.key())
-        + ", value: " + valueDeserializers.format(record.value());
+        + ", value: " + valueDeserializers.format(record.value())
+        + ", partition: " + record.partition();
   }
 
   private static String formatRowTime(final long timestamp) {

diff --git a/.../src/test/java/io/confluent/ksql/rest/server/resources/streaming/RecordFormatterTest.java b/.../src/test/java/io/confluent/ksql/rest/server/resources/streaming/RecordFormatterTest.java
@@ -207,6 +207,15 @@ public void shouldFormatRowTime() {
       assertThat(formatted, containsString("rowtime: 2020/05/01 14:45:13.111 Z, "));
     }
 
+    @Test
+    public void shouldFormatPartition() {
+      // When:
+      final String formatted = formatSingle(consumerRecord(null, null));
+
+      // Then:
+      assertThat(formatted, containsString("partition: 1"));
+    }
+
     @Test
     public void shouldFormatNoRowTime() {
       // Given: