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                         // PER ORACLE, this isn't actually needed, but Sonar idiocy
  70                         // requires something or flags as error.
  71                         return null;
  72                     }
  73                 }
  74                 last = System.currentTimeMillis() + sleep;
  75             }
  76             return session.execute(sb.toString());
  77         } else {
  78             return null;
  79         }
  80     }
  81
  82     public ResultSet execute(boolean dryRun) {
  83         ResultSet rv = null;
  84         if(dryRun) {
  85             if(sleep>0) {
  86                 long left = last - System.currentTimeMillis();
  87                 if(left>0) {
  88                     try {
  89                         Thread.sleep(left);
  90                     } catch (InterruptedException e) {
  91                         // PER ORACLE, this isn't actually needed, but Sonar idiocy
  92                         // requires something or flags as error.
  93                         return null;
  94                     }
  95                 }
  96                 last = System.currentTimeMillis() + sleep;
  97             }
  98             end();
  99         } else {
 100             rv = execute();
 101         }
 102         sb.setLength(0);
 103         return rv;
 104     }
 105
 106     public ResultSet singleExec(StringBuilder query, boolean dryRun) {
 107         if(dryRun) {
 108             return null;
 109         } else {
 110             return session.execute(query.toString());
 111         }
 112     }
 113
 114     public void touch(String table, int begin, int end, boolean dryRun) {
 115         StringBuilder sb = begin();
 116         for(int i=begin; i<end; ++i) {
 117             sb.append("UPDATE cache SET touched=dateof(now()) WHERE name='");
 118             sb.append(table);
 119             sb.append("' AND seg=");
 120             sb.append(i);
 121             sb.append(";\n");
 122         }
 123         execute(dryRun);
 124     }
 125
 126     public void sleep(long j) {
 127         sleep = j*1000;
 128     }
 129
 130     public String toString() {
 131         return sb.toString();
 132     }
 133 }