io.github.heykb.sqlhelper.spring.dynamicds.SqlHelperDynamicDataSourceProxy Maven / Gradle / Ivy

Go to download

package io.github.heykb.sqlhelper.spring.dynamicds;

import io.github.heykb.sqlhelper.dynamicdatasource.SqlHelperDsManager;
import org.springframework.core.InfrastructureProxy;

import javax.sql.DataSource;
import java.util.function.Function;

class SqlHelperDynamicDataSourceProxy extends io.github.heykb.sqlhelper.dynamicdatasource.SqlHelperDynamicDataSourceProxy implements InfrastructureProxy {


    public SqlHelperDynamicDataSourceProxy(SqlHelperDsManager sqlHelperDsManager) {
        super(sqlHelperDsManager);
    }

    /**
     * 适应spring事务，做到数据源切换后再开启的事务为一个新的事务，保证正常切换。
     * @See    org.springframework.transaction.support.TransactionSynchronizationUtils#unwrapResourceIfNecessary()
     * @return
     */
    @Override
    public Object getWrappedObject() {
        return getDatasource();
    }
}