2 * ============LICENSE_START====================================================
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
11 * http://www.apache.org/licenses/LICENSE-2.0
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====================================================
22 package org.onap.aaf.auth.batch;
24 import java.io.IOException;
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;
33 import com.datastax.driver.core.ResultSet;
34 import com.datastax.driver.core.exceptions.InvalidQueryException;
36 public abstract class CassBatch extends Batch {
38 protected CassBatch(AuthzTrans trans, String log4JName) throws APIException, IOException, OrganizationException {
40 // Flow all Env Logs to Log4j
41 Log4JLogTarget.setLog4JEnv(log4JName, env);
43 TimeTaken tt = trans.start("Connect to Cluster", Env.REMOTE);
45 session = cluster.connect();
52 protected void _close(AuthzTrans trans) {
54 trans.info().log("Closed Session");
57 public ResultSet executeQuery(String cql) {
58 return executeQuery(cql,"");
61 public ResultSet executeQuery(String cql, String extra) {
62 if (isDryRun() && !cql.startsWith("SELECT")) {
64 env.info().log("Would query" + extra + ": " + cql);
68 env.info().log("query" + extra + ": " + cql);
71 return session.execute(cql);
72 } catch (InvalidQueryException e) {
74 env.info().log("query: " + cql);