Batch Test improvements

[aaf/authz.git] / auth / auth-batch / src / main / java / org / onap / aaf / auth / batch / helpers / CQLBatch.java

diff --git a/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/helpers/CQLBatch.java b/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/helpers/CQLBatch.java
index 5df5dcd..9f685ad 100644 (file)
--- a/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/helpers/CQLBatch.java
+++ b/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/helpers/CQLBatch.java
@@ -20,6 +20,8 @@
 
 package org.onap.aaf.auth.batch.helpers;
 
+import org.onap.aaf.misc.env.LogTarget;
+
 import com.datastax.driver.core.ResultSet;
 import com.datastax.driver.core.Session;
 
@@ -27,8 +29,10 @@ public class CQLBatch {
        private Session session;
        private StringBuilder sb;
        private int hasAdded;
+       private LogTarget log;
 
-       public CQLBatch(Session session) {
+       public CQLBatch(LogTarget log, Session session) {
+               this.log = log;
                this.session = session;
                sb = new StringBuilder();
                hasAdded = 0;
@@ -42,11 +46,10 @@ public class CQLBatch {
        
        private boolean end() {
                if(sb.length()==hasAdded) {
-                       System.out.println("Nothing to Process");
                        return false;
                } else {
                        sb.append("APPLY BATCH;\n");
-                       System.out.println(sb);
+                       log.log(sb);
                        return true;
                }
        }
@@ -60,12 +63,29 @@ public class CQLBatch {
        }
        
        public ResultSet execute(boolean dryRun) {
+               ResultSet rv = null;
                if(dryRun) {
                        end();
-                       return null;
                } else {
-                       return execute();
+                       rv = execute();
+               }
+               sb.setLength(0);
+               return rv;
+       }
+       
+       public void touch(String table, int begin, int end, boolean dryRun) {
+               StringBuilder sb = begin();
+               for(int i=begin;i<end;++i) {
+                       sb.append("UPDATE cache SET touched=dateof(now()) WHERE name='");
+                       sb.append(table);
+                       sb.append("' AND seg=");
+                       sb.append(i);
+                       sb.append(";\n");
                }
-               
+               execute(dryRun);
+       }
+       
+       public String toString() {
+               return sb.toString();
        }
 }