auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/CassBatch.java

   1 /**
   2  * ============LICENSE_START====================================================
   3  * org.onap.aaf
   4  * ===========================================================================
   5  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ===========================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END====================================================
  19  *
  20  */
  21
  22 package org.onap.aaf.auth.batch;
  23
  24 import java.io.IOException;
  25
  26 import org.onap.aaf.auth.env.AuthzTrans;
  27 import org.onap.aaf.auth.org.OrganizationException;
  28 import org.onap.aaf.misc.env.APIException;
  29 import org.onap.aaf.misc.env.Env;
  30 import org.onap.aaf.misc.env.TimeTaken;
  31 import org.onap.aaf.misc.env.impl.Log4JLogTarget;
  32
  33 import com.datastax.driver.core.ResultSet;
  34 import com.datastax.driver.core.exceptions.InvalidQueryException;
  35
  36 public abstract class CassBatch extends Batch {
  37
  38     protected CassBatch(AuthzTrans trans, String log4JName) throws APIException, IOException, OrganizationException {
  39         super(trans.env());
  40         // Flow all Env Logs to Log4j
  41         Log4JLogTarget.setLog4JEnv(log4JName, env);
  42
  43         TimeTaken tt = trans.start("Connect to Cluster", Env.REMOTE);
  44         try {
  45             session = cluster.connect();
  46         } finally {
  47             tt.done();
  48         }
  49     }
  50
  51     @Override
  52     protected void _close(AuthzTrans trans) {
  53         session.close();
  54         trans.info().log("Closed Session");
  55     }
  56
  57     public ResultSet executeQuery(String cql) {
  58         return executeQuery(cql,"");
  59     }
  60
  61     public ResultSet executeQuery(String cql, String extra) {
  62         if (isDryRun() && !cql.startsWith("SELECT")) {
  63             if (extra!=null) {
  64                 env.info().log("Would query" + extra + ": " + cql);
  65             }
  66         } else {
  67             if (extra!=null) {
  68                 env.info().log("query" + extra + ": " + cql);
  69             }
  70             try {
  71                 return session.execute(cql);
  72             } catch (InvalidQueryException e) {
  73                 if (extra==null) {
  74                     env.info().log("query: " + cql);
  75                 }
  76                 throw e;
  77             }
  78         }
  79         return null;
  80     }
  81
  82 }