Update project structure to org.onap.aaf

[aaf/authz.git] / authz-batch / src / main / java / com / att / authz / CassBatch.java

diff --git a/authz-batch/src/main/java/com/att/authz/CassBatch.java b/authz-batch/src/main/java/com/att/authz/CassBatch.java
new file mode 100644 (file)
index 0000000..f251582
--- /dev/null
+++ b/authz-batch/src/main/java/com/att/authz/CassBatch.java
@@ -0,0 +1,58 @@
+/*******************************************************************************
+ * Copyright (c) 2016 AT&T Intellectual Property. All rights reserved.
+ *******************************************************************************/
+package com.att.authz;
+
+import java.io.IOException;
+
+import com.att.authz.env.AuthzTrans;
+import org.onap.aaf.inno.env.APIException;
+import org.onap.aaf.inno.env.Env;
+import org.onap.aaf.inno.env.TimeTaken;
+import org.onap.aaf.inno.env.impl.Log4JLogTarget;
+import com.datastax.driver.core.ResultSet;
+import com.datastax.driver.core.exceptions.InvalidQueryException;
+
+public abstract class CassBatch extends Batch {
+
+       protected CassBatch(AuthzTrans trans, String log4JName) throws APIException, IOException {
+               super(trans.env());
+               // Flow all Env Logs to Log4j
+               Log4JLogTarget.setLog4JEnv(log4JName, env);
+               
+               TimeTaken tt = trans.start("Connect to Cluster", Env.REMOTE);
+               try {
+                       session = cluster.connect();
+               } finally {
+                       tt.done();
+               }
+       }
+
+       @Override
+       protected void _close(AuthzTrans trans) {
+           session.close();
+               trans.info().log("Closed Session");
+       }
+
+       public ResultSet executeQuery(String cql) {
+               return executeQuery(cql,"");
+       }
+
+       public ResultSet executeQuery(String cql, String extra) {
+               if(isDryRun() && !cql.startsWith("SELECT")) {
+                       if(extra!=null)env.info().log("Would query" + extra + ": " + cql);
+               } else {
+                       if(extra!=null)env.info().log("query" + extra + ": " + cql);
+                       try {
+                               return session.execute(cql);
+                       } catch (InvalidQueryException e) {
+                               if(extra==null) {
+                                       env.info().log("query: " + cql);
+                               }
+                               throw e;
+                       }
+               } 
+               return null;
+       }
+
+}
\ No newline at end of file