authz-batch/src/main/java/com/att/authz/helpers/NsAttrib.java

   1 /*******************************************************************************
   2  * Copyright (c) 2016 AT&T Intellectual Property. All rights reserved.
   3  *******************************************************************************/
   4 package com.att.authz.helpers;
   5
   6 import java.util.ArrayList;
   7 import java.util.List;
   8 import java.util.TreeMap;
   9
  10 import org.onap.aaf.inno.env.Env;
  11 import org.onap.aaf.inno.env.TimeTaken;
  12 import org.onap.aaf.inno.env.Trans;
  13 import com.datastax.driver.core.ResultSet;
  14 import com.datastax.driver.core.Row;
  15 import com.datastax.driver.core.Session;
  16 import com.datastax.driver.core.SimpleStatement;
  17 import com.datastax.driver.core.Statement;
  18
  19 public class NsAttrib  {
  20         public static final List<NsAttrib> data = new ArrayList<NsAttrib>();
  21     public static final TreeMap<String,List<NsAttrib>> byKey = new TreeMap<String,List<NsAttrib>>();
  22     public static final TreeMap<String,List<NsAttrib>> byNS = new TreeMap<String,List<NsAttrib>>();
  23
  24         public final String ns,key,value;
  25
  26         public NsAttrib(String ns, String key, String value) {
  27                 this.ns = ns;
  28                 this.key = key;
  29                 this.value = value;
  30         }
  31
  32         public static void load(Trans trans, Session session, Creator<NsAttrib> creator ) {
  33                 trans.info().log( "query: " + creator.select() );
  34         ResultSet results;
  35         TimeTaken tt = trans.start("Load NsAttributes", Env.REMOTE);
  36                 try {
  37                 Statement stmt = new SimpleStatement(creator.select());
  38                 results = session.execute(stmt);
  39         } finally {
  40                 tt.done();
  41         }
  42                 int count = 0;
  43         tt = trans.start("Process NsAttributes", Env.SUB);
  44
  45         try {
  46                 for(Row row : results.all()) {
  47                         ++count;
  48                         NsAttrib ur = creator.create(row);
  49                         data.add(ur);
  50
  51                         List<NsAttrib> lna = byKey.get(ur.key);
  52                         if(lna==null) {
  53                                 lna = new ArrayList<NsAttrib>();
  54                                 byKey.put(ur.key, lna);
  55                         }
  56                         lna.add(ur);
  57
  58                         lna = byNS.get(ur.ns);
  59                         if(lna==null) {
  60                                 lna = new ArrayList<NsAttrib>();
  61                                 byNS.put(ur.ns, lna);
  62                         }
  63                         lna.add(ur);
  64                 }
  65         } finally {
  66                 tt.done();
  67                 trans.info().log("Found",count,"NS Attributes");
  68         }
  69         }
  70
  71         public static Creator<NsAttrib> v2_0_11 = new Creator<NsAttrib>() {
  72                 @Override
  73                 public NsAttrib create(Row row) {
  74                         return new NsAttrib(row.getString(0), row.getString(1), row.getString(2));
  75                 }
  76
  77                 @Override
  78                 public String select() {
  79                         return "select ns,key,value from authz.ns_attrib";
  80                 }
  81         };
  82
  83
  84         public String toString() {
  85                 return "\"" + ns + "\",\"" + key + "\",\""  + value;
  86         }
  87
  88 }