package com.core.datasource;

import java.util.Map;  
import java.util.Map.Entry;  
  
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;

import com.alibaba.fastjson.JSON;  
  
 
/** 
 * 动态切换数据源切面 
 * @ClassName: DataSourceAspect    
 
 */  

public class DataSourceAspect{   
    private String defaultDataSource;  
    private Map<String, Object> targetDataSources;  
    	
    public void doBefore(JoinPoint joinPoint) {  
        boolean isSetDataSource = false;  
        String targetName = joinPoint.getTarget().getClass().getName(); 
        System.out.println("DataSourceAspect dataSourceChange doBefore 类名:"+targetName);
        System.out.println("targetDataSources:"+JSON.toJSONString(targetDataSources));
        System.out.println("defaultDataSource:"+JSON.toJSONString(defaultDataSource));
        System.out.println("------");
        for(Entry<String, Object> entry : targetDataSources.entrySet()) {  
            if(targetName.contains(entry.getKey())){  
                String value = entry.getValue().toString();  
                System.out.println("切换选择数据源："+value);
                DynamicDataSourceHolder.setDataSource(value);                  
                isSetDataSource = true;  
                break;  
            }  
        }  
        if(!isSetDataSource) {  
        	System.out.println("切换为默认数据源"+defaultDataSource);
        	DynamicDataSourceHolder.setDataSource(defaultDataSource);  
        }  
          
    }  
      
    public void doAfterReturning(JoinPoint joinPoint) {  
    	DynamicDataSourceHolder.clearDataSource();  
    }  
  
    public Map<String, Object> getTargetDataSources() {  
        return targetDataSources;  
    }  
  
    public void setTargetDataSources(Map<String, Object> targetDataSources) {  
        this.targetDataSources = targetDataSources;  
    }  
  
    public String getDefaultDataSource() {  
        return defaultDataSource;  
    }  
  
    public void setDefaultDataSource(String defaultDataSource) {  
        this.defaultDataSource = defaultDataSource;  
    }  
      
}