/*
* COMSAT
* Copyright (c) 2013-2014, Parallel Universe Software Co. All rights reserved.
*
* This program and the accompanying materials are dual-licensed under
* either the terms of the Eclipse Public License v1.0 as published by
* the Eclipse Foundation
*
* or (per the licensee's choosing)
*
* under the terms of the GNU Lesser General Public License version 3.0
* as published by the Free Software Foundation.
*/
package co.paralleluniverse.fibers.jdbc;
import java.util.Hashtable;
import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.Name;
import javax.naming.NamingException;
import javax.naming.RefAddr;
import javax.naming.Reference;
import javax.naming.spi.ObjectFactory;
import javax.sql.DataSource;
public class FiberDataSourceFactory implements ObjectFactory {
@Override
public Object getObjectInstance(Object obj, Name name, Context nameCtx, Hashtable<?, ?> environment) throws Exception {
if ((obj == null) || !(obj instanceof Reference))
return null;
Reference ref = (Reference) obj;
if (!"javax.sql.DataSource".equals(ref.getClassName()))
return null;
RefAddr ra;
ra = ref.get("rawDataSource");
if (ra == null)
throw new RuntimeException("mising rawDataSource name");
String rawDS = ra.getContent().toString();
ra = ref.get("threadsCount");
if (ra == null)
throw new RuntimeException("mising rawDataSource name");
int tc = Integer.parseInt(ra.getContent().toString());
return create(rawDS, tc);
}
public static DataSource create(String rawDS, int tc) throws NamingException {
Context envCtx = (Context) new InitialContext().lookup("java:comp/env");
DataSource ds = (DataSource) envCtx.lookup(rawDS);
return FiberDataSource.wrap(ds, tc);
}
}