auth/auth-batch/src/main/java/org/onap/aaf/auth/helpers/MiscID.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.Map;
  25 import java.util.TreeMap;
  26
  27 import org.onap.aaf.auth.BatchException;
  28 import org.onap.aaf.misc.env.Env;
  29 import org.onap.aaf.misc.env.TimeTaken;
  30 import org.onap.aaf.misc.env.Trans;
  31
  32 import com.datastax.driver.core.ResultSet;
  33 import com.datastax.driver.core.Row;
  34 import com.datastax.driver.core.Session;
  35 import com.datastax.driver.core.SimpleStatement;
  36 import com.datastax.driver.core.Statement;
  37
  38 public class MiscID  {
  39     public static final TreeMap<String,MiscID> data = new TreeMap<>();
  40     /*
  41     Sample Record
  42     aad890|mj9030|20040902|20120207
  43
  44     **** Field Definitions ****
  45     MISCID - AT&T Miscellaneous ID - Non-User ID (Types: Internal Mechanized ID, External Mechanized ID, Datagate ID, Customer ID, Vendor ID, Exchange Mail ID, CLEC ID, Specialized ID, Training ID)
  46     SPONSOR_ATTUID - ATTUID of MiscID Sponsor (Owner)
  47     CREATE_DATE - Date when MiscID was created
  48     LAST_RENEWAL_DATE - Date when MiscID Sponsorship was last renewed
  49     */
  50     public String id,sponsor,created,renewal;
  51
  52     private static final String fieldString = "id,created,sponsor,renewal";
  53
  54     /**
  55      * Load a Row of Strings (from CSV file).
  56      *
  57      * Be CAREFUL that the Row lists match the Fields above!!!  If this changes, change
  58      * 1) This Object
  59      * 2) DB "suits.cql"
  60      * 3) Alter existing Tables
  61      * @param row
  62      * @throws BatchException
  63      * @throws IllegalAccessException
  64      * @throws IllegalArgumentException
  65      */
  66     public void set(String row []) throws BatchException {
  67         if(row.length<4) {throw new BatchException("Row of MiscID_XRef is too short");}
  68         id      = row[0];
  69         sponsor = row[1];
  70         created = row[2];
  71         renewal = row[3];
  72     }
  73
  74     public void set(Row row) {
  75         id      = row.getString(0);
  76         sponsor = row.getString(1);
  77         created = row.getString(2);
  78         renewal = row.getString(3);
  79     }
  80
  81
  82     public static void load(Trans trans, Session session ) {
  83         load(trans, session,"SELECT " + fieldString + " FROM authz.miscid;",data);
  84     }
  85
  86     public static void load(Trans trans, Session session, Map<String,MiscID> map ) {
  87         load(trans, session,"SELECT " + fieldString + " FROM authz.miscid;",map);
  88     }
  89
  90     public static void loadOne(Trans trans, Session session, String id ) {
  91         load(trans, session,"SELECT " + fieldString + " FROM authz.miscid WHERE id ='" + id + "';", data);
  92     }
  93
  94     public static void load(Trans trans, Session session, String query, Map<String,MiscID> map) {
  95         trans.info().log( "query: " + query );
  96         TimeTaken tt = trans.start("Read MiscID", Env.REMOTE);
  97
  98         ResultSet results;
  99         try {
 100             Statement stmt = new SimpleStatement( query );
 101             results = session.execute(stmt);
 102         } finally {
 103             tt.done();
 104         }
 105         int count = 0;
 106         try {
 107             tt = trans.start("Load Map", Env.SUB);
 108             try {
 109                 for( Row row : results.all()) {
 110                     MiscID miscID = new MiscID();
 111                     miscID.set(row);
 112                     data.put(miscID.id,miscID);
 113                     ++count;
 114                 }
 115             } finally {
 116                 tt.done();
 117             }
 118         } finally {
 119             trans.info().log("Found",count,"miscID records");
 120         }
 121     }
 122
 123     /* (non-Javadoc)
 124      * @see java.lang.Object#hashCode()
 125      */
 126     @Override
 127     public int hashCode() {
 128         return id.hashCode();
 129     }
 130
 131     /* (non-Javadoc)
 132      * @see java.lang.Object#equals(java.lang.Object)
 133      */
 134     @Override
 135     public boolean equals(Object obj) {
 136         if(obj!=null && obj instanceof MiscID) {
 137             return id.equals(((MiscID)obj).id);
 138         }
 139         return false;
 140     }
 141
 142     public StringBuilder insertStmt() throws IllegalArgumentException, IllegalAccessException {
 143         StringBuilder sb = new StringBuilder("INSERT INTO authz.miscid (");
 144         sb.append(fieldString);
 145         sb.append(") VALUES ('");
 146         sb.append(id);
 147         sb.append("','");
 148         sb.append(sponsor);
 149         sb.append("','");
 150         sb.append(created);
 151         sb.append("','");
 152         sb.append(renewal);
 153         sb.append("')");
 154         return sb;
 155     }
 156
 157     public StringBuilder updateStmt(MiscID source) {
 158         StringBuilder sb = null;
 159         if(id.equals(source.id)) {
 160             sb = addField(sb,"sponser",sponsor,source.sponsor);
 161             sb = addField(sb,"created",created,source.created);
 162             sb = addField(sb,"renewal",renewal,source.renewal);
 163         }
 164         if(sb!=null) {
 165             sb.append(" WHERE id='");
 166             sb.append(id);
 167             sb.append('\'');
 168         }
 169         return sb;
 170     }
 171
 172     private StringBuilder addField(StringBuilder sb, String name, String a, String b) {
 173         if(!a.equals(b)) {
 174             if(sb==null) {
 175                 sb = new StringBuilder("UPDATE authz.miscid SET ");
 176             } else {
 177                 sb.append(',');
 178             }
 179             sb.append(name);
 180             sb.append("='");
 181             sb.append(b);
 182             sb.append('\'');
 183         }
 184         return sb;
 185     }
 186
 187
 188 }