src/main/java/org/embulk/output/elasticsearch/ElasticsearchRecordBuffer.java in embulk-output-elasticsearch-0.4.2 vs src/main/java/org/embulk/output/elasticsearch/ElasticsearchRecordBuffer.java in embulk-output-elasticsearch-0.4.3

@@ -86,14 +86,17 @@
     }
 
     @Override
     public TaskReport commitWithTaskReportUpdated(TaskReport taskReport)
     {
-        if (records.size() > 0) {
-            client.push(records, task, retryHelper);
-            log.info("Inserted {} records", records.size());
+        try {
+            if (records.size() > 0) {
+                client.push(records, task, retryHelper);
+                log.info("Inserted {} records", records.size());
+            }
+            return Exec.newTaskReport().set("inserted", totalCount);
         }
-
-        this.retryHelper.close();
-        return Exec.newTaskReport().set("inserted", totalCount);
+        finally {
+            this.retryHelper.close();
+        }
     }
 }