org.eclipse.persistence.platform.database.oracle.ucp.GridLinkDataPartitioningCallback Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of eclipselink Show documentation
Show all versions of eclipselink Show documentation
EclipseLink build based upon Git transaction 774c696
/*******************************************************************************
* Copyright (c) 2010, 2013 Oracle and/or its affiliates. All rights reserved.
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0
* which accompanies this distribution.
* The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
* and the Eclipse Distribution License is available at
* http://www.eclipse.org/org/documents/edl-v10.php.
*
* Contributors:
* James Sutherland - initial API and implementation
******************************************************************************/
package org.eclipse.persistence.platform.database.oracle.ucp;
import java.lang.reflect.Method;
import javax.sql.DataSource;
import oracle.ucp.jdbc.oracle.DataBasedConnectionAffinityCallback;
import org.eclipse.persistence.internal.security.PrivilegedAccessHelper;
import org.eclipse.persistence.logging.SessionLog;
import org.eclipse.persistence.sessions.Session;
/**
* PUBLIC:
* Integrates with WebLogic GirdLink's data affinity support.
*
* @see org.eclipse.persistence.descriptors.partitioning.PartitioningPolicy
* @author James Sutherland
* @since EclipseLink 2.3
*/
public class GridLinkDataPartitioningCallback extends UCPDataPartitioningCallback {
/** The id is stored in a static thread local. */
protected static ThreadLocal partitionId = new ThreadLocal();
public static boolean isRegistered = false;
/**
* Registration only occurs once in WLS (against all data sources), so must be static registered.
*/
public void register(DataSource datSource, Session session) {
if (isRegistered) {
return;
}
register(session);
}
/**
* Register with WLS through reflection.
*/
public static synchronized void register(Session session) {
if (isRegistered) {
return;
}
try {
Class dataSourceManager = PrivilegedAccessHelper.getClassForName("weblogic.jdbc.common.internal.DataSourceManager");
Method getInstance = PrivilegedAccessHelper.getMethod(dataSourceManager, "getInstance", null, false);
Object instance = PrivilegedAccessHelper.invokeMethod(getInstance, null, null);
Method getDataSourceService = PrivilegedAccessHelper.getMethod(instance.getClass(), "getDataSourceService", null, false);
Object service = PrivilegedAccessHelper.invokeMethod(getDataSourceService, instance, null);
Class[] argumentTypes = new Class[] {DataBasedConnectionAffinityCallback.class};
Method registerDataAffinityCallback = PrivilegedAccessHelper.getMethod(service.getClass(), "registerDataAffinityCallback", argumentTypes, false);
Object[] arguments = new Object[] {new GridLinkDataPartitioningCallback()};
PrivilegedAccessHelper.invokeMethod(registerDataAffinityCallback, service, arguments);
isRegistered = true;
} catch (Exception exception) {
session.getSessionLog().logThrowable(SessionLog.WARNING, SessionLog.CONNECTION, exception);
}
}
/**
* Set the partition id for this thread.
*/
public void setPartitionId(int id) {
partitionId.set(id);
}
public int getPartitionId() {
Integer id = (Integer)partitionId.get();
if (id == null) {
return 0;
}
return id;
}
}