com.jaspersoft.jasperserver.api.engine.jasperreports.service.impl.SQLDataSourceWrapper.java Source code

Introduction

Here is the source code for com.jaspersoft.jasperserver.api.engine.jasperreports.service.impl.SQLDataSourceWrapper.java

Source

/*
 * Copyright (C) 2005 - 2014 TIBCO Software Inc. All rights reserved.
 * http://www.jaspersoft.com.
 *
 * Unless you have purchased  a commercial license agreement from Jaspersoft,
 * the following license terms  apply:
 *
 * This program is free software: you can redistribute it and/or  modify
 * it under the terms of the GNU Affero General Public License  as
 * published by the Free Software Foundation, either version 3 of  the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero  General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public  License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

package com.jaspersoft.jasperserver.api.engine.jasperreports.service.impl;

import com.jaspersoft.jasperserver.api.common.util.TibcoDriverManager;
import org.apache.commons.beanutils.BeanUtils;
import org.apache.commons.dbcp.BasicDataSource;
import javax.sql.DataSource;
import java.lang.reflect.Method;
import java.sql.Connection;
import java.sql.Driver;
import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;
import java.util.logging.Logger;

public class SQLDataSourceWrapper implements DataSource {

    private DataSource basicDataSource;
    private TibcoDriverManager tibcoDriverManager;

    public SQLDataSourceWrapper(DataSource basicDataSource) {
        this.basicDataSource = basicDataSource;
        tibcoDriverManager = TibcoDriverManagerImpl.getInstance();
    }

    public Connection getConnection() throws SQLException {
        Connection connection = basicDataSource.getConnection();
        tibcoDriverManager.unlockDSConnection(connection);
        return connection;
    }

    public Connection getConnection(String username, String password) throws SQLException {
        Connection connection = basicDataSource.getConnection(username, password);
        tibcoDriverManager.unlockDSConnection(connection);
        return connection;
    }

    public <T> T unwrap(Class<T> iface) throws SQLException {
        try {
            return (T) invokeMethod(basicDataSource, "unwrap", iface);
        } catch (Exception ex) {
            throw new SQLException(ex);
        }
    }

    public boolean isWrapperFor(Class<?> iface) throws SQLException {
        try {
            return (Boolean) invokeMethod(basicDataSource, "isWrapperFor", iface);
        } catch (Exception ex) {
            throw new SQLException(ex);
        }
    }

    public Logger getParentLogger() throws SQLFeatureNotSupportedException {
        try {
            return (Logger) invokeMethod(basicDataSource, "getParentLogger", null);
        } catch (Exception ex) {
            throw new SQLFeatureNotSupportedException(ex);
        }
    }

    public int getLoginTimeout() throws SQLException {
        return basicDataSource.getLoginTimeout();
    }

    public java.io.PrintWriter getLogWriter() throws SQLException {
        return basicDataSource.getLogWriter();
    }

    public void setLoginTimeout(int loginTimeout) throws SQLException {
        basicDataSource.setLoginTimeout(loginTimeout);
    }

    public void setLogWriter(java.io.PrintWriter logWriter) throws SQLException {
        basicDataSource.setLogWriter(logWriter);
    }

    private static Object invokeMethod(Object object, String methodName, Object arg1) throws Exception {
        Method[] methods = object.getClass().getMethods();
        for (Method method : methods) {
            if (method.getName().equals(methodName)) {
                if (arg1 == null)
                    return method.invoke(object);
                else
                    return method.invoke(object, arg1);
            }
        }
        throw new Exception("Class " + object.getClass().getName() + " Invalid Method: " + methodName);
    }
}