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                 if(dryRun) {
  67                         end();
  68                         return null;
  69                 } else {
  70                         return execute();
  71                 }
  72         }
  73
  74         public void touch(String table, int begin, int end, boolean dryRun) {
  75                 StringBuilder sb = begin();
  76                 for(int i=begin;i<end;++i) {
  77                         sb.append("UPDATE cache SET touched=dateof(now()) WHERE name='");
  78                         sb.append(table);
  79                         sb.append("' AND seg=");
  80                         sb.append(i);
  81                         sb.append(";\n");
  82                 }
  83                 execute(dryRun);
  84         }
  85 }