1 files changed, 117 insertions, 0 deletions
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
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();
+		}
+	}
+}