1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
/*-
* ============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.*;
import java.util.*;
import com.att.eelf.configuration.EELFLogger;
import com.att.eelf.configuration.EELFManager;
import java.util.concurrent.TimeUnit;
import org.onap.dmaap.dbcapi.util.DmaapConfig;
public class ConnectionFactory {
static final EELFLogger logger = EELFManager.getInstance().getLogger( ConnectionFactory.class );
static final EELFLogger appLogger = EELFManager.getInstance().getApplicationLogger();
static final EELFLogger auditLogger = EELFManager.getInstance().getAuditLogger();
static final EELFLogger debugLogger = EELFManager.getInstance().getDebugLogger();
static final EELFLogger errorLogger = EELFManager.getInstance().getErrorLogger();
static final EELFLogger metricsLogger = EELFManager.getInstance().getMetricsLogger();
static final EELFLogger serverLogger = EELFManager.getInstance().getServerLogger();
static final int PREPARE_PSQL_CONNECTION_ATTEMPTS = 5;
static {
try {
Class.forName("org.postgresql.Driver");
} catch (Exception e) {
logger.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 = p.getProperty("DB.user", "dmaap_admin");
dbcr = p.getProperty("DB.cred", "test234-ftl");
schema = p.getProperty("DB.schema", "public");
}
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){
logger.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) {
logger.error("Error", e);
}
}
private void waitFor(long seconds){
try {
TimeUnit.SECONDS.sleep(seconds);
} catch (InterruptedException e) {
logger.debug("Waiting interrupted. ", e);
}
}
}
|
