auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/helpers/CQLBatchLoop.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 public class CQLBatchLoop {
  24         private static final int MAX_CHARS = (50 * 1024)/2;
  25
  26         private final CQLBatch cqlBatch;
  27         private final int maxBatch;
  28         private final StringBuilder sb;
  29         private final boolean dryRun;
  30         private int i;
  31         private int total;
  32         private int batches;
  33         private final StringBuilder current;
  34         private boolean showProgress;
  35
  36         public CQLBatchLoop(CQLBatch cb, int max, boolean dryRun) {
  37                 cqlBatch = cb;
  38                 i=0;
  39                 total = 0;
  40                 maxBatch = max;
  41                 sb = cqlBatch.begin();
  42                 current = new StringBuilder();
  43                 this.dryRun = dryRun;
  44                 showProgress = false;
  45         }
  46
  47         public CQLBatchLoop showProgress() {
  48                 showProgress = true;
  49                 return this;
  50         }
  51         /**
  52          * Assume this is another line in the Batch
  53          * @return
  54          */
  55         public StringBuilder inc() {
  56                 if(i>=maxBatch || current.length()+sb.length()>MAX_CHARS) {
  57                         if(i>0) {
  58                                 cqlBatch.execute(dryRun);
  59                                 i = -1;
  60                                 incBatch();
  61                         }
  62                 }
  63                 if(i<0) {
  64                         cqlBatch.begin();
  65                         i=0;
  66                 }
  67                 if(current.length() > MAX_CHARS) {
  68                         cqlBatch.singleExec(current, dryRun);
  69                 } else {
  70                         sb.append(current);
  71                 }
  72                 current.setLength(0);
  73                 ++i;
  74                 ++total;
  75                 return current;
  76         }
  77
  78         /**
  79          * Close up when finished.
  80          */
  81         public void flush() {
  82                 if(current.length()+sb.length()>MAX_CHARS) {
  83                         if(i>0) {
  84                                 cqlBatch.execute(dryRun);
  85                                 incBatch();
  86                         }
  87                         if(current.length()>0) {
  88                                 cqlBatch.singleExec(current, dryRun);
  89                                 current.setLength(0);
  90                                 incBatch();
  91                         }
  92                 } else {
  93                         if(i<0) {
  94                                 cqlBatch.begin();
  95                         }
  96                         sb.append(current);
  97                         current.setLength(0);
  98                         cqlBatch.execute(dryRun);
  99                         incBatch();
 100                 }
 101                 i=-1;
 102         }
 103
 104         private void incBatch() {
 105                 ++batches;
 106                 if(showProgress) {
 107                         System.out.print('.');
 108                         if(batches%70==0) {
 109                                 System.out.println();
 110                         }
 111                 }
 112         }
 113
 114         public int total() {
 115                 return total;
 116         }
 117
 118         public int batches() {
 119                 return batches;
 120         }
 121
 122         public void reset() {
 123                 total = 0;
 124                 batches = 0;
 125                 i = -1;
 126         }
 127
 128         public String toString() {
 129                 return cqlBatch.toString();
 130         }
 131 }