Improve Upload Batch

[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 efd1ec9..4547fb1 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
@@ -63,11 +63,21 @@ public class CQLBatch {
        }
        
        public ResultSet execute(boolean dryRun) {
+               ResultSet rv = null;
                if(dryRun) {
                        end();
+               } else {
+                       rv = execute();
+               }
+               sb.setLength(0);
+               return rv;
+       }
+       
+       public ResultSet singleExec(StringBuilder query, boolean dryRun) {
+               if(dryRun) {
                        return null;
                } else {
-                       return execute();
+                       return session.execute(query.toString());
                }
        }
        
@@ -82,4 +92,8 @@ public class CQLBatch {
                }
                execute(dryRun);
        }
+       
+       public String toString() {
+               return sb.toString();
+       }
 }