com.dream.boot.share.DataSourceConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dream-orm-spring-boot-starter Show documentation
The newest version!
package com.dream.boot.share;

import com.dream.boot.autoconfigure.DreamProperties;
import com.dream.mate.share.datasource.ShareDataSource;
import com.dream.mate.share.listener.ShardListener;
import com.dream.system.core.listener.Listener;
import com.dream.system.core.listener.factory.ListenerFactory;
import com.dream.util.common.ObjectUtil;
import com.dream.util.exception.DreamRunTimeException;
import com.dream.util.reflect.ReflectUtil;
import org.springframework.beans.BeansException;
import org.springframework.cglib.beans.BeanMap;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.ImportAware;
import org.springframework.core.type.AnnotationMetadata;

import javax.sql.DataSource;
import java.util.HashMap;
import java.util.Map;

public class DataSourceConfiguration implements ImportAware, ApplicationContextAware {
    private Class dataSourceType;


    @Bean
    public DataSource dataSource(DreamProperties dreamProperties) {
        Map> datasource = dreamProperties.getDatasource();
        if (datasource == null || datasource.isEmpty()) {
            throw new DreamRunTimeException("数据源未配置");
        }
        Map dataSourceMap = new HashMap<>(4);
        datasource.forEach((k, v) -> {
            DataSource dataSource = ReflectUtil.create(dataSourceType);
            BeanMap beanMap = BeanMap.create(dataSource);
            beanMap.putAll(v);
            dataSourceMap.put(k, dataSource);
        });
        return new ShareDataSource(dataSourceMap);
    }

    @Override
    public void setImportMetadata(AnnotationMetadata importMetadata) {
        Map annotationAttributes = importMetadata.getAnnotationAttributes(EnableShare.class.getName());
        this.dataSourceType = (Class) annotationAttributes.get("value");
    }

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        ListenerFactory listenerFactory = applicationContext.getBean(ListenerFactory.class);
        Listener[] listeners = listenerFactory.getListeners();
        if (!ObjectUtil.isNull(listeners)) {
            for (Listener listener : listeners) {
                if (listener instanceof ShardListener) {
                    return;
                }
            }
        }
        listeners = ObjectUtil.merge(listeners, new Listener[]{new ShardListener()});
        listenerFactory.listeners(listeners);
    }
}