From d36fa6a210261c147453144ef2fb202745319c84 Mon Sep 17 00:00:00 2001
From: dnsbtchr <dnsbtchr@googlemail.com>
Date: Tue, 26 Feb 2019 11:47:17 +0100
Subject: [PATCH] NIFI-6016 PutCassandraRecord batch size

Adds resetting the batch size to fix broken batch processing

Removes empty line

Signed-off-by: Matthew Burgess <mattyb149@apache.org>

This closes #3337
---
 .../org/apache/nifi/processors/cassandra/PutCassandraRecord.java | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nifi-nar-bundles/nifi-cassandra-bundle/nifi-cassandra-processors/src/main/java/org/apache/nifi/processors/cassandra/PutCassandraRecord.java b/nifi-nar-bundles/nifi-cassandra-bundle/nifi-cassandra-processors/src/main/java/org/apache/nifi/processors/cassandra/PutCassandraRecord.java
index 84016dcf89..8acc36aecd 100644
--- a/nifi-nar-bundles/nifi-cassandra-bundle/nifi-cassandra-processors/src/main/java/org/apache/nifi/processors/cassandra/PutCassandraRecord.java
+++ b/nifi-nar-bundles/nifi-cassandra-bundle/nifi-cassandra-processors/src/main/java/org/apache/nifi/processors/cassandra/PutCassandraRecord.java
@@ -167,6 +167,7 @@ public class PutCassandraRecord extends AbstractCassandraProcessor {
                 if (recordsAdded.incrementAndGet() == batchSize) {
                     connectionSession.execute(batchStatement);
                     batchStatement.clear();
+                    recordsAdded.set(0);
                 }
             }