SaasDataSource.java

package org.dayatang.datasource4saas;

import org.dayatang.utils.Assert;
import org.dayatang.utils.BeanUtils;

import javax.sql.DataSource;
import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;
import java.util.logging.Logger;

/**
 * 用于SaaS的数据源实现。是本项目的中心类。它作为代理，自动为不同的租户分配不同的实际数据源。
 * @author yyang (<a href="mailto:gdyangyu@gmail.com">gdyangyu@gmail.com</a>)
 *
 */
public class SaasDataSource implements DataSource {

	private TenantService tenantService;
	private DataSourceRegistry dataSourceRegistry;

	public SaasDataSource(TenantService tenantService, DataSourceRegistry dataSourceRegistry) {
		Assert.notNull(dataSourceRegistry, "data source registry is null!");
		Assert.notNull(tenantService, "Tenant service is null!");
		this.tenantService = tenantService;
		this.dataSourceRegistry = dataSourceRegistry;
	}

	@Override
	public PrintWriter getLogWriter() throws SQLException {
		return getActualDataSource().getLogWriter();
	}

	@Override
	public int getLoginTimeout() throws SQLException {
		return getActualDataSource().getLoginTimeout();
	}

	@Override
	public void setLogWriter(PrintWriter out) throws SQLException {
		getActualDataSource().setLogWriter(out);
	}

	@Override
	public void setLoginTimeout(int seconds) throws SQLException {
		getActualDataSource().setLoginTimeout(seconds);
	}

	@Override
	public boolean isWrapperFor(Class<?> iface) throws SQLException {
		return getActualDataSource().isWrapperFor(iface);
	}

	@Override
	public <T> T unwrap(Class<T> iface) throws SQLException {
		return getActualDataSource().unwrap(iface);
	}

	@Override
	public Connection getConnection() throws SQLException {
		return getActualDataSource().getConnection();
	}

	@Override
	public Connection getConnection(String username, String password)
			throws SQLException {
		return getActualDataSource().getConnection(username, password);
	}

	private DataSource getActualDataSource() {
		return dataSourceRegistry.getDataSourceOfTenant(tenantService.getTenant());
	}

	//For JDK 7 compatability
	public Logger getParentLogger() throws SQLFeatureNotSupportedException {
        BeanUtils beanUtils = new BeanUtils(getActualDataSource());
        return (Logger) beanUtils.getPropValue("parentLogger");
	}
	
}