tech.guyi.web.quick.service.datasource.DataSourceGetter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of quick-web-service Show documentation

Quick-Web依赖

The newest version!

package tech.guyi.web.quick.service.datasource;

import org.springframework.context.ApplicationContext;

import javax.annotation.Resource;
import javax.sql.DataSource;
import java.util.Collection;
import java.util.Map;

public class DataSourceGetter {

    @Resource
    private ApplicationContext context;
    @Resource
    private DataSource dataSource;

    public DataSource get(String name){
        return this.context.getBeansOfType(DataSource.class).entrySet()
                .stream()
                .filter(e -> e.getKey().equals(name))
                .findFirst()
                .map(Map.Entry::getValue)
                .orElse(this.dataSource);
    }

    public Collection all(){
        return this.context.getBeansOfType(DataSource.class).values();
    }

}