| /* |
| * Copyright (c) 2010, 2021 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 v. 2.0 which is available at |
| * http://www.eclipse.org/legal/epl-2.0, |
| * or the Eclipse Distribution License v. 1.0 which is available at |
| * http://www.eclipse.org/org/documents/edl-v10.php. |
| * |
| * SPDX-License-Identifier: EPL-2.0 OR BSD-3-Clause |
| */ |
| |
| // 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; |
| |
| /** |
| * Default constructor. |
| */ |
| public GridLinkDataPartitioningCallback() { |
| super(); |
| } |
| |
| /** |
| * Registration only occurs once in WLS (against all data sources), so must be static registered. |
| */ |
| @Override |
| 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<Object> 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. |
| */ |
| @Override |
| public void setPartitionId(int id) { |
| partitionId.set(id); |
| } |
| |
| @Override |
| public int getPartitionId() { |
| Integer id = (Integer)partitionId.get(); |
| if (id == null) { |
| return 0; |
| } |
| return id; |
| } |
| } |