--- /dev/null
+/*******************************************************************************\r
+ * ============LICENSE_START==================================================\r
+ * * org.onap.dmaap\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
+\r
+\r
+package org.onap.dmaap.datarouter.provisioning.utils;\r
+\r
+import java.io.File;\r
+import java.io.FileReader;\r
+import java.io.IOException;\r
+import java.io.InputStream;\r
+import java.io.LineNumberReader;\r
+import java.lang.reflect.Constructor;\r
+import java.lang.reflect.InvocationTargetException;\r
+import java.sql.Connection;\r
+import java.sql.DatabaseMetaData;\r
+import java.sql.DriverManager;\r
+import java.sql.PreparedStatement;\r
+import java.sql.ResultSet;\r
+import java.sql.SQLException;\r
+import java.sql.Statement;\r
+import java.util.HashSet;\r
+import java.util.LinkedList;\r
+import java.util.NoSuchElementException;\r
+import java.util.Properties;\r
+import java.util.Queue;\r
+import java.util.Set;\r
+\r
+import org.apache.log4j.Logger;\r
+import org.onap.dmaap.datarouter.provisioning.beans.DeliveryRecord;\r
+import org.onap.dmaap.datarouter.provisioning.beans.ExpiryRecord;\r
+import org.onap.dmaap.datarouter.provisioning.beans.Loadable;\r
+import org.onap.dmaap.datarouter.provisioning.beans.PublishRecord;\r
+\r
+/**\r
+ * Load the DB JDBC driver, and manage a simple pool of connections to the DB.\r
+ *\r
+ * @author Robert Eby\r
+ * @version $Id$\r
+ */\r
+public class DB {\r
+ /** The name of the properties file (in CLASSPATH) */\r
+ public static final String CONFIG_FILE = "provserver.properties";\r
+\r
+ private static String DB_DRIVER = "com.mysql.jdbc.Driver";\r
+ private static String DB_URL = "jdbc:mysql://127.0.0.1:3306/datarouter";\r
+ private static String DB_LOGIN = "datarouter";\r
+ private static String DB_PASSWORD = "datarouter";\r
+ private static Properties props;\r
+ private static Logger intlogger = Logger.getLogger("org.onap.dmaap.datarouter.provisioning.internal");\r
+ private static Queue<Connection> queue = new LinkedList<Connection>();\r
+\r
+ public static String HTTPS_PORT;\r
+ public static String HTTP_PORT;\r
+\r
+ /**\r
+ * Construct a DB object. If this is the very first creation of this object, it will load a copy\r
+ * of the properties for the server, and attempt to load the JDBC driver for the database. If a fatal\r
+ * error occurs (e.g. either the properties file or the DB driver is missing), the JVM will exit.\r
+ */\r
+ public DB() {\r
+ if (props == null) {\r
+ props = new Properties();\r
+ InputStream inStream = getClass().getClassLoader().getResourceAsStream(CONFIG_FILE);\r
+ try {\r
+ props.load(inStream);\r
+ DB_DRIVER = (String) props.get("com.att.research.datarouter.db.driver");\r
+ DB_URL = (String) props.get("com.att.research.datarouter.db.url");\r
+ DB_LOGIN = (String) props.get("com.att.research.datarouter.db.login");\r
+ DB_PASSWORD = (String) props.get("com.att.research.datarouter.db.password");\r
+ HTTPS_PORT = (String) props.get("com.att.research.datarouter.provserver.https.port");\r
+ HTTP_PORT = (String) props.get("com.att.research.datarouter.provserver.http.port");\r
+ Class.forName(DB_DRIVER);\r
+ } catch (IOException e) {\r
+ intlogger.fatal("PROV9003 Opening properties: "+e.getMessage());\r
+ e.printStackTrace();\r
+ System.exit(1);\r
+ } catch (ClassNotFoundException e) {\r
+ intlogger.fatal("PROV9004 cannot find the DB driver: "+e);\r
+ e.printStackTrace();\r
+ System.exit(1);\r
+ } finally {\r
+ try {\r
+ inStream.close();\r
+ } catch (IOException e) {\r
+ }\r
+ }\r
+ }\r
+ }\r
+ /**\r
+ * Get the provisioning server properties (loaded from provserver.properties).\r
+ * @return the Properties object\r
+ */\r
+ public Properties getProperties() {\r
+ return props;\r
+ }\r
+ /**\r
+ * Get a JDBC connection to the DB from the pool. Creates a new one if none are available.\r
+ * @return the Connection\r
+ * @throws SQLException\r
+ */\r
+ @SuppressWarnings("resource")\r
+ public Connection getConnection() throws SQLException {\r
+ Connection c = null;\r
+ while (c == null) {\r
+ synchronized (queue) {\r
+ try {\r
+ c = queue.remove();\r
+ } catch (NoSuchElementException e) {\r
+ int n = 0;\r
+ do {\r
+ // Try up to 3 times to get a connection\r
+ try {\r
+ c = DriverManager.getConnection(DB_URL, DB_LOGIN, DB_PASSWORD);\r
+ } catch (SQLException e1) {\r
+ if (++n >= 3)\r
+ throw e1;\r
+ }\r
+ } while (c == null);\r
+ }\r
+ }\r
+ if (c != null && !c.isValid(1)) {\r
+ c.close();\r
+ c = null;\r
+ }\r
+ }\r
+ return c;\r
+ }\r
+ /**\r
+ * Returns a JDBC connection to the pool.\r
+ * @param c the Connection to return\r
+ * @throws SQLException\r
+ */\r
+ public void release(Connection c) {\r
+ if (c != null) {\r
+ synchronized (queue) {\r
+ if (!queue.contains(c))\r
+ queue.add(c);\r
+ }\r
+ }\r
+ }\r
+\r
+ /**\r
+ * Run all necessary retrofits required to bring the database up to the level required for this version\r
+ * of the provisioning server. This should be run before the server itself is started.\r
+ * @return true if all retrofits worked, false otherwise\r
+ */\r
+ public boolean runRetroFits() {\r
+ return retroFit1()\r
+ && retroFit2()\r
+ && retroFit3()\r
+ && retroFit4()\r
+ && retroFit5()\r
+ && retroFit6()\r
+ && retroFit7()\r
+ && retroFit8()\r
+ && retroFit9() //New retroFit call to add CREATED_DATE column Rally:US674199 - 1610\r
+ && retroFit10() //New retroFit call to add BUSINESS_DESCRIPTION column Rally:US708102 - 1610\r
+ && retroFit11() //New retroFit call for groups feature Rally:US708115 - 1610 \r
+ ;\r
+ }\r
+ /**\r
+ * Retrofit 1 - Make sure the expected tables are in MySQL and are initialized.\r
+ * Uses mysql_init_0000 and mysql_init_0001 to setup the DB.\r
+ * @return true if the retrofit worked, false otherwise\r
+ */\r
+ private boolean retroFit1() {\r
+ final String[] expected_tables = {\r
+ "FEEDS", "FEED_ENDPOINT_ADDRS", "FEED_ENDPOINT_IDS", "PARAMETERS", "SUBSCRIPTIONS"\r
+ };\r
+ Connection c = null;\r
+ try {\r
+ c = getConnection();\r
+ Set<String> tables = getTableSet(c);\r
+ boolean initialize = false;\r
+ for (String s : expected_tables) {\r
+ initialize |= !tables.contains(s);\r
+ }\r
+ if (initialize) {\r
+ intlogger.info("PROV9001: First time startup; The database is being initialized.");\r
+ runInitScript(c, 0); // script 0 creates the provisioning tables\r
+ runInitScript(c, 1); // script 1 initializes PARAMETERS\r
+ }\r
+ } catch (SQLException e) {\r
+ intlogger.fatal("PROV9000: The database credentials are not working: "+e.getMessage());\r
+ return false;\r
+ } finally {\r
+ if (c != null)\r
+ release(c);\r
+ }\r
+ return true;\r
+ }\r
+ /**\r
+ * Retrofit 2 - if the LOG_RECORDS table is missing, add it.\r
+ * Uses mysql_init_0002 to create this table.\r
+ * @return true if the retrofit worked, false otherwise\r
+ */\r
+ private boolean retroFit2() {\r
+ Connection c = null;\r
+ try {\r
+ // If LOG_RECORDS table is missing, add it\r
+ c = getConnection();\r
+ Set<String> tables = getTableSet(c);\r
+ if (!tables.contains("LOG_RECORDS")) {\r
+ intlogger.info("PROV9002: Creating LOG_RECORDS table.");\r
+ runInitScript(c, 2); // script 2 creates the LOG_RECORDS table\r
+ }\r
+ } catch (SQLException e) {\r
+ intlogger.fatal("PROV9000: The database credentials are not working: "+e.getMessage());\r
+ return false;\r
+ } finally {\r
+ if (c != null)\r
+ release(c);\r
+ }\r
+ return true;\r
+ }\r
+ /**\r
+ * Retrofit 3 - if the FEEDS_UNIQUEID table (from release 1.0.*) exists, drop it.\r
+ * If SUBSCRIPTIONS.SUBID still has the auto_increment attribute, remove it.\r
+ * @return true if the retrofit worked, false otherwise\r
+ */\r
+ @SuppressWarnings("resource")\r
+ private boolean retroFit3() {\r
+ Connection c = null;\r
+ try {\r
+ // if SUBSCRIPTIONS.SUBID still has auto_increment, remove it\r
+ boolean doremove = false;\r
+ c = getConnection();\r
+ DatabaseMetaData md = c.getMetaData();\r
+ ResultSet rs = md.getColumns("datarouter", "", "SUBSCRIPTIONS", "SUBID");\r
+ if (rs != null) {\r
+ while (rs.next()) {\r
+ doremove = rs.getString("IS_AUTOINCREMENT").equals("YES");\r
+ }\r
+ rs.close();\r
+ rs = null;\r
+ }\r
+ if (doremove) {\r
+ intlogger.info("PROV9002: Modifying SUBSCRIPTIONS SUBID column to remove auto increment.");\r
+ Statement s = c.createStatement();\r
+ s.execute("ALTER TABLE SUBSCRIPTIONS MODIFY COLUMN SUBID INT UNSIGNED NOT NULL");\r
+ s.close();\r
+ }\r
+\r
+ // Remove the FEEDS_UNIQUEID table, if it exists\r
+ Set<String> tables = getTableSet(c);\r
+ if (tables.contains("FEEDS_UNIQUEID")) {\r
+ intlogger.info("PROV9002: Dropping FEEDS_UNIQUEID table.");\r
+ Statement s = c.createStatement();\r
+ s.execute("DROP TABLE FEEDS_UNIQUEID");\r
+ s.close();\r
+ }\r
+ } catch (SQLException e) {\r
+ intlogger.fatal("PROV9000: The database credentials are not working: "+e.getMessage());\r
+ return false;\r
+ } finally {\r
+ if (c != null)\r
+ release(c);\r
+ }\r
+ return true;\r
+ }\r
+ private long nextid = 0; // used for initial creation of LOG_RECORDS table.\r
+ /**\r
+ * Retrofit 4 - if old log tables exist (from release 1.0.*), copy them to LOG_RECORDS, then drop them.\r
+ * @return true if the retrofit worked, false otherwise\r
+ */\r
+ @SuppressWarnings("resource")\r
+ private boolean retroFit4() {\r
+ Connection c = null;\r
+ try {\r
+ c = getConnection();\r
+ Set<String> tables = getTableSet(c);\r
+ if (tables.contains("PUBLISH_RECORDS")) {\r
+ intlogger.info("PROV9002: Copying PUBLISH_RECORDS to LOG_RECORDS table.");\r
+ copyLogTable("PUBLISH_RECORDS", PublishRecord.class);\r
+ intlogger.info("PROV9002: Dropping PUBLISH_RECORDS table.");\r
+ Statement s = c.createStatement();\r
+ s.execute("DROP TABLE PUBLISH_RECORDS");\r
+ s.close();\r
+ }\r
+ if (tables.contains("DELIVERY_RECORDS")) {\r
+ intlogger.info("PROV9002: Copying DELIVERY_RECORDS to LOG_RECORDS table.");\r
+ copyLogTable("DELIVERY_RECORDS", DeliveryRecord.class);\r
+ intlogger.info("PROV9002: Dropping DELIVERY_RECORDS table.");\r
+ Statement s = c.createStatement();\r
+ s.execute("DROP TABLE DELIVERY_RECORDS");\r
+ s.close();\r
+ }\r
+ if (tables.contains("EXPIRY_RECORDS")) {\r
+ intlogger.info("PROV9002: Copying EXPIRY_RECORDS to LOG_RECORDS table.");\r
+ copyLogTable("EXPIRY_RECORDS", ExpiryRecord.class);\r
+ intlogger.info("PROV9002: Dropping EXPIRY_RECORDS table.");\r
+ Statement s = c.createStatement();\r
+ s.execute("DROP TABLE EXPIRY_RECORDS");\r
+ s.close();\r
+ }\r
+ } catch (SQLException e) {\r
+ intlogger.fatal("PROV9000: The database credentials are not working: "+e.getMessage());\r
+ return false;\r
+ } finally {\r
+ if (c != null)\r
+ release(c);\r
+ }\r
+ return true;\r
+ }\r
+ /**\r
+ * Retrofit 5 - Create the new routing tables required for Release 2.\r
+ * Adds a new "SUSPENDED" column to FEEDS and SUBSCRIPTIONS.\r
+ * Modifies the LOG_RECORDS table to handle new R2 records.\r
+ * @return true if the retrofit worked, false otherwise\r
+ */\r
+ @SuppressWarnings("resource")\r
+ private boolean retroFit5() {\r
+ final String[] expected_tables = {\r
+ "INGRESS_ROUTES", "EGRESS_ROUTES", "NETWORK_ROUTES", "NODESETS", "NODES"\r
+ };\r
+ Connection c = null;\r
+ try {\r
+ // If expected tables are not present, then add new routing tables\r
+ c = getConnection();\r
+ Set<String> tables = getTableSet(c);\r
+ boolean initialize = false;\r
+ for (String s : expected_tables) {\r
+ initialize |= !tables.contains(s);\r
+ }\r
+ if (initialize) {\r
+ intlogger.info("PROV9002: Adding routing tables for Release 2.0.");\r
+ runInitScript(c, 3); // script 3 creates the routing tables\r
+ }\r
+\r
+ // Add SUSPENDED column to FEEDS/SUBSCRIPTIONS\r
+ DatabaseMetaData md = c.getMetaData();\r
+ for (String tbl : new String[] {"FEEDS", "SUBSCRIPTIONS" }) {\r
+ boolean add_col = true;\r
+ ResultSet rs = md.getColumns("datarouter", "", tbl, "SUSPENDED");\r
+ if (rs != null) {\r
+ add_col = !rs.next();\r
+ rs.close();\r
+ rs = null;\r
+ }\r
+ if (add_col) {\r
+ intlogger.info("PROV9002: Adding SUSPENDED column to "+tbl+" table.");\r
+ Statement s = c.createStatement();\r
+ s.execute("ALTER TABLE "+tbl+" ADD COLUMN SUSPENDED BOOLEAN DEFAULT FALSE");\r
+ s.close();\r
+ }\r
+ }\r
+\r
+ // Modify LOG_RECORDS for R2\r
+ intlogger.info("PROV9002: Modifying LOG_RECORDS table.");\r
+ Statement s = c.createStatement();\r
+ s.execute("ALTER TABLE LOG_RECORDS MODIFY COLUMN TYPE ENUM('pub', 'del', 'exp', 'pbf', 'dlx') NOT NULL");\r
+ s.close();\r
+ s = c.createStatement();\r
+ s.execute("ALTER TABLE LOG_RECORDS MODIFY COLUMN REASON ENUM('notRetryable', 'retriesExhausted', 'diskFull', 'other')");\r
+ s.close();\r
+ boolean add_col = true;\r
+ ResultSet rs = md.getColumns("datarouter", "", "LOG_RECORDS", "CONTENT_LENGTH_2");\r
+ if (rs != null) {\r
+ add_col = !rs.next();\r
+ rs.close();\r
+ rs = null;\r
+ }\r
+ if (add_col) {\r
+ intlogger.info("PROV9002: Fixing two columns in LOG_RECORDS table (this may take some time).");\r
+ s = c.createStatement();\r
+ s.execute("ALTER TABLE LOG_RECORDS MODIFY COLUMN CONTENT_LENGTH BIGINT NOT NULL, ADD COLUMN CONTENT_LENGTH_2 BIGINT AFTER RECORD_ID");\r
+ s.close();\r
+ }\r
+ } catch (SQLException e) {\r
+ intlogger.fatal("PROV9000: The database credentials are not working: "+e.getMessage());\r
+ return false;\r
+ } finally {\r
+ if (c != null)\r
+ release(c);\r
+ }\r
+ return true;\r
+ }\r
+ /**\r
+ * Retrofit 6 - Adjust LOG_RECORDS.USER to be 50 chars (MR #74).\r
+ * @return true if the retrofit worked, false otherwise\r
+ */\r
+ @SuppressWarnings("resource")\r
+ private boolean retroFit6() {\r
+ Connection c = null;\r
+ try {\r
+ c = getConnection();\r
+ // Modify LOG_RECORDS for R2\r
+ intlogger.info("PROV9002: Modifying LOG_RECORDS.USER length.");\r
+ Statement s = c.createStatement();\r
+ s.execute("ALTER TABLE LOG_RECORDS MODIFY COLUMN USER VARCHAR(50)");\r
+ s.close();\r
+ } catch (SQLException e) {\r
+ intlogger.fatal("PROV9000: The database credentials are not working: "+e.getMessage());\r
+ return false;\r
+ } finally {\r
+ if (c != null)\r
+ release(c);\r
+ }\r
+ return true;\r
+ }\r
+ /**\r
+ * Retrofit 7 - Adjust LOG_RECORDS.FEED_FILEID and LOG_RECORDS.DELIVERY_FILEID to be 256 chars.\r
+ * @return true if the retrofit worked, false otherwise\r
+ */\r
+ @SuppressWarnings("resource")\r
+ private boolean retroFit7() {\r
+ Connection c = null;\r
+ try {\r
+ c = getConnection();\r
+ // Modify LOG_RECORDS for long (>128) FILEIDs\r
+ intlogger.info("PROV9002: Modifying LOG_RECORDS.USER length.");\r
+ Statement s = c.createStatement();\r
+ s.execute("ALTER TABLE LOG_RECORDS MODIFY COLUMN FEED_FILEID VARCHAR(256), MODIFY COLUMN DELIVERY_FILEID VARCHAR(256)");\r
+ s.close();\r
+ } catch (SQLException e) {\r
+ intlogger.fatal("PROV9000: The database credentials are not working: "+e.getMessage());\r
+ return false;\r
+ } finally {\r
+ if (c != null)\r
+ release(c);\r
+ }\r
+ return true;\r
+ }\r
+ /**\r
+ * Retrofit 8 - Adjust FEEDS.NAME to be 255 chars (MR #74).\r
+ * @return true if the retrofit worked, false otherwise\r
+ */\r
+ @SuppressWarnings("resource")\r
+ private boolean retroFit8() {\r
+ Connection c = null;\r
+ try {\r
+ c = getConnection();\r
+ intlogger.info("PROV9002: Modifying FEEDS.NAME length.");\r
+ Statement s = c.createStatement();\r
+ s.execute("ALTER TABLE FEEDS MODIFY COLUMN NAME VARCHAR(255)");\r
+ s.close();\r
+ } catch (SQLException e) {\r
+ intlogger.fatal("PROV9000: The database credentials are not working: "+e.getMessage());\r
+ return false;\r
+ } finally {\r
+ if (c != null)\r
+ release(c);\r
+ }\r
+ return true;\r
+ }\r
+ \r
+ /**\r
+ * Retrofit 9 - Add column FEEDS.CREATED_DATE and SUBSCRIPTIONS.CREATED_DATE, 1610 release user story US674199.\r
+ * @return true if the retrofit worked, false otherwise\r
+ */\r
+\r
+ @SuppressWarnings("resource") \r
+ private boolean retroFit9() { \r
+ Connection c = null; \r
+ try { \r
+ c = getConnection(); \r
+ // Add CREATED_DATE column to FEEDS/SUBSCRIPTIONS tables\r
+ DatabaseMetaData md = c.getMetaData(); \r
+ for (String tbl : new String[] {"FEEDS", "SUBSCRIPTIONS" }) { \r
+ boolean add_col = true; \r
+ ResultSet rs = md.getColumns("datarouter", "", tbl, "CREATED_DATE"); \r
+ if (rs != null) { \r
+ add_col = !rs.next(); \r
+ rs.close(); \r
+ rs = null; \r
+ } \r
+ if (add_col) { \r
+ intlogger.info("PROV9002: Adding CREATED_DATE column to "+tbl+" table."); \r
+ Statement s = c.createStatement();\r
+ s.execute("ALTER TABLE "+tbl+" ADD COLUMN CREATED_DATE timestamp DEFAULT CURRENT_TIMESTAMP"); \r
+ s.close(); \r
+ } \r
+ } \r
+ } catch (SQLException e) { \r
+ intlogger.fatal("PROV9000: The database credentials are not working: "+e.getMessage()); \r
+ return false; \r
+ } finally { \r
+ if (c != null) \r
+ release(c); \r
+ } \r
+ return true; \r
+ }\r
+\r
+ /**\r
+ * Retrofit 10 -Adding business BUSINESS_DESCRIPTION to FEEDS table (Rally\r
+ * US708102).\r
+ * \r
+ * @return true if the retrofit worked, false otherwise\r
+ */\r
+\r
+ @SuppressWarnings("resource")\r
+ private boolean retroFit10() {\r
+ Connection c = null;\r
+ boolean addColumn = true;\r
+ \r
+ try {\r
+\r
+ c = getConnection(); \r
+ // Add BUSINESS_DESCRIPTION column to FEEDS table\r
+ DatabaseMetaData md = c.getMetaData(); \r
+ boolean add_col = true; \r
+ ResultSet rs = md.getColumns("datarouter", "", "FEEDS", "BUSINESS_DESCRIPTION"); \r
+ if (rs != null) { \r
+ add_col = !rs.next(); \r
+ rs.close(); \r
+ rs = null; \r
+ } \r
+ if(add_col) {\r
+ intlogger\r
+ .info("PROV9002: Adding BUSINESS_DESCRIPTION column to FEEDS table.");\r
+ Statement s = c.createStatement();\r
+ s.execute("ALTER TABLE FEEDS ADD COLUMN BUSINESS_DESCRIPTION varchar(1000) DEFAULT NULL AFTER DESCRIPTION, MODIFY COLUMN DESCRIPTION VARCHAR(1000)");\r
+ s.close();\r
+ }\r
+ }\r
+ catch (SQLException e) {\r
+ intlogger\r
+ .fatal("PROV9000: The database credentials are not working: "\r
+ + e.getMessage());\r
+ return false;\r
+ } finally {\r
+ if (c != null)\r
+ release(c);\r
+ }\r
+ return true;\r
+ }\r
+\r
+\r
+ /*New retroFit method is added for groups feature Rally:US708115 - 1610 \r
+ * @retroFit11()\r
+ * @parmas: none\r
+ * @return - boolean if table and fields are created (Group table, group id in FEEDS, SUBSCRIPTION TABLES)\r
+ */\r
+ @SuppressWarnings("resource") \r
+ private boolean retroFit11() { \r
+ final String[] expected_tables = { \r
+ "GROUPS" \r
+ }; \r
+ Connection c = null; \r
+ \r
+ try { \r
+ // If expected tables are not present, then add new routing tables \r
+ c = getConnection(); \r
+ Set<String> tables = getTableSet(c); \r
+ boolean initialize = false; \r
+ for (String s : expected_tables) { \r
+ initialize |= !tables.contains(s); \r
+ } \r
+ if (initialize) { \r
+ intlogger.info("PROV9002: Adding GROUPS table for Release 1610."); \r
+ runInitScript(c, 4); // script 4 creates the routing tables \r
+ } \r
+ \r
+ // Add GROUPID column to FEEDS/SUBSCRIPTIONS \r
+ DatabaseMetaData md = c.getMetaData(); \r
+ for (String tbl : new String[] {"FEEDS", "SUBSCRIPTIONS" }) { \r
+ boolean add_col = true; \r
+ ResultSet rs = md.getColumns("datarouter", "", tbl, "GROUPID"); \r
+ if (rs != null) { \r
+ add_col = !rs.next(); \r
+ rs.close(); \r
+ rs = null; \r
+ } \r
+ if (add_col) { \r
+ intlogger.info("PROV9002: Adding GROUPID column to "+tbl+" table."); \r
+ Statement s = c.createStatement(); \r
+ s.execute("ALTER TABLE "+tbl+" ADD COLUMN GROUPID INT(10) UNSIGNED NOT NULL DEFAULT 0 AFTER FEEDID"); \r
+ s.close(); \r
+ } \r
+ } \r
+ } catch (SQLException e) { \r
+ intlogger.fatal("PROV9000: The database credentials are not working: "+e.getMessage()); \r
+ return false; \r
+ } finally { \r
+ if (c != null) \r
+ release(c); \r
+ } \r
+ return true; \r
+ }\r
+\r
+\r
+ /**\r
+ * Copy the log table <i>table_name</i> to LOG_RECORDS;\r
+ * @param table_name the name of the old (1.0.*) table to copy\r
+ * @param table_class the class used to instantiate a record from the table\r
+ * @throws SQLException if there is a problem getting a MySQL connection\r
+ */\r
+ @SuppressWarnings("resource")\r
+ private void copyLogTable(String table_name, Class<? extends Loadable> table_class) throws SQLException {\r
+ long start = System.currentTimeMillis();\r
+ int n = 0;\r
+ Connection c1 = getConnection();\r
+ Connection c2 = getConnection();\r
+\r
+ try {\r
+ Constructor<? extends Loadable> cnst = table_class.getConstructor(ResultSet.class);\r
+ PreparedStatement ps = c2.prepareStatement(LogfileLoader.INSERT_SQL);\r
+ Statement stmt = c1.createStatement();\r
+ ResultSet rs = stmt.executeQuery("select * from "+table_name);\r
+ while (rs.next()) {\r
+ Loadable rec = cnst.newInstance(rs);\r
+ rec.load(ps);\r
+ ps.setLong(18, ++nextid);\r
+ ps.executeUpdate();\r
+ if ((++n % 10000) == 0)\r
+ intlogger.debug(" "+n+" records done.");\r
+ }\r
+ stmt.close();\r
+ ps.close();\r
+ } catch (SQLException e) {\r
+ e.printStackTrace();\r
+ } catch (NoSuchMethodException e) {\r
+ e.printStackTrace();\r
+ } catch (SecurityException e) {\r
+ e.printStackTrace();\r
+ } catch (InstantiationException e) {\r
+ e.printStackTrace();\r
+ } catch (IllegalAccessException e) {\r
+ e.printStackTrace();\r
+ } catch (IllegalArgumentException e) {\r
+ e.printStackTrace();\r
+ } catch (InvocationTargetException e) {\r
+ e.printStackTrace();\r
+ }\r
+\r
+ release(c1);\r
+ release(c2);\r
+ long x = (System.currentTimeMillis() - start);\r
+ intlogger.debug(" "+n+" records done in "+x+" ms.");\r
+ }\r
+\r
+ /**\r
+ * Get a set of all table names in the DB.\r
+ * @param c a DB connection\r
+ * @return the set of table names\r
+ */\r
+ private Set<String> getTableSet(Connection c) {\r
+ Set<String> tables = new HashSet<String>();\r
+ try {\r
+ DatabaseMetaData md = c.getMetaData();\r
+ ResultSet rs = md.getTables("datarouter", "", "", null);\r
+ if (rs != null) {\r
+ while (rs.next()) {\r
+ tables.add(rs.getString("TABLE_NAME"));\r
+ }\r
+ rs.close();\r
+ }\r
+ } catch (SQLException e) {\r
+ }\r
+ return tables;\r
+ }\r
+ /**\r
+ * Initialize the tables by running the initialization scripts located in the directory specified\r
+ * by the property <i>com.att.research.datarouter.provserver.dbscripts</i>. Scripts have names of\r
+ * the form mysql_init_NNNN.\r
+ * @param c a DB connection\r
+ * @param n the number of the mysql_init_NNNN script to run\r
+ */\r
+ private void runInitScript(Connection c, int n) {\r
+ String scriptdir = (String) props.get("com.att.research.datarouter.provserver.dbscripts");\r
+ StringBuilder sb = new StringBuilder();\r
+ try {\r
+ String scriptfile = String.format("%s/mysql_init_%04d", scriptdir, n);\r
+ if (!(new File(scriptfile)).exists())\r
+ return;\r
+\r
+ LineNumberReader in = new LineNumberReader(new FileReader(scriptfile));\r
+ String line;\r
+ while ((line = in.readLine()) != null) {\r
+ if (!line.startsWith("--")) {\r
+ line = line.trim();\r
+ sb.append(line);\r
+ if (line.endsWith(";")) {\r
+ // Execute one DDL statement\r
+ String sql = sb.toString();\r
+ sb.setLength(0);\r
+ Statement s = c.createStatement();\r
+ s.execute(sql);\r
+ s.close();\r
+ }\r
+ }\r
+ }\r
+ in.close();\r
+ sb.setLength(0);\r
+ } catch (Exception e) {\r
+ intlogger.fatal("PROV9002 Error when initializing table: "+e.getMessage());\r
+ System.exit(1);\r
+ }\r
+ }\r
+}\r