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.cadi.Access;
  24 import org.onap.aaf.misc.env.LogTarget;
  25
  26 import com.datastax.driver.core.ResultSet;
  27 import com.datastax.driver.core.Session;
  28
  29 public class CQLBatch {
  30     private Session session;
  31     private StringBuilder sb;
  32     private int hasAdded;
  33     private LogTarget log;
  34     private long sleep;
  35     private long last;
  36
  37     public CQLBatch(LogTarget log, Session session) {
  38         this.log = log;
  39         this.session = session;
  40         sb = new StringBuilder();
  41         hasAdded = 0;
  42         sleep = 0L;
  43     }
  44     public StringBuilder begin() {
  45         sb.setLength(0);
  46         sb.append("BEGIN BATCH\n");
  47         hasAdded = sb.length();
  48         return sb;
  49     }
  50
  51     private boolean end() {
  52         if(sb.length()==hasAdded) {
  53             return false;
  54         } else {
  55             sb.append("APPLY BATCH;\n");
  56             log.log(sb);
  57             return true;
  58         }
  59     }
  60
  61     public ResultSet execute() {
  62         if(end()) {
  63             if(sleep>0) {
  64                 long left = last - System.currentTimeMillis();
  65                 if(left>0) {
  66                     try {
  67                         Thread.sleep(left);
  68                     } catch (InterruptedException e) {
  69                         Access.NULL.log(e); // Keep code check idiocy at bay
  70                     }
  71                 }
  72                 last = System.currentTimeMillis()+sleep;
  73             }
  74             return session.execute(sb.toString());
  75         } else {
  76             return null;
  77         }
  78     }
  79
  80     public ResultSet execute(boolean dryRun) {
  81         ResultSet rv = null;
  82         if(dryRun) {
  83             if(sleep>0) {
  84                 long left = last - System.currentTimeMillis();
  85                 if(left>0) {
  86                     try {
  87                         Thread.sleep(left);
  88                     } catch (InterruptedException e) {
  89                         Access.NULL.log(e); // Keep code check idiocy at bay
  90                     }
  91                 }
  92                 last = System.currentTimeMillis()+sleep;
  93             }
  94             end();
  95         } else {
  96             rv = execute();
  97         }
  98         sb.setLength(0);
  99         return rv;
 100     }
 101
 102     public ResultSet singleExec(StringBuilder query, boolean dryRun) {
 103         if(dryRun) {
 104             return null;
 105         } else {
 106             return session.execute(query.toString());
 107         }
 108     }
 109
 110     public void touch(String table, int begin, int end, boolean dryRun) {
 111         StringBuilder sb = begin();
 112         for(int i=begin;i<end;++i) {
 113             sb.append("UPDATE cache SET touched=dateof(now()) WHERE name='");
 114             sb.append(table);
 115             sb.append("' AND seg=");
 116             sb.append(i);
 117             sb.append(";\n");
 118         }
 119         execute(dryRun);
 120     }
 121
 122     public void sleep(int j) {
 123         sleep = j*1000;
 124     }
 125
 126     public String toString() {
 127         return sb.toString();
 128     }
 129 }