X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Forg%2Fonap%2Fdmaap%2Fdbcapi%2Fdatabase%2FTableHandler.java;fp=src%2Fmain%2Fjava%2Forg%2Fonap%2Fdmaap%2Fdbcapi%2Fdatabase%2FTableHandler.java;h=b43edc5ef0c0bfd1889ee738fa1843f76b3250b9;hb=0bff051a842b164b680bc938f4a56db435dd5841;hp=0000000000000000000000000000000000000000;hpb=89e6d8f0b87a84b2de6cdab641b28692b9cf3be1;p=dmaap%2Fdbcapi.git

diff --git a/src/main/java/org/onap/dmaap/dbcapi/database/TableHandler.java b/src/main/java/org/onap/dmaap/dbcapi/database/TableHandler.java
new file mode 100644
index 0000000..b43edc5
--- /dev/null
+++ b/src/main/java/org/onap/dmaap/dbcapi/database/TableHandler.java
@@ -0,0 +1,115 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * org.onap.dmaap
+ * ================================================================================
+ * Copyright (C) 2017 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 org.onap.dmaap.dbcapi.database;
+
+import java.util.*;
+import java.lang.reflect.*;
+import java.sql.*;
+
+class TableHandler<C>	{
+	protected ConnectionFactory cf;
+	protected boolean	haskey;
+	protected String	delstmt;
+	protected String	insorreplstmt;
+	protected String	getstmt;
+	protected String	liststmt;
+	protected String	initstmt;
+	protected Class<C>	cls;
+	protected DBFieldHandler[] fields;
+	private static Map<String, Map<String, DBFieldHandler.SqlOp>> exceptions = new HashMap<String, Map<String, DBFieldHandler.SqlOp>>();
+	public static void setSpecialCase(String dbtabname, String dbfldname, DBFieldHandler.SqlOp handler) {
+		Map<String, DBFieldHandler.SqlOp> m = exceptions.get(dbtabname);
+		if (m == null) {
+			m = new HashMap<String, DBFieldHandler.SqlOp>();
+			exceptions.put(dbtabname, m);
+		}
+		m.put(dbfldname, handler);
+	}
+	public static DBFieldHandler.SqlOp getSpecialCase(String dbtabname, String dbfldname) {
+		Map<String, DBFieldHandler.SqlOp> m = exceptions.get(dbtabname);
+		if (m != null) {
+			return(m.get(dbfldname));
+		}
+		return(null);
+	}
+	protected TableHandler(Class<C> cls, String tabname, String keyname) throws Exception {
+		this(ConnectionFactory.getDefaultInstance(), cls, tabname, keyname);
+	}
+	protected TableHandler(ConnectionFactory cf, Class<C> cls, String tabname, String keyname) throws Exception {
+		this.cf = cf;
+		Connection c = null;
+		try {
+			c = cf.get(false);
+			setup(c.getMetaData(), cls, tabname, keyname);
+		} finally {
+			if (c != null) {
+				cf.release(c);
+			}
+		}
+	}
+	private void setup(DatabaseMetaData dmd, Class<C> cls, String tabname, String keyname) throws Exception {
+		this.cls = cls;
+		Vector<DBFieldHandler> h = new Vector<DBFieldHandler>();
+		ResultSet rs = dmd.getColumns("", "public", tabname, null);
+		StringBuffer sb1 = new StringBuffer();
+		StringBuffer sb2 = new StringBuffer();
+		StringBuffer sb3 = new StringBuffer();
+		int	count = 0;
+		while (rs.next()) {
+			if (!rs.getString(3).equals(tabname)) {
+				continue;
+			}
+			String cname = rs.getString(4);
+			if (cname.equals(keyname)) {
+				haskey = true;
+				continue;
+			}
+			sb1.append(", ").append(cname);
+			sb2.append(", ?");
+			sb3.append(", EXCLUDED.").append(cname);
+			count++;
+			h.add(new DBFieldHandler(cls, cname, count, getSpecialCase(tabname, cname)));
+		}
+		if (count == 0) {
+			throw new SQLException("Table " + tabname + " not found");
+		}
+		String clist = sb1.substring(2);
+		String qlist = sb2.substring(2);
+		String elist = sb3.substring(2);
+		if (keyname != null && !haskey) {
+			throw new SQLException("Table " + tabname + " does not have key column " + keyname + " not found");
+		}
+		if (haskey) {
+			count++;
+			h.add(new DBFieldHandler(cls, keyname, count, getSpecialCase(tabname, keyname)));
+			delstmt = "DELETE FROM " + tabname + " WHERE " + keyname + " = ?";
+			insorreplstmt = "INSERT INTO " + tabname + " (" + clist + ", " + keyname + ") VALUES (" + qlist + ", ?) ON CONFLICT(" + keyname + ") DO UPDATE SET (" + clist + ") = (" + elist + ")";
+			getstmt = "SELECT " + clist + ", " + keyname + " FROM " + tabname + " WHERE " + keyname + " = ?";
+			liststmt = "SELECT " + clist + ", " + keyname + " FROM " + tabname;
+		} else {
+			delstmt = "DELETE FROM " + tabname;
+			initstmt = "INSERT INTO " + tabname + " (" + clist + ") VALUES (" + qlist + ")";
+			insorreplstmt = "UPDATE " + tabname + " SET (" + clist + ") = (" + qlist + ")";
+			getstmt = "SELECT " + clist + ", " + keyname + " FROM " + tabname;
+		}
+		fields = h.toArray(new DBFieldHandler[h.size()]);
+	}
+}