auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/helpers/CQLBatch.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 package org.onap.aaf.auth.batch.helpers;
  22
  23 import org.onap.aaf.misc.env.LogTarget;
  24
  25 import com.datastax.driver.core.ResultSet;
  26 import com.datastax.driver.core.Session;
  27
  28 public class CQLBatch {
  29     private Session session;
  30     private StringBuilder sb;
  31     private int hasAdded;
  32     private LogTarget log;
  33
  34     public CQLBatch(LogTarget log, Session session) {
  35         this.log = log;
  36         this.session = session;
  37         sb = new StringBuilder();
  38         hasAdded = 0;
  39     }
  40     public StringBuilder begin() {
  41         sb.setLength(0);
  42         sb.append("BEGIN BATCH\n");
  43         hasAdded = sb.length();
  44         return sb;
  45     }
  46
  47     private boolean end() {
  48         if(sb.length()==hasAdded) {
  49             return false;
  50         } else {
  51             sb.append("APPLY BATCH;\n");
  52             log.log(sb);
  53             return true;
  54         }
  55     }
  56
  57     public ResultSet execute() {
  58         if(end()) {
  59             return session.execute(sb.toString());
  60         } else {
  61             return null;
  62         }
  63     }
  64
  65     public ResultSet execute(boolean dryRun) {
  66         ResultSet rv = null;
  67         if(dryRun) {
  68             end();
  69         } else {
  70             rv = execute();
  71         }
  72         sb.setLength(0);
  73         return rv;
  74     }
  75
  76     public ResultSet singleExec(StringBuilder query, boolean dryRun) {
  77         if(dryRun) {
  78             return null;
  79         } else {
  80             return session.execute(query.toString());
  81         }
  82     }
  83
  84     public void touch(String table, int begin, int end, boolean dryRun) {
  85         StringBuilder sb = begin();
  86         for(int i=begin;i<end;++i) {
  87             sb.append("UPDATE cache SET touched=dateof(now()) WHERE name='");
  88             sb.append(table);
  89             sb.append("' AND seg=");
  90             sb.append(i);
  91             sb.append(";\n");
  92         }
  93         execute(dryRun);
  94     }
  95
  96     public String toString() {
  97         return sb.toString();
  98     }
  99 }