AT&T 2.0.19 Code drop, stage 3
[aaf/authz.git] / auth / auth-cass / src / test / java / com / att / dao / aaf / test / NS_ChildUpdate.java
diff --git a/auth/auth-cass/src/test/java/com/att/dao/aaf/test/NS_ChildUpdate.java b/auth/auth-cass/src/test/java/com/att/dao/aaf/test/NS_ChildUpdate.java
new file mode 100644 (file)
index 0000000..8e2f78b
--- /dev/null
@@ -0,0 +1,74 @@
+/**
+ * ============LICENSE_START====================================================
+ * org.onap.aaf
+ * ===========================================================================
+ * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
+ * ===========================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END====================================================
+ *
+ */
+
+package com.att.dao.aaf.test;
+
+import org.onap.aaf.auth.dao.CassAccess;
+import org.onap.aaf.auth.env.AuthzEnv;
+
+import com.datastax.driver.core.Cluster;
+import com.datastax.driver.core.ResultSet;
+import com.datastax.driver.core.Row;
+import com.datastax.driver.core.Session;
+
+public class NS_ChildUpdate {
+
+       public static void main(String[] args) {
+               if(args.length < 3 ) {
+                       System.out.println("usage: NS_ChildUpdate machine mechid (encrypted)passwd");
+               } else {
+                       try {
+                               AuthzEnv env = new AuthzEnv();
+                               env.setLog4JNames("log.properties","authz","authz","audit","init","trace");
+                               
+                               Cluster cluster = Cluster.builder()
+                                               .addContactPoint(args[0])
+                                               .withCredentials(args[1],env.decrypt(args[2], false))
+                                               .build();
+       
+                               Session session = cluster.connect(CassAccess.KEYSPACE);
+                               try {
+                                       ResultSet result = session.execute("SELECT name,parent FROM ns");
+                                       int count = 0;
+                                       for(Row r : result.all()) {
+                                               ++count;
+                                               String name = r.getString(0);
+                                               String parent = r.getString(1);
+                                               if(parent==null) {
+                                                       int idx = name.lastIndexOf('.');
+                                                       
+                                                       parent = idx>0?name.substring(0, idx):".";
+                                                       System.out.println("UPDATE " + name + " to " + parent);
+                                                       session.execute("UPDATE ns SET parent='" + parent + "' WHERE name='" + name + "';");
+                                               }
+                                       }
+                                       System.out.println("Processed " + count + " records");
+                               } finally {
+                                       session.close();
+                                       cluster.close();
+                               }
+                       } catch (Exception e) {
+                               e.printStackTrace();
+                       }
+               }
+       }
+
+}