auth/auth-batch/src/main/java/org/onap/aaf/auth/helpers/Approval.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.util.ArrayList;
  25 import java.util.Date;
  26 import java.util.List;
  27 import java.util.TreeMap;
  28 import java.util.UUID;
  29
  30 import org.onap.aaf.auth.dao.cass.ApprovalDAO;
  31 import org.onap.aaf.auth.env.AuthzTrans;
  32 import org.onap.aaf.auth.layer.Result;
  33 import org.onap.aaf.misc.env.Env;
  34 import org.onap.aaf.misc.env.TimeTaken;
  35 import org.onap.aaf.misc.env.Trans;
  36
  37 import com.datastax.driver.core.ResultSet;
  38 import com.datastax.driver.core.Row;
  39 import com.datastax.driver.core.Session;
  40 import com.datastax.driver.core.SimpleStatement;
  41 import com.datastax.driver.core.Statement;
  42
  43 public class Approval implements CacheChange.Data  {
  44         public static final String RE_APPROVAL_IN_ROLE = "Re-Approval in Role '";
  45         public static final String RE_VALIDATE_ADMIN = "Re-Validate as Administrator for AAF Namespace '";
  46         public static final String RE_VALIDATE_OWNER = "Re-Validate Ownership for AAF Namespace '";
  47
  48         public static TreeMap<String,List<Approval>> byApprover = new TreeMap<>();
  49         public static TreeMap<String,List<Approval>> byUser = new TreeMap<>();
  50         public static TreeMap<UUID,List<Approval>> byTicket = new TreeMap<>();
  51         private final static CacheChange<Approval> cache = new CacheChange<>();
  52
  53         public final ApprovalDAO.Data add;
  54         private String role;
  55
  56         public Approval(UUID id, UUID ticket, String approver, Date last_notified,
  57                         String user, String memo, String operation, String status, String type, long updated) {
  58                 add = new ApprovalDAO.Data();
  59                 add.id = id;
  60                 add.ticket = ticket;
  61                 add.approver = approver;
  62                 add.last_notified = last_notified;
  63                 add.user = user;
  64                 add.memo = memo;
  65                 add.operation = operation;
  66                 add.status = status;
  67                 add.type = type;
  68                 add.updated = new Date(updated);
  69                 role = roleFromMemo(memo);
  70         }
  71
  72         public static String roleFromMemo(String memo) {
  73                 if(memo==null) {
  74                         return null;
  75                 }
  76                 int first = memo.indexOf('\'');
  77                 if(first>=0) {
  78                         int second = memo.indexOf('\'', ++first);
  79                         if(second>=0) {
  80                                 String role = memo.substring(first, second);
  81                                 if(memo.startsWith(RE_VALIDATE_ADMIN)) {
  82                                         return role + ".admin";
  83                                 } else if(memo.startsWith(RE_VALIDATE_OWNER)) {
  84                                         return role + ".owner";
  85                                 } else if(memo.startsWith(RE_APPROVAL_IN_ROLE)) {
  86                                         return role;
  87                                 }
  88                         }
  89                 }
  90                 return null;
  91         }
  92
  93         public static void load(Trans trans, Session session, Creator<Approval> creator ) {
  94                 trans.info().log( "query: " + creator.select() );
  95         TimeTaken tt = trans.start("Load Notify", Env.REMOTE);
  96
  97         ResultSet results;
  98                 try {
  99                 Statement stmt = new SimpleStatement(creator.select());
 100                 results = session.execute(stmt);
 101         } finally {
 102                 tt.done();
 103         }
 104                 int count = 0;
 105         tt = trans.start("Process Notify", Env.SUB);
 106
 107         try {
 108                         List<Approval> ln;
 109                         for(Row row : results.all()) {
 110                                 ++count;
 111                                 try {
 112                                                 Approval app = creator.create(row);
 113                                                 String person = app.getApprover();
 114                                                 if(person!=null) {
 115                                                 ln = byApprover.get(person);
 116                                                         if(ln==null) {
 117                                                                 ln = new ArrayList<Approval>();
 118                                                                 byApprover.put(app.getApprover(), ln);
 119                                                         }
 120                                                         ln.add(app);
 121                                                 }
 122
 123
 124                                         person = app.getUser();
 125                                                 if(person!=null) {
 126                                                         ln = byUser.get(person);
 127                                                         if(ln==null) {
 128                                                                 ln = new ArrayList<Approval>();
 129                                                                 byUser.put(app.getUser(), ln);
 130                                                         }
 131                                                         ln.add(app);
 132                                                 }
 133                                                 UUID ticket = app.getTicket();
 134                                                 if(ticket!=null) {
 135                                                         ln = byTicket.get(ticket);
 136                                                         if(ln==null) {
 137                                                                 ln = new ArrayList<Approval>();
 138                                                                 byTicket.put(app.getTicket(), ln);
 139                                                         }
 140                                                 ln.add(app);
 141                                                 }
 142                                 } finally {
 143                                         tt.done();
 144                                 }
 145                         }
 146         } finally {
 147                 tt.done();
 148                 trans.info().log("Found",count,"Approval Records");
 149         }
 150         }
 151
 152         @Override
 153         public void expunge() {
 154                 List<Approval> la = byApprover.get(getApprover());
 155                 if(la!=null) {
 156                         la.remove(this);
 157                 }
 158
 159                 la = byUser.get(getUser());
 160                 if(la!=null) {
 161                         la.remove(this);
 162                 }
 163                 UUID ticket = this.add==null?null:this.add.ticket;
 164                 if(ticket!=null) {
 165                         la = byTicket.get(this.add.ticket);
 166                         if(la!=null) {
 167                                 la.remove(this);
 168                         }
 169                 }
 170         }
 171
 172         public void update(AuthzTrans trans, ApprovalDAO apprDAO, boolean dryRun) {
 173                 if(dryRun) {
 174                         trans.info().printf("Would update Approval %s, %s, last_notified %s",add.id,add.status,add.last_notified);
 175                 } else {
 176                         trans.info().printf("Update Approval %s, %s, last_notified %s",add.id,add.status,add.last_notified);
 177                         apprDAO.update(trans, add);
 178                 }
 179         }
 180
 181         public static Creator<Approval> v2_0_17 = new Creator<Approval>() {
 182                 @Override
 183                 public Approval create(Row row) {
 184                         return new Approval(row.getUUID(0), row.getUUID(1), row.getString(2), row.getTimestamp(3),
 185                                         row.getString(4),row.getString(5),row.getString(6),row.getString(7),row.getString(8)
 186                                         ,row.getLong(9)/1000);
 187                 }
 188
 189                 @Override
 190                 public String select() {
 191                         return "select id,ticket,approver,last_notified,user,memo,operation,status,type,WRITETIME(status) from authz.approval";
 192                 }
 193         };
 194
 195         /**
 196          * @return the lastNotified
 197          */
 198         public Date getLast_notified() {
 199                 return add.last_notified;
 200         }
 201         /**
 202          * @param lastNotified the lastNotified to set
 203          */
 204         public void setLastNotified(Date last_notified) {
 205                 add.last_notified = last_notified;
 206         }
 207         /**
 208          * @return the status
 209          */
 210         public String getStatus() {
 211                 return add.status;
 212         }
 213         /**
 214          * @param status the status to set
 215          */
 216         public void setStatus(String status) {
 217                 add.status = status;
 218         }
 219         /**
 220          * @return the id
 221          */
 222         public UUID getId() {
 223                 return add.id;
 224         }
 225         /**
 226          * @return the ticket
 227          */
 228         public UUID getTicket() {
 229                 return add.ticket;
 230         }
 231         /**
 232          * @return the approver
 233          */
 234         public String getApprover() {
 235                 return add.approver;
 236         }
 237         /**
 238          * @return the user
 239          */
 240         public String getUser() {
 241                 return add.user;
 242         }
 243         /**
 244          * @return the memo
 245          */
 246         public String getMemo() {
 247                 return add.memo;
 248         }
 249         /**
 250          * @return the operation
 251          */
 252         public String getOperation() {
 253                 return add.operation;
 254         }
 255         /**
 256          * @return the type
 257          */
 258         public String getType() {
 259                 return add.type;
 260         }
 261         public void lapsed() {
 262                 add.ticket=null;
 263                 add.status="lapsed";
 264         }
 265
 266         public String getRole() {
 267                 return role;
 268         }
 269
 270         public String toString() {
 271                 return getUser() + ' ' + getMemo();
 272         }
 273
 274         public void delayDelete(AuthzTrans trans, ApprovalDAO ad, boolean dryRun, String text) {
 275                 if(dryRun) {
 276                         trans.info().log(text,"- Would Delete: Approval",getId(),"on ticket",getTicket(),"for",getApprover());
 277                 } else {
 278                         Result<Void> rv = ad.delete(trans, add, false);
 279                         if(rv.isOK()) {
 280                                 trans.info().log(text,"- Deleted: Approval",getId(),"on ticket",getTicket(),"for",getApprover());
 281                                 cache.delayedDelete(this);
 282                         } else {
 283                                 trans.info().log(text,"- Failed to Delete Approval",getId());
 284                         }
 285                 }
 286         }
 287
 288
 289         public static void resetLocalData() {
 290                 cache.resetLocalData();
 291         }
 292
 293         public static int sizeForDeletion() {
 294                 return cache.cacheSize();
 295         }
 296
 297         public static void delayDelete(AuthzTrans noAvg, ApprovalDAO apprDAO, boolean dryRun, List<Approval> list, String text) {
 298                 if(list!=null) {
 299                         for(Approval a : list) {
 300                                 a.delayDelete(noAvg, apprDAO, dryRun,text);
 301                         }
 302                 }
 303         }
 304
 305         public static boolean pendingDelete(Approval a) {
 306                 return cache.contains(a);
 307         }
 308
 309 }