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 }