/**
*
* Copyright 2003-2005 The Apache Software Foundation
*
* 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.
*/
package org.apache.geronimo.connector.outbound;
//import EDU.oswego.cs.dl.util.concurrent.ConcurrentHashMap;
import java.util.HashMap;
import java.util.Map;
import java.util.Iterator;
import javax.resource.ResourceException;
import javax.resource.spi.ConnectionRequestInfo;
import javax.security.auth.Subject;
import org.apache.geronimo.connector.outbound.connectionmanagerconfig.PoolingSupport;
/**
* MultiPoolConnectionInterceptor maps the provided subject and connection request info to a
* "SinglePool". This can be used to make sure all matches will succeed, avoiding synchronization
* slowdowns.
*
* Created: Fri Oct 10 12:53:11 2003
*
* @version $Rev: 355877 $ $Date: 2005-12-11 03:48:27 +0100 (Sun, 11 Dec 2005) $
*/
public class MultiPoolConnectionInterceptor implements ConnectionInterceptor, PoolingAttributes{
private final ConnectionInterceptor next;
private final PoolingSupport singlePoolFactory;
private final boolean useSubject;
private final boolean useCRI;
private final Map pools = new HashMap();
// volatile is not necessary, here, because of synchronization. but maintained for consistency with other Interceptors...
private volatile boolean destroyed = false;
public MultiPoolConnectionInterceptor(
final ConnectionInterceptor next,
PoolingSupport singlePoolFactory,
final boolean useSubject,
final boolean useCRI) {
this.next = next;
this.singlePoolFactory = singlePoolFactory;
this.useSubject = useSubject;
this.useCRI = useCRI;
}
public void getConnection(ConnectionInfo connectionInfo) throws ResourceException {
ManagedConnectionInfo mci = connectionInfo.getManagedConnectionInfo();
SubjectCRIKey key =
new SubjectCRIKey(
useSubject ? mci.getSubject() : null,
useCRI ? mci.getConnectionRequestInfo() : null);
ConnectionInterceptor poolInterceptor = null;
synchronized (pools) {
if (destroyed) {
throw new ResourceException("ConnectionManaged has been destroyed");
}
poolInterceptor = (ConnectionInterceptor) pools.get(key);
if (poolInterceptor == null) {
poolInterceptor = singlePoolFactory.addPoolingInterceptors(next);
pools.put(key, poolInterceptor);
}
}
mci.setPoolInterceptor(poolInterceptor);
poolInterceptor.getConnection(connectionInfo);
}
// let underlying pools handle destroyed processing...
public void returnConnection(
ConnectionInfo connectionInfo,
ConnectionReturnAction connectionReturnAction) {
ManagedConnectionInfo mci = connectionInfo.getManagedConnectionInfo();
ConnectionInterceptor poolInterceptor = mci.getPoolInterceptor();
poolInterceptor.returnConnection(connectionInfo, connectionReturnAction);
}
public void destroy() {
synchronized (pools) {
destroyed = true;
for (Iterator it = pools.entrySet().iterator(); it.hasNext(); ) {
((ConnectionInterceptor)((Map.Entry)it.next()).getValue()).destroy();
it.remove();
}
}
next.destroy();
}
public int getPartitionCount() {
return pools.size();
}
public int getPartitionMaxSize() {
return singlePoolFactory.getPartitionMaxSize();
}
public void setPartitionMaxSize(int maxSize) throws InterruptedException {
singlePoolFactory.setPartitionMaxSize(maxSize);
for (Iterator iterator = pools.entrySet().iterator(); iterator.hasNext();) {
PoolingAttributes poolingAttributes = (PoolingAttributes) ((Map.Entry) iterator.next()).getValue();
poolingAttributes.setPartitionMaxSize(maxSize);
}
}
public int getPartitionMinSize() {
return singlePoolFactory.getPartitionMinSize();
}
public void setPartitionMinSize(int minSize) {
singlePoolFactory.setPartitionMinSize(minSize);
for (Iterator iterator = pools.entrySet().iterator(); iterator.hasNext();) {
PoolingAttributes poolingAttributes = (PoolingAttributes) ((Map.Entry) iterator.next()).getValue();
poolingAttributes.setPartitionMinSize(minSize);
}
}
public int getIdleConnectionCount() {
int count = 0;
for (Iterator iterator = pools.entrySet().iterator(); iterator.hasNext();) {
PoolingAttributes poolingAttributes = (PoolingAttributes) ((Map.Entry) iterator.next()).getValue();
count += poolingAttributes.getIdleConnectionCount();
}
return count;
}
public int getConnectionCount() {
int count = 0;
for (Iterator iterator = pools.entrySet().iterator(); iterator.hasNext();) {
PoolingAttributes poolingAttributes = (PoolingAttributes) ((Map.Entry) iterator.next()).getValue();
count += poolingAttributes.getConnectionCount();
}
return count;
}
public int getBlockingTimeoutMilliseconds() {
return singlePoolFactory.getBlockingTimeoutMilliseconds();
}
public void setBlockingTimeoutMilliseconds(int timeoutMilliseconds) {
singlePoolFactory.setBlockingTimeoutMilliseconds(timeoutMilliseconds);
for (Iterator iterator = pools.entrySet().iterator(); iterator.hasNext();) {
PoolingAttributes poolingAttributes = (PoolingAttributes) ((Map.Entry) iterator.next()).getValue();
poolingAttributes.setBlockingTimeoutMilliseconds(timeoutMilliseconds);
}
}
public int getIdleTimeoutMinutes() {
return singlePoolFactory.getIdleTimeoutMinutes();
}
public void setIdleTimeoutMinutes(int idleTimeoutMinutes) {
singlePoolFactory.setIdleTimeoutMinutes(idleTimeoutMinutes);
for (Iterator iterator = pools.entrySet().iterator(); iterator.hasNext();) {
PoolingAttributes poolingAttributes = (PoolingAttributes) ((Map.Entry) iterator.next()).getValue();
poolingAttributes.setIdleTimeoutMinutes(idleTimeoutMinutes);
}
}
static class SubjectCRIKey {
private final Subject subject;
private final ConnectionRequestInfo cri;
private final int hashcode;
public SubjectCRIKey(
final Subject subject,
final ConnectionRequestInfo cri) {
this.subject = subject;
this.cri = cri;
this.hashcode =
(subject == null ? 17 : subject.hashCode() * 17)
^ (cri == null ? 1 : cri.hashCode());
}
public int hashCode() {
return hashcode;
}
public boolean equals(Object other) {
if (!(other instanceof SubjectCRIKey)) {
return false;
} // end of if ()
SubjectCRIKey o = (SubjectCRIKey) other;
if (hashcode != o.hashcode) {
return false;
} // end of if ()
return subject == null
? o.subject == null
: subject.equals(o.subject)
&& cri == null ? o.cri == null : cri.equals(o.cri);
}
}
} // MultiPoolConnectionInterceptor