[AAF-21] Initial code import

[aaf/authz.git] / authz-batch / src / main / java / com / att / authz / helpers / NS.java

diff --git a/authz-batch/src/main/java/com/att/authz/helpers/NS.java b/authz-batch/src/main/java/com/att/authz/helpers/NS.java
new file mode 100644 (file)
index 0000000..a97b2d2
--- /dev/null
+++ b/authz-batch/src/main/java/com/att/authz/helpers/NS.java
@@ -0,0 +1,134 @@
+/*******************************************************************************
+ * Copyright (c) 2016 AT&T Intellectual Property. All rights reserved.
+ *******************************************************************************/
+package com.att.authz.helpers;
+
+import java.util.Iterator;
+import java.util.Map;
+import java.util.TreeMap;
+
+import com.att.inno.env.Env;
+import com.att.inno.env.TimeTaken;
+import com.att.inno.env.Trans;
+import com.datastax.driver.core.ResultSet;
+import com.datastax.driver.core.Row;
+import com.datastax.driver.core.Session;
+import com.datastax.driver.core.SimpleStatement;
+import com.datastax.driver.core.Statement;
+
+public class NS implements Comparable<NS> {
+       public final static Map<String,NS> data = new TreeMap<String,NS>();
+
+       public final String name, description, parent;
+       public final int scope,type;
+
+       public NS(String name, String description, String parent, int type, int scope) {
+               this.name = name;
+               this.description = description;
+               this.parent = parent;
+               this.scope = scope;
+               this.type = type;
+       }
+       
+       public static void load(Trans trans, Session session, Creator<NS> creator) {
+               load(trans,session,
+                               "select name, description, parent, type, scope from authz.ns;"
+                               ,creator);
+       }
+       
+       public static void loadOne(Trans trans, Session session, Creator<NS> creator, String ns) {
+           load(trans,session,
+                               ("select name, description, parent, type, scope from authz.ns WHERE name='"+ns+"';")
+                               ,creator
+                               );
+       }
+
+       private static void load(Trans trans, Session session, String query, Creator<NS> creator) {
+        trans.info().log( "query: " + query );
+        ResultSet results;
+        TimeTaken tt;
+
+        tt = trans.start("Read Namespaces", Env.REMOTE);
+        try {
+               Statement stmt = new SimpleStatement( query );
+               results = session.execute(stmt);
+        } finally {
+               tt.done();
+        }
+        
+
+        try {
+               Iterator<Row> iter = results.iterator();
+               Row row;
+               tt = trans.start("Load Namespaces", Env.SUB);
+               try {
+                       while(iter.hasNext()) {
+                               row = iter.next();
+                               NS ns = creator.create(row);
+                               data.put(ns.name,ns);
+                       }
+               } finally {
+                       tt.done();
+               }
+        } finally {
+               trans.info().log("Found",data.size(),"Namespaces");
+        }
+
+       }
+
+       public String toString() {
+               return name;
+       }
+
+       /* (non-Javadoc)
+        * @see java.lang.Object#hashCode()
+        */
+       @Override
+       public int hashCode() {
+               return name.hashCode();
+       }
+
+       /* (non-Javadoc)
+        * @see java.lang.Object#equals(java.lang.Object)
+        */
+       @Override
+       public boolean equals(Object obj) {
+               return name.equals(obj);
+       }
+
+       @Override
+       public int compareTo(NS o) {
+               return name.compareTo(o.name);
+       }
+       
+       public static class NSSplit {
+               public String ns;
+               public String other;
+               public NSSplit(String s, int dot) {
+                       ns = s.substring(0,dot);
+                       other = s.substring(dot+1);
+               }
+       }
+       public static NSSplit deriveParent(String dotted) {
+               if(dotted==null)return null;
+               for(int idx = dotted.lastIndexOf('.');idx>=0; idx=dotted.lastIndexOf('.',idx-1)) {
+                       if(data.get(dotted.substring(0, idx))!=null) {
+                               return new NSSplit(dotted,idx);
+                       }
+               }
+               return null;
+       }
+       
+       public static Creator<NS> v2_0_11 = new Creator<NS> () {
+               @Override
+               public NS create(Row row) {
+                       return new NS(row.getString(0),row.getString(1), row.getString(2),row.getInt(3),row.getInt(4));
+               }
+               
+               @Override
+               public String select() {
+                       return "SELECT name, description, parent, type, scope FROM authz.ns ";
+               }
+       };
+
+}
\ No newline at end of file