Addresses #202

Author: Scott Stafford

infrastructure/src/main/java/com/marklogic/spring/batch/item/writer/MarkLogicItemWriter.java

@@ -29,6 +29,8 @@ public class MarkLogicItemWriter extends LoggingObject implements ItemWriter<Doc
 
     private UriTransformer uriTransformer;
     private BatchWriter batchWriter;
+    protected long writeCount = 0;
+    protected long writeCalled = 0;
 
     public MarkLogicItemWriter(DatabaseClient client) {
         this(Arrays.asList(client));
@@ -46,6 +48,8 @@ public MarkLogicItemWriter(BatchWriter batchWriter) {
 
     @Override
     public void write(List<? extends DocumentWriteOperation> items) throws Exception {
+        writeCalled += 1;
+        writeCount += items.size();
         if (uriTransformer != null) {
             List<DocumentWriteOperation> newItems = new ArrayList<>();
             for (DocumentWriteOperation op : items) {
@@ -57,6 +61,9 @@ public void write(List<? extends DocumentWriteOperation> items) throws Exception
         } else {
             batchWriter.write(items);
         }
+        if (writeCalled % 250 == 0) {
+            logger.info("Write Count: " + writeCount);
+        }
     }
 
     @Override
@@ -83,6 +90,7 @@ public void close() throws ItemStreamException {
         batchWriter.waitForCompletion();
         if (logger.isInfoEnabled()) {
             logger.info("On stream close, finished waiting for BatchWriter to complete");
+            logger.info("Final Write Count: " + writeCount);
         }
     }
 

infrastructure/src/main/java/com/marklogic/spring/batch/item/writer/support/TempRestBatchWriter.java

@@ -11,9 +11,6 @@
 import java.util.Arrays;
 import java.util.List;
 
-/**
- * Created by sstafford on 1/27/2017.
- */
 public class TempRestBatchWriter extends RestBatchWriter {
 
     private Format returnFormat;