[DMAAP-BC] Consolidate bus controller repos

[dmaap/buscontroller.git] / dmaap-bc / src / main / java / org / onap / dmaap / dbcapi / database / TableHandler.java

diff --git a/dmaap-bc/src/main/java/org/onap/dmaap/dbcapi/database/TableHandler.java b/dmaap-bc/src/main/java/org/onap/dmaap/dbcapi/database/TableHandler.java
new file mode 100644 (file)
index 0000000..a85bdae
--- /dev/null
+++ b/dmaap-bc/src/main/java/org/onap/dmaap/dbcapi/database/TableHandler.java
@@ -0,0 +1,126 @@
+/*-
+ * ============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.sql.Connection;
+import java.sql.DatabaseMetaData;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Vector;
+import org.onap.dmaap.dbcapi.logging.BaseLoggingClass;
+
+class TableHandler<C> extends BaseLoggingClass {
+       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<>();
+       private String select = "SELECT ";
+       private String from = " FROM ";
+       
+       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);
+                       }
+               }
+       }
+       
+       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<>();
+                       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);
+       }
+       
+       private void setup(DatabaseMetaData dmd, Class<C> cls, String tabname, String keyname) throws Exception {
+               this.cls = cls;
+               Vector<DBFieldHandler> h = new Vector<>();
+               String qualifiedTableName = String.format( "%s.%s", cf.getSchema(), tabname );
+               ResultSet rs = dmd.getColumns("", cf.getSchema(), tabname, null);
+               StringBuilder sb1 = new StringBuilder();
+               StringBuilder sb2 = new StringBuilder();
+               StringBuilder sb3 = new StringBuilder();
+               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 " + qualifiedTableName + " WHERE " + keyname + " = ?";
+                       insorreplstmt = "INSERT INTO " + qualifiedTableName + " (" + clist + ", " + keyname + ") VALUES (" + qlist + ", ?) ON CONFLICT(" + keyname + ") DO UPDATE SET (" + clist + ") = (" + elist + ")";
+                       getstmt = select + clist + ", " + keyname + from + qualifiedTableName + " WHERE " + keyname + " = ?";
+                       liststmt = select + clist + ", " + keyname + from + qualifiedTableName;
+               } else {
+                       delstmt = "DELETE FROM " + qualifiedTableName;
+                       initstmt = "INSERT INTO " + qualifiedTableName + " (" + clist + ") VALUES (" + qlist + ")";
+                       insorreplstmt = "UPDATE " + qualifiedTableName + " SET (" + clist + ") = (" + qlist + ")";
+                       getstmt = select + clist + ", " + keyname + from + qualifiedTableName;
+               }
+               fields = h.toArray(new DBFieldHandler[h.size()]);
+       }
+}