X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=auth%2Fauth-batch%2Fsrc%2Fmain%2Fjava%2Forg%2Fonap%2Faaf%2Fauth%2Fbatch%2Fhelpers%2FCQLBatch.java;h=d62c77811291a10ed5131f861caf3af13eb54d86;hb=49a8ce0053b7a4c9655e87020b7eed83ae8c312f;hp=c65026a8d430523b49b50070c896fd450e35f397;hpb=6a37b8869b1f79eaf1ccb9e6cf44accf40e99f6b;p=aaf%2Fauthz.git

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 c65026a8..d62c7781 100644
--- 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,63 +20,110 @@
 
 package org.onap.aaf.auth.batch.helpers;
 
+import org.onap.aaf.cadi.Access;
+import org.onap.aaf.misc.env.LogTarget;
+
 import com.datastax.driver.core.ResultSet;
 import com.datastax.driver.core.Session;
 
 public class CQLBatch {
-	private Session session;
-	private StringBuilder sb;
-	private int hasAdded;
+    private Session session;
+    private StringBuilder sb;
+    private int hasAdded;
+    private LogTarget log;
+    private long sleep;
+    private long last;
 
-	public CQLBatch(Session session) {
-		this.session = session;
-		sb = new StringBuilder();
-		hasAdded = 0;
-	}
-	public StringBuilder begin() {
-		sb.setLength(0);
-		sb.append("BEGIN BATCH\n");
-		hasAdded = sb.length();
-		return sb;
-	}
-	
-	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);
-			return true;
-		}
-	}
-	
-	public ResultSet execute() {
-		if(end()) {
-			return session.execute(sb.toString());
-		} else {
-			return null;
-		}
-	}
-	
-	public ResultSet execute(boolean dryRun) {
-		if(dryRun) {
-			end();
-			return null;
-		} else {
-			return execute();
-		}
-	}
-	
-	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 CQLBatch(LogTarget log, Session session) {
+        this.log = log;
+        this.session = session;
+        sb = new StringBuilder();
+        hasAdded = 0;
+        sleep = 0L;
+    }
+    public StringBuilder begin() {
+        sb.setLength(0);
+        sb.append("BEGIN BATCH\n");
+        hasAdded = sb.length();
+        return sb;
+    }
+    
+    private boolean end() {
+        if(sb.length()==hasAdded) {
+            return false;
+        } else {
+            sb.append("APPLY BATCH;\n");
+            log.log(sb);
+            return true;
+        }
+    }
+    
+    public ResultSet execute() {
+        if(end()) {
+            if(sleep>0) {
+                long left = last - System.currentTimeMillis();
+                if(left>0) {
+                    try {
+                        Thread.sleep(left);
+                    } catch (InterruptedException e) {
+                        Access.NULL.log(e); // Keep code check idiocy at bay
+                    }
+                }
+                last = System.currentTimeMillis()+sleep;
+            }
+            return session.execute(sb.toString());
+        } else {
+            return null;
+        }
+    }
+    
+    public ResultSet execute(boolean dryRun) {
+        ResultSet rv = null;
+        if(dryRun) {
+            if(sleep>0) {
+                long left = last - System.currentTimeMillis();
+                if(left>0) {
+                    try {
+                        Thread.sleep(left);
+                    } catch (InterruptedException e) {
+                        Access.NULL.log(e); // Keep code check idiocy at bay
+                    }
+                }
+                last = System.currentTimeMillis()+sleep;
+            }
+            end();
+        } else {
+            rv = execute();
+        }
+        sb.setLength(0);
+        return rv;
+    }
+    
+    public ResultSet singleExec(StringBuilder query, boolean dryRun) {
+        if(dryRun) {
+            return null;
+        } else {
+            return session.execute(query.toString());
+        }
+    }
+    
+    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 void sleep(int j) {
+        sleep = j*1000;
+    }
+    
+    public String toString() {
+        return sb.toString();
+    }
 }