[DMAAP-BC] Consolidate bus controller repos

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

diff --git a/dmaap-bc/src/main/java/org/onap/dmaap/dbcapi/database/ConnectionFactory.java b/dmaap-bc/src/main/java/org/onap/dmaap/dbcapi/database/ConnectionFactory.java
new file mode 100644 (file)
index 0000000..6004d9b
--- /dev/null
+++ b/dmaap-bc/src/main/java/org/onap/dmaap/dbcapi/database/ConnectionFactory.java
@@ -0,0 +1,117 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * org.onap.dmaap
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * Modifications Copyright (C) 2019 IBM.
+ * ================================================================================
+ * 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.DriverManager;
+import java.sql.SQLException;
+import java.util.Properties;
+import java.util.concurrent.TimeUnit;
+import org.onap.dmaap.dbcapi.logging.BaseLoggingClass;
+import org.onap.dmaap.dbcapi.util.DmaapConfig;
+
+public class ConnectionFactory extends BaseLoggingClass {
+
+        static final int PREPARE_PSQL_CONNECTION_ATTEMPTS = 5;
+
+       static {
+               try {
+                       Class.forName("org.postgresql.Driver");
+               } catch (Exception e) {
+                       errorLogger.error("Unable to load postgres driver " + e, e);
+               }
+       }
+       private static ConnectionFactory instance = new ConnectionFactory();
+       private String  host;
+       private String  dbname;
+       private String  dbuser;
+       private String  dbcr;
+       private String  schema;
+       
+       public ConnectionFactory() {
+               Properties p = DmaapConfig.getConfig();
+               host = p.getProperty("DB.host", "dcae-pstg-write-ftl.domain.notset.com");
+               dbname = p.getProperty("DB.name", "dmaap");
+               dbuser = getValue(p, "DB.user", "dmaap_admin");
+               dbcr = getValue(p, "DB.cred", "test234-ftl");
+               schema = p.getProperty("DB.schema", "dmaap_admin");
+       }
+
+       private static String getValue(final Properties props, final String value, final String defaultValue) {
+               String prop = props.getProperty(value, defaultValue);
+               if (prop != null && prop.matches("[$][{].*[}]$")) {
+                       return System.getenv(prop.substring(2, prop.length() - 1));
+               }
+               return prop;
+       }
+
+       public static ConnectionFactory getDefaultInstance() {
+               return(instance);
+       }
+       private Connection[] pool = new Connection[5];
+       private int     cur;
+       public Connection get(boolean fresh) throws SQLException {
+               if (!fresh) {
+                       synchronized(this) {
+                               if (cur > 0) {
+                                       return(pool[--cur]);
+                               }
+                       }
+               }
+               Properties p = new Properties();
+               p.put("user", dbuser);
+               p.put("password", dbcr);
+               for (int i=1; i<PREPARE_PSQL_CONNECTION_ATTEMPTS; i++){
+                       try{
+                               return(DriverManager.getConnection("jdbc:postgresql://" + host + "/" + dbname, p));
+                       }catch(SQLException e){
+                               errorLogger.error("Unable to connect to the postgres server. " + i + "attempt failed. ", e);
+                               waitFor(1);
+                       }
+               }
+               return(DriverManager.getConnection("jdbc:postgresql://" + host + "/" + dbname, p));
+       }
+       public String getSchema() {
+               return(schema);
+       }
+       public void release(Connection c) {
+               synchronized(this) {
+                       if (cur < pool.length) {
+                               pool[cur++] = c;
+                               return;
+                       }
+               }
+               try { 
+                       c.close(); 
+               } catch (Exception e) {
+                       errorLogger.error("Error", e);
+               }
+       }
+       private void waitFor(long seconds){
+               try {
+                       TimeUnit.SECONDS.sleep(seconds);
+               } catch (InterruptedException e) {
+                       debugLogger.debug("Waiting interrupted. ", e);
+                       Thread.currentThread().interrupt();
+               }
+       }
+}