X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=auth%2Fauth-batch%2Fsrc%2Fmain%2Fjava%2Forg%2Fonap%2Faaf%2Fauth%2Fbatch%2Fupdate%2FUpload.java;h=537cd9327e5e3271ea0f6a73bf6954ec5e3e284a;hb=66424b306877435b7e71e119a8d1498b4b263719;hp=a3d37a0ed20e4053748a5972c6bbc496993ce405;hpb=d087d3ddd2829ced1d2ffccc5147c377cbcc92cb;p=aaf%2Fauthz.git

diff --git a/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/update/Upload.java b/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/update/Upload.java
index a3d37a0e..537cd932 100644
--- a/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/update/Upload.java
+++ b/auth/auth-batch/src/main/java/org/onap/aaf/auth/batch/update/Upload.java
@@ -22,73 +22,33 @@
 
 package org.onap.aaf.auth.batch.update;
 
-import java.io.BufferedReader;
 import java.io.File;
-import java.io.FileReader;
 import java.io.IOException;
 import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 import org.onap.aaf.auth.batch.Batch;
 import org.onap.aaf.auth.batch.helpers.CQLBatch;
 import org.onap.aaf.auth.batch.helpers.CQLBatchLoop;
 import org.onap.aaf.auth.env.AuthzTrans;
 import org.onap.aaf.auth.org.OrganizationException;
+import org.onap.aaf.cadi.util.CSV;
 import org.onap.aaf.misc.env.APIException;
 import org.onap.aaf.misc.env.Env;
 import org.onap.aaf.misc.env.LogTarget;
 import org.onap.aaf.misc.env.TimeTaken;
+import org.onap.aaf.misc.env.util.Split;
 
 public class Upload extends Batch {
 
-	private CQLBatchLoop cqlBatch;
+	private static final String DAT = ".dat";
 
-	// APPROVALS
-	private static final String APPR_INS_FMT="  INSERT INTO authz.approval "
-			+ "(id,approver,last_notified,memo,operation,status,ticket,type,user) "
-			+ "VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s);\n";
-	private static final Boolean[] APPR_QUOTES = new Boolean[]{false,true,true,true,true,true,false,true,true};
+	private CQLBatch cqlBatch;
 
-	// ARTIFACTS
-	private static final String ARTI_INS_FMT="  INSERT INTO authz.artifact "
-			+ "(mechid,machine,ca,dir,expires,notify,ns,os_user,renewdays,sans,sponsor,type) "
-			+ "VALUES (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s);\n";
-	private static final Boolean[] ARTI_QUOTES = new Boolean[]
-			{true,true,true,true,true,true,true,true,false,false,true,false};
+	private Map<String,Feed> feeds;
 
-	// CREDS
-	private static final String CRED_INS_FMT="  INSERT INTO authz.cred "
-			+ "(id,type,expires,cred,notes,ns,other,prev) "
-			+ "VALUES (%s,%s,%s,%s,%s,%s,%s,%s);\n";
-	private static final Boolean[] CRED_QUOTES = new Boolean[]
-			{true,false,true,false,true,true,false,false};
-
-	// NS
-	private static final String NS_INS_FMT="  INSERT INTO authz.ns "
-			+ "(name,description,parent,scope,type) "
-			+ "VALUES (%s,%s,%s,%s,%s);\n";
-	private static final Boolean[] NS_QUOTES = new Boolean[]
-			{true,true,true,false,false};
-
-	// x509
-	private static final String X509_INS_FMT="  INSERT INTO authz.x509 "
-			+ "(ca,serial,id,x500,x509) "
-			+ "VALUES (%s,%s,%s,%s,%s);\n";
-	private static final Boolean[] X509_QUOTES = new Boolean[]
-			{true,false,true,true,true};
-
-	// ROLE
-	private static final String ROLE_INS_FMT="  INSERT INTO authz.role "
-			+ "(ns,name,description,perms) "
-			+ "VALUES (%s,%s,%s,%s);\n";
-	private static final Boolean[] ROLE_QUOTES = new Boolean[]
-			{true,true,true,false};
-	// ROLE
-	private static final String PERM_INS_FMT="  INSERT INTO authz.perm "
-			+ "(ns,type,instance,action,description,roles) "
-			+ "VALUES (%s,%s,%s,%s,%s,%s);\n";
-	private static final Boolean[] PERM_QUOTES = new Boolean[]
-			{true,true,true,true,true,false};
 
 	public Upload(AuthzTrans trans) throws APIException, IOException, OrganizationException {
 		super(trans.env());
@@ -103,192 +63,160 @@ public class Upload extends Batch {
 				tt.done();
 			}
 			
-			cqlBatch = new CQLBatchLoop(new CQLBatch(LogTarget.NULL,session),50,dryRun);
+			cqlBatch = new CQLBatch(LogTarget.NULL,session);
+			
+			feeds=new HashMap<>();
+			new Feed(feeds,"ns",1,"name,description,parent,scope=int,type=int",300);
+			new Feed(feeds,"notified",3,"user,target,key,last",300);
+			new Feed(feeds,"approval",1,"id=UUID,approver,last_notified,memo,operation,status,ticket=UUID,type,user",200);
+			new Feed(feeds,"artifact",2,"mechid,machine,ca,dir,expires,notify,ns,os_user,renewdays=int,sans=set,sponsor,type=set",200);
+			new Feed(feeds,"cred",1,"id,type=int,expires,cred=blob,notes,ns,other=int,prev=blob,tag",200);
+			new Feed(feeds,"x509",2,"ca,serial=blob,id,x500,x509=C/R",200);
+			new Feed(feeds,"role",2,"ns,name,description,perms=set",200);
+			new Feed(feeds,"perm",4,"ns,type,instance,action,description,roles=set",200);
+			new Feed(feeds,"history",1,"id=UUID,action,memo,reconstruct=blob,subject,target,user,yr_mon=int",300);
+
 	    } finally {
 	    	tt0.done();
 	    }
 	}
 
+
 	@Override
 	protected void run(AuthzTrans trans) {
-		String line;
-		StringBuilder sb = new StringBuilder();
-		List<String> array = new ArrayList<String>();
-		for(String feed : args()) {
-			File file;
-			if(feed.endsWith(".dat")) {
-				file = new File(feed);
-				feed = file.getName();
-				feed = feed.substring(0,feed.length()-4);
+		List<File> files = new ArrayList<>();
+		if(args().length>0) {
+			File dir = new File(args()[0]);
+			if(dir.isDirectory()) {
+				for(File f : dir.listFiles(pathname -> {
+					return pathname.getName().endsWith(DAT);
+				})) {
+					files.add(f);
+				}
 			} else {
-				file = new File(feed+".dat");
-			}
-			TimeTaken tt = trans.start(file.getAbsolutePath(), Env.SUB);
-			String msg = String.format("#### Running %s.dat Feed ####",feed);
-			trans.info().log(msg);
-			System.out.println(msg);
-	    	BufferedReader br = null;
-		    try {
-				if(file.exists()) {
-					try {
-						br = new BufferedReader(new FileReader(file));
-						try {
-							while((line=br.readLine())!=null) {
-								if(line.length()>5000) {
-									cqlBatch.flush();
-								}
-								cqlBatch.preLoop();
-
-								// Split into fields, first turning Escaped values into something we can convert back from
-								char c=0;
-								boolean inQuote = false;
-								int fldcnt = 0;
-								for(int i=0;i<line.length();++i) {
-									switch(c=line.charAt(i)) {
-										case '"':
-											inQuote = !inQuote;
-											break;
-										case '|':
-											if(inQuote) {
-												sb.append(c);
-											} else {
-												addField(feed,fldcnt++,array,sb);
-											}
-											break;
-										default:
-											sb.append(c);
-									}
-								}
-								addField(feed,fldcnt,array,sb);
-								cqlBatch.inc().append(build(feed, array));
-							}
-							cqlBatch.flush();
-						} catch (Exception t) {
-							trans.error().log(t);
-						} finally {
-							br.close();
-						}
-					} catch (IOException e) {
-						trans.error().log(e);
+				File f;
+				for(String arg : args()) {
+					if(arg.endsWith(DAT)) {
+						f=new File(arg);
+					} else {
+						f=new File(arg+DAT);
 					}
-				} else {
-					trans.error().log("No file found: ", file.getAbsolutePath());
+					files.add(f);
 				}
-			} finally {
-				tt.done();
-				System.err.flush();
-				msg = String.format("\n%d applied in %d batches\n",cqlBatch.total(), cqlBatch.batches());
+			}
+		}
+		for(File file : files) {
+			String f = file.getName();
+			final Feed feed = feeds.get(f.substring(0,f.length()-4));
+			if(feed!=null) {
+				TimeTaken tt = trans.start(file.getAbsolutePath(), Env.SUB);
+				String msg = String.format("#### Running %s.dat Feed ####",feed.getName());
 				trans.info().log(msg);
 				System.out.println(msg);
+				CQLBatchLoop cbl = new CQLBatchLoop(cqlBatch,feed.batchSize,dryRun).showProgress();
+				
+			    try {
+					if(file.exists()) {
+						CSV csv = new CSV(trans.env().access(),file).setDelimiter('|');
+						csv.visit( row -> {
+							feed.insert(cbl.inc(),row);
+						});
+					}
+					cbl.flush();
+				} catch (Throwable e) {
+					e.printStackTrace();
+				} finally {
+					tt.done();
+					System.err.flush();
+					msg = String.format("\n%d applied in %d batches\n",cbl.total(), cbl.batches());
+					trans.info().log(msg);
+					System.out.println(msg);
+				}
 			}
 		}
-
+	}
+	
+	@Override
+	protected void _close(AuthzTrans trans) {
+        session.close();
 	}
 
-
-	private String build(String feed, List<String> array) {
-		String rv;
-		if(array.size()>0) {
-			switch(feed) {
-				case "approval":
-					rv = String.format(APPR_INS_FMT,array.toArray());
-					break;
-				case "artifact":
-					rv = String.format(ARTI_INS_FMT,array.toArray());
-					break;
-				case "cred":
-					rv = String.format(CRED_INS_FMT,array.toArray());
-					break;
-				case "ns":
-					rv = String.format(NS_INS_FMT,array.toArray());
-					break;
-				case "role":
-					rv = String.format(ROLE_INS_FMT,array.toArray());
-					break;
-				case "perm":
-					rv = String.format(PERM_INS_FMT,array.toArray());
-					break;
-				case "x509":
-					rv = String.format(X509_INS_FMT,array.toArray());
-					break;
-				default:
-					rv = "";
+	private class Feed {
+		private final String name;
+		private final String[] flds;
+		private final String[] types;
+		private final int key;
+		private final int batchSize;
+		public Feed(Map<String, Feed> feeds, String feed, int keyLength, String fields,int batchSize) {
+			name=feed;
+			key = keyLength;
+			flds = Split.splitTrim(',', fields);
+			types = new String[flds.length];
+			this.batchSize = batchSize;
+			int equals;
+			for(int i=0;i<flds.length;++i) {
+				if((equals = flds[i].indexOf('='))>0) {
+					types[i]=flds[i].substring(equals+1);
+					flds[i]=flds[i].substring(0, equals);
+				}
 			}
-			array.clear();
-		} else {
-			rv = ""; 
+			feeds.put(feed,this);
 		}
-		return rv;
-	}
-	
-	private void addField(String feed, int fldcnt, List<String> array, StringBuilder sb) {
-		Boolean[] ba;
-		switch(feed) {
-			case "approval":
-				ba = APPR_QUOTES;
-				break;
-			case "artifact":
-				ba = ARTI_QUOTES;
-				break;
-			case "cred":
-				ba = CRED_QUOTES;
-				break;
-			case "ns":
-				ba = NS_QUOTES;
-				break;
-			case "role":
-				ba = ROLE_QUOTES;
-				break;
-			case "perm":
-				ba = PERM_QUOTES;
-				break;
-			case "x509":
-				ba = X509_QUOTES;
-				break;
-			default:
-				ba = null;
+		
+		public String getName() {
+			return name;
 		}
-		if(ba!=null) {
-			if(sb.toString().length()==0) {
-				array.add("null");
-			} else {
-				if(ba[fldcnt]) {
-					String s = null;
-					if(sb.indexOf("'")>=0) {
-						s = sb.toString().replace("'","''");
-					}
-					if(sb.indexOf("\\n")>=0) {
-						if(s==null) {
-							s = sb.toString().replace("\\n","\n");
-						} else {
-							s = s.replace("\\n","\n");
-						}
+
+		public void insert(StringBuilder sb,List<String> row) {
+			sb.append("INSERT INTO authz.");
+			sb.append(name);
+			sb.append(" (");
+			boolean first = true;
+			StringBuilder values = new StringBuilder(") VALUES (");
+			String value;
+			String type;
+			for(int idx=0;idx<row.size();++idx) {
+				value = row.get(idx).trim();
+				if(idx<key || !(value.isEmpty() || "null".equals(value))) {
+					if(first) {
+						first = false;
+					} else {
+						sb.append(',');
+						values.append(',');
 					}
-					if(sb.indexOf("\\t")>=0) {
-						if(s==null) {
-							s = sb.toString().replace("\\t","\t");
-						} else {
-							s = s.replace("\\t","\t");
+					sb.append(flds[idx]);
+					type=types[idx];
+					if(type==null) { // String is default.
+						switch(value) {
+						  case "":
+							if(idx<key) {
+								// Key value has to be something, but can't be actual null
+								values.append("''");
+							} else {
+								values.append("null");
+							}
+							break;
+						  default:
+							values.append('\'');
+							values.append(value.replaceAll("'","''"));							
+							values.append('\'');
 						}
+					} else switch(type) {
+							case "C/R":
+								values.append('\'');
+								values.append(value.replaceAll("\\\\n", "\n"));
+								values.append('\'');
+								break;
+							default:
+								values.append(value);
+								break;
+								
 					}
-					if(s==null) {
-						array.add("'" + sb + '\'');
-					} else {
-						array.add("'" + s + '\'');
-					}
-				} else {
-					array.add(sb.toString());
 				}
 			}
-			sb.setLength(0);
+			sb.append(values);
+			sb.append(");\n");
 		}
 	}
-
-
-	@Override
-	protected void _close(AuthzTrans trans) {
-        session.close();
-	}
-
 }