auth/auth-batch/src/main/java/org/onap/aaf/auth/helpers/Future.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
  22 package org.onap.aaf.auth.helpers;
  23
  24 import java.nio.ByteBuffer;
  25 import java.util.ArrayList;
  26 import java.util.Date;
  27 import java.util.List;
  28 import java.util.Map;
  29 import java.util.TreeMap;
  30 import java.util.UUID;
  31
  32 import org.onap.aaf.auth.dao.cass.FutureDAO;
  33 import org.onap.aaf.auth.env.AuthzTrans;
  34 import org.onap.aaf.auth.layer.Result;
  35 import org.onap.aaf.misc.env.Env;
  36 import org.onap.aaf.misc.env.TimeTaken;
  37 import org.onap.aaf.misc.env.Trans;
  38 import org.onap.aaf.misc.env.util.Chrono;
  39
  40 import com.datastax.driver.core.ResultSet;
  41 import com.datastax.driver.core.Row;
  42 import com.datastax.driver.core.Session;
  43 import com.datastax.driver.core.SimpleStatement;
  44 import com.datastax.driver.core.Statement;
  45
  46 public class Future implements CacheChange.Data, Comparable<Future> {
  47     public static final Map<UUID,Future> data = new TreeMap<>();
  48     public static final Map<String,List<Future>> byRole = new TreeMap<>();
  49
  50     public final FutureDAO.Data fdd;
  51     public final String role; // derived
  52     private static final CacheChange<Future> cache = new CacheChange<>();
  53
  54
  55     public final UUID id() {
  56         return fdd.id;
  57     }
  58
  59     public final String memo() {
  60         return fdd.memo;
  61     }
  62
  63     public final String target() {
  64         return fdd.target;
  65     }
  66
  67     public final Date start() {
  68         return fdd.start;
  69     }
  70
  71     public final Date expires() {
  72         return fdd.expires;
  73     }
  74
  75
  76     public Future(UUID id, String memo, String target, Date start, Date expires, ByteBuffer construct) {
  77         fdd = new FutureDAO.Data();
  78         fdd.id = id;
  79         fdd.memo = memo;
  80         fdd.target = target;
  81         fdd.start = start;
  82         fdd.expires = expires;
  83         fdd.construct = construct;
  84         role = Approval.roleFromMemo(memo);
  85     }
  86
  87     public static void load(Trans trans, Session session, Creator<Future> creator) {
  88         trans.info().log( "query: " + creator.select() );
  89         ResultSet results;
  90         TimeTaken tt = trans.start("Load Futures", Env.REMOTE);
  91         try {
  92             Statement stmt = new SimpleStatement(creator.select());
  93             results = session.execute(stmt);
  94         } finally {
  95             tt.done();
  96         }
  97
  98         int count = 0;
  99         tt = trans.start("Process Futures", Env.SUB);
 100         try {
 101             for(Row row : results.all()) {
 102                 ++count;
 103                 Future f = creator.create(row);
 104                 data.put(f.fdd.id,f);
 105                 if(f.role==null) {
 106                     continue;
 107                 }
 108                 List<Future> lf = byRole.get(f.role);
 109                 if(lf==null) {
 110                     lf = new ArrayList<>();
 111                     byRole.put(f.role,lf);
 112                 }
 113                 lf.add(f);
 114
 115             }
 116         } finally {
 117             tt.done();
 118             trans.info().log("Found",count,"Futures");
 119         }
 120     }
 121
 122     public static Creator<Future> v2_0_17 = new Creator<Future>() {
 123         @Override
 124         public Future create(Row row) {
 125             return new Future(row.getUUID(0),row.getString(1),row.getString(2),
 126                     row.getTimestamp(3),row.getTimestamp(4), null);
 127         }
 128
 129         @Override
 130         public String select() {
 131             return "select id,memo,target,start,expires from authz.future";
 132         }
 133     };
 134
 135     public static Creator<Future> withConstruct = new Creator<Future>() {
 136         @Override
 137         public String select() {
 138             return "select id,memo,target,start,expires,construct from authz.future";
 139         }
 140
 141         @Override
 142         public Future create(Row row) {
 143             return new Future(row.getUUID(0),row.getString(1),row.getString(2),
 144                     row.getTimestamp(3),row.getTimestamp(4), row.getBytes(5));
 145         }
 146
 147     };
 148
 149     public Result<Void> delayedDelete(AuthzTrans trans, FutureDAO fd, boolean dryRun, String text) {
 150         Result<Void> rv;
 151         if(dryRun) {
 152             trans.info().log(text,"- Would Delete: ",fdd.id,fdd.memo,"expiring on",Chrono.dateOnlyStamp(fdd.expires));
 153             rv = Result.ok();
 154         } else {
 155             rv = fd.delete(trans, fdd, true); // need to read for undelete
 156             if(rv.isOK()) {
 157                 trans.info().log(text, "- Deleted:",fdd.id,fdd.memo,"expiring on",Chrono.dateOnlyStamp(fdd.expires));
 158                 cache.delayedDelete(this);
 159             } else {
 160                 if(rv.status!=6) {
 161                     trans.info().log(text,"- Failed to Delete Future", fdd.id);
 162                 }
 163             }
 164         }
 165         return rv;
 166     }
 167
 168     /* (non-Javadoc)
 169      * @see org.onap.aaf.auth.helpers.CacheChange.Data#resetLocalData()
 170      */
 171     @Override
 172     public void expunge() {
 173         data.remove(fdd.id);
 174         if(role!=null) {
 175             List<Future> lf = byRole.get(role);
 176             if(lf!=null) {
 177                 lf.remove(this);
 178             }
 179         }
 180     }
 181
 182     @Override
 183     public int compareTo(Future o) {
 184         if(o==null) {
 185             return -1;
 186         }
 187         return fdd.id.compareTo(o.fdd.id);
 188     }
 189
 190     public static void resetLocalData() {
 191         cache.resetLocalData();
 192     }
 193
 194     public static int sizeForDeletion() {
 195         return cache.cacheSize();
 196     }
 197
 198     public static boolean pendingDelete(Future f) {
 199         return cache.contains(f);
 200     }
 201
 202
 203 }