Improve Upload Batch

[aaf/authz.git] / auth / auth-batch / src / main / java / org / onap / aaf / auth / batch / update / Remove.java

diff --git a/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/update/Remove.java b/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/update/Remove.java
index 4472a31..b6b16fe 100644 (file)
--- a/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/update/Remove.java
+++ b/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/update/Remove.java
@@ -36,6 +36,7 @@ import org.onap.aaf.auth.batch.helpers.CQLBatch;
 import org.onap.aaf.auth.batch.helpers.CQLBatchLoop;
 import org.onap.aaf.auth.batch.helpers.Cred;
 import org.onap.aaf.auth.batch.helpers.Future;
+import org.onap.aaf.auth.batch.helpers.LastNotified;
 import org.onap.aaf.auth.batch.helpers.UserRole;
 import org.onap.aaf.auth.batch.helpers.X509;
 import org.onap.aaf.auth.dao.CassAccess;
@@ -75,7 +76,7 @@ public class Remove extends Batch {
                        } finally {
                                tt2.done();
                        }
-                       cqlBatch = new CQLBatch(noAvg.debug(),session); 
+                       cqlBatch = new CQLBatch(noAvg.info(),session); 
 
 
                } finally {
@@ -123,7 +124,6 @@ public class Remove extends Batch {
                                        CSV removeCSV = new CSV(env.access(),f);
                                        try {
                                                removeCSV.visit( row -> {
-                                                       cbl.preLoop();
                                                        switch(row.get(0)) {
                                                                case "info":
                                                                        switch(row.get(1)) {
@@ -175,6 +175,9 @@ public class Remove extends Batch {
                                                                        // Not cached
                                                                        Approval.deleteByIDBatch(cbl.inc(),row.get(1));
                                                                        break;
+                                                               case "notified":
+                                                                       LastNotified.delete(cbl.inc(),row);
+                                                                       break;
                                                        }
                                                });
                                                cbl.flush();