package com.changhong.epc.config.datasource;

import com.changhong.epc.constter.system.DataSourceKey;
import com.iemsoft.framework.cloud.config.mybatis.MapperConfig;
import com.iemsoft.framework.cloud.core.thread.Keys;
import com.iemsoft.framework.cloud.core.thread.ThreadData;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.mapping.MappedStatement;
import org.springframework.boot.autoconfigure.AutoConfigureAfter;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.Environment;
import tk.mybatis.spring.mapper.MapperScannerConfigurer;

import java.util.HashMap;
import java.util.Map;

@Configuration
@AutoConfigureAfter(DataSourceConfig.class)
@Slf4j
public class MybatisConfig extends MapperConfig implements DataSourceKey{

	/**
	 * 多数据源改成单数据源
	 * @author WangYX
	 *
	 */
	private static Map<String, String> DATA_SOURCE = new HashMap<>();
	
	static{
		DATA_SOURCE.put("^com[.]changhong[.]epc[.][^.]+[.]mapper[.]centen.+", ADMIN_DATA_SOURCE);
	}

	private static void setDataSource(MappedStatement ms, String id) {
		for(Map.Entry<String, String> entry : DATA_SOURCE.entrySet()){
        	if(id.matches(entry.getKey())){
        		ThreadData.set(Keys.DATA_SOURCE, entry.getValue());
        		break;
        	}
        }
	}

	@Override
	public MapperScannerConfigurer mapperScannerConfigurer(Environment environment) {
		log.debug(environment.getProperty(getBasePackageKey()));
		return super.mapperScannerConfigurer(environment);
	}

}