/**
 * @author jiaoyi 通过 JNDI 远程链接 JBoss 的数据源 会有问题 抛出
 *         java.lang.IllegalAccessException: Method=public abstract
 *         java.sql.Connection java.sql.Statement.getConnection() throws
 *         java.sql.SQLException does not return Serializable
 *
 *         这个 家伙 通过hook 代码 将 原始的 Connection 返回给上层 解决这个问题
 *
 */
public class JndiDataSourceProxy implements FactoryBean {
    private static final ClassLoader THIS_LOADER = JndiDataSourceProxy.class
            .getClassLoader();

    private Object jndiObject;

    public void setJndiObject(Object jndiObject) {
        this.jndiObject = jndiObject;
    }

    private class StatementHandler implements InvocationHandler {
        private DataSource ds = (DataSource) jndiObject;
        private Object target;

        StatementHandler(Object target) {
            this.target = target;
        }

        public Object invoke(Object proxy, Method method, Object[] args)
                throws Throwable {
            if ("getConnection".equals(method.getName())) {
                return ds.getConnection();
            }
            return method.invoke(target, args);
        }

    }

    private class ConnectionHandler implements InvocationHandler {
        private Connection target;

        ConnectionHandler(Connection target) {
            this.target = target;
        }

        public Object invoke(Object proxy, Method method, Object[] args)
                throws Throwable {

            if ("prepareStatement".equals(method.getName())) {
                return (Statement) Proxy.newProxyInstance(THIS_LOADER,
                        new Class[] { PreparedStatement.class },
                        new StatementHandler(method.invoke(target, args)));
            }
            return method.invoke(target, args);
        }

    }

    private class DataSourceHandler implements InvocationHandler {
        private DataSource ds = (DataSource) jndiObject;

        public Object invoke(Object proxy, Method method, Object[] args)
                throws Throwable {
            if ("getConnection".equals(method.getName())) {
                return (Connection) Proxy.newProxyInstance(THIS_LOADER,
                        new Class[] { Connection.class },
                        new ConnectionHandler(ds.getConnection()));
            }

            return method.invoke(JndiDataSourceProxy.this.jndiObject, args);
        }

    }

    public Object getObject() throws Exception {
        return (DataSource) Proxy.newProxyInstance(THIS_LOADER,
                new Class[] { DataSource.class }, new DataSourceHandler());
    }

    public Class<DataSource> getObjectType() {
        return DataSource.class;
    }

    public boolean isSingleton() {
        return true;
    }

}