Newer
Older
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
/*
************************************************************************
******************* CANADIAN ASTRONOMY DATA CENTRE *******************
************** CENTRE CANADIEN DE DONNÉES ASTRONOMIQUES **************
*
* (c) 2015. (c) 2015.
* Government of Canada Gouvernement du Canada
* National Research Council Conseil national de recherches
* Ottawa, Canada, K1A 0R6 Ottawa, Canada, K1A 0R6
* All rights reserved Tous droits réservés
*
* NRC disclaims any warranties, Le CNRC dénie toute garantie
* expressed, implied, or énoncée, implicite ou légale,
* statutory, of any kind with de quelque nature que ce
* respect to the software, soit, concernant le logiciel,
* including without limitation y compris sans restriction
* any warranty of merchantability toute garantie de valeur
* or fitness for a particular marchande ou de pertinence
* purpose. NRC shall not be pour un usage particulier.
* liable in any event for any Le CNRC ne pourra en aucun cas
* damages, whether direct or être tenu responsable de tout
* indirect, special or general, dommage, direct ou indirect,
* consequential or incidental, particulier ou général,
* arising from the use of the accessoire ou fortuit, résultant
* software. Neither the name de l'utilisation du logiciel. Ni
* of the National Research le nom du Conseil National de
* Council of Canada nor the Recherches du Canada ni les noms
* names of its contributors may de ses participants ne peuvent
* be used to endorse or promote être utilisés pour approuver ou
* products derived from this promouvoir les produits dérivés
* software without specific prior de ce logiciel sans autorisation
* written permission. préalable et particulière
* par écrit.
*
* This file is part of the Ce fichier fait partie du projet
* OpenCADC project. OpenCADC.
*
* OpenCADC is free software: OpenCADC est un logiciel libre ;
* you can redistribute it and/or vous pouvez le redistribuer ou le
* modify it under the terms of modifier suivant les termes de
* the GNU Affero General Public la “GNU Affero General Public
* License as published by the License” telle que publiée
* Free Software Foundation, par la Free Software Foundation
* either version 3 of the : soit la version 3 de cette
* License, or (at your option) licence, soit (à votre gré)
* any later version. toute version ultérieure.
*
* OpenCADC is distributed in the OpenCADC est distribué
* hope that it will be useful, dans l’espoir qu’il vous
* but WITHOUT ANY WARRANTY; sera utile, mais SANS AUCUNE
* without even the implied GARANTIE : sans même la garantie
* warranty of MERCHANTABILITY implicite de COMMERCIALISABILITÉ
* or FITNESS FOR A PARTICULAR ni d’ADÉQUATION À UN OBJECTIF
* PURPOSE. See the GNU Affero PARTICULIER. Consultez la Licence
* General Public License for Générale Publique GNU Affero
* more details. pour plus de détails.
*
* You should have received Vous devriez avoir reçu une
* a copy of the GNU Affero copie de la Licence Générale
* General Public License along Publique GNU Affero avec
* with OpenCADC. If not, see OpenCADC ; si ce n’est
* <http://www.gnu.org/licenses/>. pas le cas, consultez :
* <http://www.gnu.org/licenses/>.
*
* $Revision: 4 $
*
************************************************************************
*/
package ca.nrc.cadc.ac.server.ldap;
import org.apache.log4j.Logger;
import ca.nrc.cadc.ac.server.ldap.LdapConfig.LdapPool;
import ca.nrc.cadc.ac.server.ldap.LdapConfig.PoolPolicy;
import ca.nrc.cadc.profiler.Profiler;
import com.unboundid.ldap.sdk.FewestConnectionsServerSet;
import com.unboundid.ldap.sdk.LDAPConnection;
import com.unboundid.ldap.sdk.LDAPConnectionPool;
import com.unboundid.ldap.sdk.LDAPException;
import com.unboundid.ldap.sdk.LDAPReadWriteConnectionPool;
import com.unboundid.ldap.sdk.RoundRobinServerSet;
import com.unboundid.ldap.sdk.ServerSet;
import com.unboundid.ldap.sdk.SimpleBindRequest;
/**
* This object is designed to be shared between the DAO classes
* for access to LDAP. There should only be a single instance.
* It contains a connection pool object from the UnboundID library.
*
* @author majorb
*/
public class LdapConnectionPool
{
private static final Logger logger = Logger.getLogger(LdapConnectionPool.class);
private static final int POOL_CHECK_INTERVAL_MILLESCONDS = 10000; // 10 seconds
Profiler profiler = new Profiler(LdapConnectionPool.class);
protected LdapConfig currentConfig;
private LDAPReadWriteConnectionPool pool;
private Object poolMonitor = new Object();
private long lastPoolCheck = System.currentTimeMillis();
LdapConnectionPool()
{
this.currentConfig = LdapConfig.getLdapConfig();
pool = createPool(currentConfig);
profiler.checkpoint("Create pool");
}
LdapConnectionPool(LdapConfig config)
{
this.currentConfig = config;
pool = createPool(currentConfig);
profiler.checkpoint("Create pool");
}
protected LDAPConnection getReadOnlyConnection() throws LDAPException
{
synchronized (poolMonitor)
{
poolCheck();
LDAPConnection conn = pool.getReadConnection();
profiler.checkpoint("get read only connection");
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
return conn;
}
}
protected LDAPConnection getReadWriteConnection() throws LDAPException
{
synchronized (poolMonitor)
{
poolCheck();
LDAPConnection conn = pool.getWriteConnection();
profiler.checkpoint("get read write connection");
return conn;
}
}
protected void releaseReadOnlyConnection(LDAPConnection conn)
{
synchronized (poolMonitor)
{
pool.releaseReadConnection(conn);
}
}
protected void releaseReadWriteConnection(LDAPConnection conn)
{
synchronized (poolMonitor)
{
pool.releaseWriteConnection(conn);
}
}
protected LdapConfig getCurrentConfig()
{
return currentConfig;
}
protected void shutdown()
{
logger.debug("Shutting down pool");
pool.close();
profiler.checkpoint("Shutdown pool");
}
private void poolCheck()
{
if (timeToCheckPool())
{
synchronized (poolMonitor)
{
if (timeToCheckPool())
{
// check to see if the configuration has changed
logger.debug("checking for ldap config change");
LdapConfig newConfig = LdapConfig.getLdapConfig();
if (!newConfig.equals(currentConfig))
{
logger.debug("Detected ldap configuration change, rebuilding pools");
this.currentConfig = newConfig;
pool.close();
profiler.checkpoint("Close old pool");
pool = createPool(currentConfig);
profiler.checkpoint("Rebuild pool");
}
lastPoolCheck = System.currentTimeMillis();
}
}
}
}
private boolean timeToCheckPool()
{
return System.currentTimeMillis() - lastPoolCheck > POOL_CHECK_INTERVAL_MILLESCONDS;
}
LDAPReadWriteConnectionPool createPool(LdapConfig config)
{
LDAPConnectionPool ro = createPool(config.getReadOnlyPool(), config);
LDAPConnectionPool rw = createPool(config.getReadOnlyPool(), config);
LDAPReadWriteConnectionPool pool = new LDAPReadWriteConnectionPool(ro, rw);
return pool;
}
private LDAPConnectionPool createPool(LdapPool pool, LdapConfig config)
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
{
try
{
logger.debug("LDAP Config: " + config);
String[] hosts = pool.getServers().toArray(new String[0]);
int[] ports = new int[pool.getServers().size()];
for (int i=0; i<pool.getServers().size(); i++)
{
ports[i] = config.getPort();
}
ServerSet serverSet = null;
if (pool.getPolicy().equals(PoolPolicy.roundRobin))
{
serverSet = new RoundRobinServerSet(hosts, ports, LdapDAO.getSocketFactory(config));
}
else if (pool.getPolicy().equals(PoolPolicy.fewestConnections))
{
serverSet = new FewestConnectionsServerSet(hosts, ports, LdapDAO.getSocketFactory(config));
}
else
{
throw new IllegalStateException("Unconfigured pool policy: " + pool.getPolicy());
}
SimpleBindRequest bindRequest = new SimpleBindRequest(config.getAdminUserDN(), config.getAdminPasswd());
LDAPConnectionPool connectionPool = new LDAPConnectionPool(
serverSet, bindRequest, pool.getInitSize(), pool.getMaxSize());
return connectionPool;
}
catch (Exception e)
{
logger.error("Failed to create connection pool", e);
throw new IllegalStateException(e);
}
}
}