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
  25         private final CQLBatch cqlBatch;
  26         private final int maxBatch;
  27         private final StringBuilder sb;
  28         private final boolean dryRun;
  29         private int i;
  30
  31         public CQLBatchLoop(CQLBatch cb, int max, boolean dryRun) {
  32                 cqlBatch = cb;
  33                 i=0;
  34                 maxBatch = max;
  35                 sb = cqlBatch.begin();
  36                 this.dryRun = dryRun;
  37         }
  38
  39         /**
  40          * Put at the first part of your Loop Logic... It checks if you have enough lines to
  41          * push a batch.
  42          */
  43         public void preLoop() {
  44                 if(i<0) {
  45                         cqlBatch.begin();
  46                 } else if(i>=maxBatch) {
  47                         cqlBatch.execute(dryRun);
  48                         cqlBatch.begin();
  49                         i=0;
  50                 }
  51         }
  52
  53         /**
  54          * Assume this is another line in the Batch
  55          * @return
  56          */
  57         public StringBuilder inc() {
  58                 ++i;
  59                 return sb;
  60         }
  61
  62         /**
  63          * Close up when done.  However, can go back to "preLoop" safely.
  64          */
  65         public void flush() {
  66                 cqlBatch.execute(dryRun);
  67                 i=-1;
  68         }
  69 }