--- /dev/null
+/*******************************************************************************\r
+ * ============LICENSE_START====================================================\r
+ * * org.onap.aaf\r
+ * * ===========================================================================\r
+ * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.\r
+ * * ===========================================================================\r
+ * * Licensed under the Apache License, Version 2.0 (the "License");\r
+ * * you may not use this file except in compliance with the License.\r
+ * * You may obtain a copy of the License at\r
+ * * \r
+ * * http://www.apache.org/licenses/LICENSE-2.0\r
+ * * \r
+ * * Unless required by applicable law or agreed to in writing, software\r
+ * * distributed under the License is distributed on an "AS IS" BASIS,\r
+ * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
+ * * See the License for the specific language governing permissions and\r
+ * * limitations under the License.\r
+ * * ============LICENSE_END====================================================\r
+ * *\r
+ * * ECOMP is a trademark and service mark of AT&T Intellectual Property.\r
+ * *\r
+ ******************************************************************************/\r
+package org.onap.aaf.dao.aaf.cass;\r
+\r
+import java.io.ByteArrayOutputStream;\r
+import java.io.DataInputStream;\r
+import java.io.DataOutputStream;\r
+import java.io.IOException;\r
+import java.nio.ByteBuffer;\r
+import java.util.HashMap;\r
+import java.util.HashSet;\r
+import java.util.Iterator;\r
+import java.util.List;\r
+import java.util.Map;\r
+import java.util.Map.Entry;\r
+\r
+import org.onap.aaf.authz.env.AuthzTrans;\r
+import org.onap.aaf.authz.layer.Result;\r
+import org.onap.aaf.dao.Bytification;\r
+import org.onap.aaf.dao.Cached;\r
+import org.onap.aaf.dao.CassAccess;\r
+import org.onap.aaf.dao.CassDAOImpl;\r
+import org.onap.aaf.dao.Loader;\r
+import org.onap.aaf.dao.Streamer;\r
+\r
+import java.util.Set;\r
+\r
+import org.onap.aaf.inno.env.APIException;\r
+import org.onap.aaf.inno.env.Env;\r
+import org.onap.aaf.inno.env.TimeTaken;\r
+import com.datastax.driver.core.Cluster;\r
+import com.datastax.driver.core.ResultSet;\r
+import com.datastax.driver.core.Row;\r
+import com.datastax.driver.core.exceptions.DriverException;\r
+\r
+/**\r
+ * NsDAO\r
+ * \r
+ * Data Access Object for Namespace Data\r
+ *\r
+ */\r
+public class NsDAO extends CassDAOImpl<AuthzTrans,NsDAO.Data> {\r
+ public static final String TABLE = "ns";\r
+ public static final String TABLE_ATTRIB = "ns_attrib";\r
+ public static final int CACHE_SEG = 0x40; // yields segment 0x0-0x3F\r
+ public static final int ROOT = 1;\r
+ public static final int COMPANY=2;\r
+ public static final int APP = 3;\r
+\r
+ private static final String BEGIN_BATCH = "BEGIN BATCH\n";\r
+ private static final String APPLY_BATCH = "APPLY BATCH;\n";\r
+ private static final String SQSCCR = "';\n";\r
+ private static final String SQCSQ = "','";\r
+ \r
+ private HistoryDAO historyDAO;\r
+ private CacheInfoDAO infoDAO;\r
+ private PSInfo psNS;\r
+\r
+ public NsDAO(AuthzTrans trans, Cluster cluster, String keyspace) throws APIException, IOException {\r
+ super(trans, NsDAO.class.getSimpleName(),cluster,keyspace,Data.class,TABLE, readConsistency(trans,TABLE), writeConsistency(trans,TABLE));\r
+ init(trans);\r
+ }\r
+\r
+ public NsDAO(AuthzTrans trans, HistoryDAO hDAO, CacheInfoDAO iDAO) throws APIException, IOException {\r
+ super(trans, NsDAO.class.getSimpleName(),hDAO,Data.class,TABLE, readConsistency(trans,TABLE), writeConsistency(trans,TABLE));\r
+ historyDAO=hDAO;\r
+ infoDAO = iDAO;\r
+ init(trans);\r
+ }\r
+\r
+\r
+ //////////////////////////////////////////\r
+ // Data Definition, matches Cassandra DM\r
+ //////////////////////////////////////////\r
+ private static final int KEYLIMIT = 1;\r
+ /**\r
+ * Data class that matches the Cassandra Table "role"\r
+ * \r
+ */\r
+ public static class Data extends CacheableData implements Bytification {\r
+ public String name;\r
+ public int type;\r
+ public String description;\r
+ public String parent;\r
+ public Map<String,String> attrib;\r
+\r
+// ////////////////////////////////////////\r
+// // Getters\r
+ public Map<String,String> attrib(boolean mutable) {\r
+ if (attrib == null) {\r
+ attrib = new HashMap<String,String>();\r
+ } else if (mutable && !(attrib instanceof HashMap)) {\r
+ attrib = new HashMap<String,String>(attrib);\r
+ }\r
+ return attrib;\r
+ }\r
+\r
+ @Override\r
+ public int[] invalidate(Cached<?,?> cache) {\r
+ return new int[] {\r
+ seg(cache,name)\r
+ };\r
+ }\r
+\r
+ public NsSplit split(String name) {\r
+ return new NsSplit(this,name);\r
+ }\r
+\r
+ @Override\r
+ public ByteBuffer bytify() throws IOException {\r
+ ByteArrayOutputStream baos = new ByteArrayOutputStream();\r
+ NSLoader.deflt.marshal(this,new DataOutputStream(baos));\r
+ return ByteBuffer.wrap(baos.toByteArray());\r
+ }\r
+ \r
+ @Override\r
+ public void reconstitute(ByteBuffer bb) throws IOException {\r
+ NSLoader.deflt.unmarshal(this,toDIS(bb));\r
+ }\r
+ \r
+ @Override\r
+ public String toString() {\r
+ return name;\r
+ }\r
+ \r
+ }\r
+ \r
+ private void init(AuthzTrans trans) throws APIException, IOException {\r
+ // Set up sub-DAOs\r
+ if(historyDAO==null) {\r
+ historyDAO = new HistoryDAO(trans, this);\r
+ }\r
+ if(infoDAO==null) {\r
+ infoDAO = new CacheInfoDAO(trans,this);\r
+ }\r
+\r
+ String[] helpers = setCRUD(trans, TABLE, Data.class, NSLoader.deflt,4/*need to skip attrib */);\r
+ \r
+ psNS = new PSInfo(trans, SELECT_SP + helpers[FIELD_COMMAS] + " FROM " + TABLE +\r
+ " WHERE parent = ?", new NSLoader(1),readConsistency);\r
+\r
+ }\r
+ \r
+ private static final class NSLoader extends Loader<Data> implements Streamer<Data> {\r
+ public static final int MAGIC=250935515;\r
+ public static final int VERSION=1;\r
+ public static final int BUFF_SIZE=48;\r
+\r
+ public static final NSLoader deflt = new NSLoader(KEYLIMIT);\r
+ \r
+ public NSLoader(int keylimit) {\r
+ super(keylimit);\r
+ }\r
+\r
+ @Override\r
+ public Data load(Data data, Row row) {\r
+ // Int more efficient\r
+ data.name = row.getString(0);\r
+ data.type = row.getInt(1);\r
+ data.description = row.getString(2);\r
+ data.parent = row.getString(3);\r
+ return data;\r
+ }\r
+\r
+ @Override\r
+ protected void key(Data data, int idx, Object[] obj) {\r
+ obj[idx]=data.name;\r
+ }\r
+\r
+ @Override\r
+ protected void body(Data data, int _idx, Object[] obj) {\r
+ int idx = _idx;\r
+\r
+ obj[idx]=data.type;\r
+ obj[++idx]=data.description;\r
+ obj[++idx]=data.parent;\r
+ }\r
+ \r
+ @Override\r
+ public void marshal(Data data, DataOutputStream os) throws IOException {\r
+ writeHeader(os,MAGIC,VERSION);\r
+ writeString(os, data.name);\r
+ os.writeInt(data.type);\r
+ writeString(os,data.description);\r
+ writeString(os,data.parent);\r
+ if(data.attrib==null) {\r
+ os.writeInt(-1);\r
+ } else {\r
+ os.writeInt(data.attrib.size());\r
+ for(Entry<String, String> es : data.attrib(false).entrySet()) {\r
+ writeString(os,es.getKey());\r
+ writeString(os,es.getValue());\r
+ }\r
+ }\r
+ }\r
+\r
+ @Override\r
+ public void unmarshal(Data data, DataInputStream is) throws IOException {\r
+ /*int version = */readHeader(is,MAGIC,VERSION);\r
+ // If Version Changes between Production runs, you'll need to do a switch Statement, and adequately read in fields\r
+ \r
+ byte[] buff = new byte[BUFF_SIZE];\r
+ data.name = readString(is, buff);\r
+ data.type = is.readInt();\r
+ data.description = readString(is,buff);\r
+ data.parent = readString(is,buff);\r
+ int count = is.readInt();\r
+ if(count>0) {\r
+ Map<String, String> da = data.attrib(true);\r
+ for(int i=0;i<count;++i) {\r
+ da.put(readString(is,buff), readString(is,buff));\r
+ }\r
+ }\r
+ }\r
+\r
+ }\r
+ \r
+ @Override\r
+ public Result<Data> create(AuthzTrans trans, Data data) {\r
+ String ns = data.name;\r
+ // Ensure Parent is set\r
+ int ldot = ns.lastIndexOf('.');\r
+ data.parent=ldot<0?".":ns.substring(0,ldot);\r
+\r
+ // insert Attributes\r
+ StringBuilder stmt = new StringBuilder();\r
+ stmt.append(BEGIN_BATCH);\r
+ attribInsertStmts(stmt, data);\r
+ stmt.append(APPLY_BATCH);\r
+ try {\r
+ getSession(trans).execute(stmt.toString());\r
+//// TEST CODE for Exception \r
+// boolean force = true; \r
+// if(force) {\r
+// throw new com.datastax.driver.core.exceptions.NoHostAvailableException(new HashMap<InetSocketAddress,Throwable>());\r
+//// throw new com.datastax.driver.core.exceptions.AuthenticationException(new InetSocketAddress(9999),"Sample Message");\r
+// }\r
+////END TEST CODE\r
+\r
+ } catch (DriverException | APIException | IOException e) {\r
+ reportPerhapsReset(trans,e);\r
+ trans.info().log(stmt);\r
+ return Result.err(Result.ERR_Backend, "Backend Access");\r
+ }\r
+ return super.create(trans, data);\r
+ }\r
+\r
+ @Override\r
+ public Result<Void> update(AuthzTrans trans, Data data) {\r
+ String ns = data.name;\r
+ // Ensure Parent is set\r
+ int ldot = ns.lastIndexOf('.');\r
+ data.parent=ldot<0?".":ns.substring(0,ldot);\r
+\r
+ StringBuilder stmt = new StringBuilder();\r
+ stmt.append(BEGIN_BATCH);\r
+ try {\r
+ Map<String, String> localAttr = data.attrib;\r
+ Result<Map<String, String>> rremoteAttr = readAttribByNS(trans,ns);\r
+ if(rremoteAttr.notOK()) {\r
+ return Result.err(rremoteAttr);\r
+ }\r
+ // update Attributes\r
+ String str;\r
+ for(Entry<String, String> es : localAttr.entrySet()) {\r
+ str = rremoteAttr.value.get(es.getKey());\r
+ if(str==null || !str.equals(es.getValue())) {\r
+ attribInsertStmt(stmt, ns, es.getKey(),es.getValue());\r
+ }\r
+ }\r
+ \r
+ // No point in deleting... insert overwrites...\r
+// for(Entry<String, String> es : remoteAttr.entrySet()) {\r
+// str = localAttr.get(es.getKey());\r
+// if(str==null || !str.equals(es.getValue())) {\r
+// attribDeleteStmt(stmt, ns, es.getKey());\r
+// }\r
+// }\r
+ if(stmt.length()>BEGIN_BATCH.length()) {\r
+ stmt.append(APPLY_BATCH);\r
+ getSession(trans).execute(stmt.toString());\r
+ }\r
+ } catch (DriverException | APIException | IOException e) {\r
+ reportPerhapsReset(trans,e);\r
+ trans.info().log(stmt);\r
+ return Result.err(Result.ERR_Backend, CassAccess.ERR_ACCESS_MSG);\r
+ }\r
+\r
+ return super.update(trans,data);\r
+ }\r
+\r
+ /* (non-Javadoc)\r
+ * @see org.onap.aaf.dao.CassDAOImpl#read(org.onap.aaf.inno.env.TransStore, java.lang.Object)\r
+ */\r
+ @Override\r
+ public Result<List<Data>> read(AuthzTrans trans, Data data) {\r
+ Result<List<Data>> rld = super.read(trans, data);\r
+ \r
+ if(rld.isOKhasData()) {\r
+ for(Data d : rld.value) {\r
+ // Note: Map is null at this point, save time/mem by assignment\r
+ Result<Map<String, String>> rabn = readAttribByNS(trans,d.name);\r
+ if(rabn.isOK()) {\r
+ d.attrib = rabn.value;\r
+ } else {\r
+ return Result.err(rabn);\r
+ }\r
+ }\r
+ }\r
+ return rld;\r
+ }\r
+\r
+ /* (non-Javadoc)\r
+ * @see org.onap.aaf.dao.CassDAOImpl#read(org.onap.aaf.inno.env.TransStore, java.lang.Object[])\r
+ */\r
+ @Override\r
+ public Result<List<Data>> read(AuthzTrans trans, Object... key) {\r
+ Result<List<Data>> rld = super.read(trans, key);\r
+\r
+ if(rld.isOKhasData()) {\r
+ for(Data d : rld.value) {\r
+ // Note: Map is null at this point, save time/mem by assignment\r
+ Result<Map<String, String>> rabn = readAttribByNS(trans,d.name);\r
+ if(rabn.isOK()) {\r
+ d.attrib = rabn.value;\r
+ } else {\r
+ return Result.err(rabn);\r
+ }\r
+ }\r
+ }\r
+ return rld;\r
+ }\r
+\r
+ @Override\r
+ public Result<Void> delete(AuthzTrans trans, Data data, boolean reread) {\r
+ TimeTaken tt = trans.start("Delete NS Attributes " + data.name, Env.REMOTE);\r
+ try {\r
+ StringBuilder stmt = new StringBuilder();\r
+ attribDeleteAllStmt(stmt, data);\r
+ try {\r
+ getSession(trans).execute(stmt.toString());\r
+ } catch (DriverException | APIException | IOException e) {\r
+ reportPerhapsReset(trans,e);\r
+ trans.info().log(stmt);\r
+ return Result.err(Result.ERR_Backend, CassAccess.ERR_ACCESS_MSG);\r
+ }\r
+ } finally {\r
+ tt.done();\r
+ }\r
+ return super.delete(trans, data, reread);\r
+\r
+ }\r
+ \r
+ public Result<Map<String,String>> readAttribByNS(AuthzTrans trans, String ns) {\r
+ Map<String,String> map = new HashMap<String,String>();\r
+ TimeTaken tt = trans.start("readAttribByNS " + ns, Env.REMOTE);\r
+ try {\r
+ ResultSet rs = getSession(trans).execute("SELECT key,value FROM " \r
+ + TABLE_ATTRIB \r
+ + " WHERE ns='"\r
+ + ns\r
+ + "';");\r
+ \r
+ for(Iterator<Row> iter = rs.iterator();iter.hasNext(); ) {\r
+ Row r = iter.next();\r
+ map.put(r.getString(0), r.getString(1));\r
+ }\r
+ } catch (DriverException | APIException | IOException e) {\r
+ reportPerhapsReset(trans,e);\r
+ return Result.err(Result.ERR_Backend, CassAccess.ERR_ACCESS_MSG);\r
+ } finally {\r
+ tt.done();\r
+ }\r
+ return Result.ok(map);\r
+ }\r
+\r
+ public Result<Set<String>> readNsByAttrib(AuthzTrans trans, String key) {\r
+ Set<String> set = new HashSet<String>();\r
+ TimeTaken tt = trans.start("readNsBykey " + key, Env.REMOTE);\r
+ try {\r
+ ResultSet rs = getSession(trans).execute("SELECT ns FROM " \r
+ + TABLE_ATTRIB \r
+ + " WHERE key='"\r
+ + key\r
+ + "';");\r
+ \r
+ for(Iterator<Row> iter = rs.iterator();iter.hasNext(); ) {\r
+ Row r = iter.next();\r
+ set.add(r.getString(0));\r
+ }\r
+ } catch (DriverException | APIException | IOException e) {\r
+ reportPerhapsReset(trans,e);\r
+ return Result.err(Result.ERR_Backend, CassAccess.ERR_ACCESS_MSG);\r
+ } finally {\r
+ tt.done();\r
+ }\r
+ return Result.ok(set);\r
+ }\r
+\r
+ public Result<Void> attribAdd(AuthzTrans trans, String ns, String key, String value) {\r
+ try {\r
+ getSession(trans).execute(attribInsertStmt(new StringBuilder(),ns,key,value).toString());\r
+ return Result.ok();\r
+ } catch (DriverException | APIException | IOException e) {\r
+ reportPerhapsReset(trans,e);\r
+ return Result.err(Result.ERR_Backend, CassAccess.ERR_ACCESS_MSG);\r
+ }\r
+ }\r
+ \r
+ private StringBuilder attribInsertStmt(StringBuilder sb, String ns, String key, String value) {\r
+ sb.append("INSERT INTO ");\r
+ sb.append(TABLE_ATTRIB);\r
+ sb.append(" (ns,key,value) VALUES ('");\r
+ sb.append(ns);\r
+ sb.append(SQCSQ);\r
+ sb.append(key);\r
+ sb.append(SQCSQ);\r
+ sb.append(value);\r
+ sb.append("');");\r
+ return sb;\r
+ }\r
+ \r
+ public Result<Void> attribRemove(AuthzTrans trans, String ns, String key) {\r
+ try {\r
+ getSession(trans).execute(attribDeleteStmt(new StringBuilder(),ns,key).toString());\r
+ return Result.ok();\r
+ } catch (DriverException | APIException | IOException e) {\r
+ reportPerhapsReset(trans,e);\r
+ return Result.err(Result.ERR_Backend, CassAccess.ERR_ACCESS_MSG);\r
+ }\r
+ }\r
+ \r
+ private StringBuilder attribDeleteStmt(StringBuilder stmt, String ns, String key) {\r
+ stmt.append("DELETE FROM ");\r
+ stmt.append(TABLE_ATTRIB);\r
+ stmt.append(" WHERE ns='");\r
+ stmt.append(ns);\r
+ stmt.append("' AND key='");\r
+ stmt.append(key);\r
+ stmt.append("';");\r
+ return stmt;\r
+ }\r
+ \r
+ private void attribDeleteAllStmt(StringBuilder stmt, Data data) {\r
+ stmt.append(" DELETE FROM ");\r
+ stmt.append(TABLE_ATTRIB);\r
+ stmt.append(" WHERE ns='");\r
+ stmt.append(data.name);\r
+ stmt.append(SQSCCR);\r
+ }\r
+\r
+ private void attribInsertStmts(StringBuilder stmt, Data data) {\r
+ // INSERT new Attrib\r
+ for(Entry<String,String> es : data.attrib(false).entrySet() ) {\r
+ stmt.append(" ");\r
+ attribInsertStmt(stmt,data.name,es.getKey(),es.getValue());\r
+ }\r
+ }\r
+\r
+ /**\r
+ * Add description to Namespace\r
+ * @param trans\r
+ * @param ns\r
+ * @param description\r
+ * @return\r
+ */\r
+ public Result<Void> addDescription(AuthzTrans trans, String ns, String description) {\r
+ try {\r
+ getSession(trans).execute(UPDATE_SP + TABLE + " SET description = '" \r
+ + description + "' WHERE name = '" + ns + "';");\r
+ } catch (DriverException | APIException | IOException e) {\r
+ reportPerhapsReset(trans,e);\r
+ return Result.err(Result.ERR_Backend, CassAccess.ERR_ACCESS_MSG);\r
+ }\r
+\r
+ Data data = new Data();\r
+ data.name=ns;\r
+ wasModified(trans, CRUD.update, data, "Added description " + description + " to namespace " + ns, null );\r
+ return Result.ok();\r
+ }\r
+\r
+ public Result<List<Data>> getChildren(AuthzTrans trans, String parent) {\r
+ return psNS.read(trans, R_TEXT, new Object[]{parent});\r
+ }\r
+ \r
+\r
+ /**\r
+ * Log Modification statements to History\r
+ * \r
+ * @param modified which CRUD action was done\r
+ * @param data entity data that needs a log entry\r
+ * @param overrideMessage if this is specified, we use it rather than crafting a history message based on data\r
+ */\r
+ @Override\r
+ protected void wasModified(AuthzTrans trans, CRUD modified, Data data, String ... override) {\r
+ boolean memo = override.length>0 && override[0]!=null;\r
+ boolean subject = override.length>1 && override[1]!=null;\r
+\r
+ //TODO Must log history\r
+ HistoryDAO.Data hd = HistoryDAO.newInitedData();\r
+ hd.user = trans.user();\r
+ hd.action = modified.name();\r
+ hd.target = TABLE;\r
+ hd.subject = subject ? override[1] : data.name;\r
+ hd.memo = memo ? override[0] : (data.name + " was " + modified.name() + 'd' );\r
+ if(modified==CRUD.delete) {\r
+ try {\r
+ hd.reconstruct = data.bytify();\r
+ } catch (IOException e) {\r
+ trans.error().log(e,"Could not serialize NsDAO.Data");\r
+ }\r
+ }\r
+\r
+ if(historyDAO.create(trans, hd).status!=Status.OK) {\r
+ trans.error().log("Cannot log to History");\r
+ }\r
+ if(infoDAO.touch(trans, TABLE,data.invalidate(cache)).notOK()) {\r
+ trans.error().log("Cannot touch CacheInfo");\r
+ }\r
+ }\r
+\r
+}\r