Merge pull request #133 from Azure/bugfix/FixFlakyTest

* Fix Flaky test on TopicPartitionWriter
Azure · Dec 11, 2024 · 7279001 · 7279001
2 parents 11febbd + 6c3c80f
commit 7279001
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/main/java/com/microsoft/azure/kusto/kafka/connect/sink/TopicPartitionWriter.java b/src/main/java/com/microsoft/azure/kusto/kafka/connect/sink/TopicPartitionWriter.java
@@ -104,8 +104,13 @@ public void handleRollFile(SourceFile fileDescriptor) {
                         continue;
                     }
                 }
+                IngestionStatus ingestionStatus = null;
+                if(ingestionResult!=null && ingestionResult.getIngestionStatusCollection()!=null
+                        && !ingestionResult.getIngestionStatusCollection().isEmpty()){
+                    ingestionStatus = ingestionResult.getIngestionStatusCollection().get(0);
+                }
                 log.info("Kusto ingestion: file ({}) of size ({}) at current offset ({}) with status ({})",
-                        fileDescriptor.path, fileDescriptor.rawBytes, currentOffset, ingestionResult.getIngestionStatusCollection().get(0));
+                        fileDescriptor.path, fileDescriptor.rawBytes, currentOffset,ingestionStatus);
                 this.lastCommittedOffset = currentOffset;
                 return;
             } catch (IngestionServiceException exception) {