/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.naming.factory;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.sql.SQLException;
import java.util.Hashtable;

import javax.naming.Context;
import javax.naming.Name;
import javax.naming.NamingException;
import javax.naming.RefAddr;
import javax.naming.Reference;
import javax.sql.DataSource;



Object factory for resource links for shared data sources.

/** * <p>Object factory for resource links for shared data sources.</p> * */
public class DataSourceLinkFactory extends ResourceLinkFactory { public static void setGlobalContext(Context newGlobalContext) { ResourceLinkFactory.setGlobalContext(newGlobalContext); } // ------------------------------------------------- ObjectFactory Methods
Create a new DataSource instance.
Params:
  • obj – The reference object describing the DataSource
/** * Create a new DataSource instance. * * @param obj The reference object describing the DataSource */
@Override public Object getObjectInstance(Object obj, Name name, Context nameCtx, Hashtable<?,?> environment) throws NamingException { Object result = super.getObjectInstance(obj, name, nameCtx, environment); // Can we process this request? if (result!=null) { Reference ref = (Reference) obj; RefAddr userAttr = ref.get("username"); RefAddr passAttr = ref.get("password"); if (userAttr.getContent()!=null && passAttr.getContent()!=null) { result = wrapDataSource(result,userAttr.getContent().toString(), passAttr.getContent().toString()); } } return result; } protected Object wrapDataSource(Object datasource, String username, String password) throws NamingException { try { DataSourceHandler handler = new DataSourceHandler((DataSource)datasource, username, password); return Proxy.newProxyInstance(datasource.getClass().getClassLoader(), datasource.getClass().getInterfaces(), handler); }catch (Exception x) { if (x instanceof InvocationTargetException) { Throwable cause = x.getCause(); if (cause instanceof ThreadDeath) { throw (ThreadDeath) cause; } if (cause instanceof VirtualMachineError) { throw (VirtualMachineError) cause; } if (cause instanceof Exception) { x = (Exception) cause; } } if (x instanceof NamingException) throw (NamingException)x; else { NamingException nx = new NamingException(x.getMessage()); nx.initCause(x); throw nx; } } }
Simple wrapper class that will allow a user to configure a ResourceLink for a data source so that when DataSource.getConnection() is called, it will invoke DataSource.getConnection(String, String) with the preconfigured username and password.
/** * Simple wrapper class that will allow a user to configure a ResourceLink for a data source * so that when {@link javax.sql.DataSource#getConnection()} is called, it will invoke * {@link javax.sql.DataSource#getConnection(String, String)} with the preconfigured username and password. */
public static class DataSourceHandler implements InvocationHandler { private final DataSource ds; private final String username; private final String password; private final Method getConnection; public DataSourceHandler(DataSource ds, String username, String password) throws Exception { this.ds = ds; this.username = username; this.password = password; getConnection = ds.getClass().getMethod("getConnection", String.class, String.class); } @Override public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { if ("getConnection".equals(method.getName()) && (args==null || args.length==0)) { args = new String[] {username,password}; method = getConnection; } else if ("unwrap".equals(method.getName())) { return unwrap((Class<?>)args[0]); } try { return method.invoke(ds,args); }catch (Throwable t) { if (t instanceof InvocationTargetException && t.getCause() != null) { throw t.getCause(); } else { throw t; } } } public Object unwrap(Class<?> iface) throws SQLException { if (iface == DataSource.class) { return ds; } else { throw new SQLException(sm.getString("dataSourceLinkFactory.badWrapper", iface.getName())); } } } }