com.spider.autoswitching.basis.bean.SpringConfigurableEnvironmentDataSourceCreateable Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spider-boot-apollo-autoswitch Show documentation

The newest version!

package com.spider.autoswitching.basis.bean;

import com.spider.autoswitching.Createable;
import com.spider.autoswitching.bind.ConfigurationPropertiesBinding;
import org.springframework.boot.autoconfigure.jdbc.DataSourceProperties;

import javax.sql.DataSource;
import java.util.Set;

/**
 * 通过 spring 环境配置 创建DataSource
 *
 * @author liuzhongkai
 */
public class SpringConfigurableEnvironmentDataSourceCreateable implements Createable {

    private ConfigurationPropertiesBinding binding;

    private Set prefixSet;

    private Class type;


    public SpringConfigurableEnvironmentDataSourceCreateable(ConfigurationPropertiesBinding binding, Class type, Set prefixSet) {
        this.binding = binding;
        this.prefixSet = prefixSet;
        this.type = type;
    }

    @Override
    public T create() {
        DataSourceProperties properties = new DataSourceProperties();
        prefixSet.forEach(item -> binding.bind(properties, item));
        T dataSource = properties.initializeDataSourceBuilder().type(type).build();
        prefixSet.forEach(item -> binding.bind(dataSource, item));
        return dataSource;
    }


}