org.onetwo.common.db.filequery.SimpleSqlFileScanner Maven / Gradle / Ivy

Go to download
package org.onetwo.common.db.filequery;

import java.io.File;
import java.util.Arrays;
import java.util.Collection;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

import org.onetwo.common.db.spi.SqlFileScanner;
import org.onetwo.common.file.FileUtils;
import org.onetwo.common.log.JFishLoggerFactory;
import org.onetwo.common.propconf.ResourceAdapter;
import org.onetwo.common.spring.utils.SpringResourceAdapterImpl;
import org.onetwo.common.utils.ClassUtils;
import org.onetwo.common.utils.LangUtils;
import org.onetwo.common.utils.StringUtils;
import org.onetwo.dbm.exception.DbmException;
import org.slf4j.Logger;
import org.springframework.core.io.ClassPathResource;

import com.google.common.collect.Maps;

public class SimpleSqlFileScanner implements SqlFileScanner {
	protected Logger logger = JFishLoggerFactory.getLogger(this.getClass());
	protected ClassLoader classLoader = ClassUtils.getDefaultClassLoader();
	protected String dir = "sql";
	protected String sqlPostfix = SQL_POSTFIX;
	protected String jfishPostfix = ".jfish"+sqlPostfix;
	

	public SimpleSqlFileScanner(ClassLoader classLoader) {
		super();
		this.classLoader = classLoader;
	}

	@SuppressWarnings("unchecked")
	@Override
	public ResourceAdapter getClassPathResource(String dialect, String classPath) {
		String postfix = getDialetSqlPostfix(dialect);
		String path = dir+"/"+classPath+postfix;
		ClassPathResource cpr = new ClassPathResource(path);
		SpringResourceAdapterImpl springRes = new SpringResourceAdapterImpl(cpr, postfix);
		return springRes;
	}

	public String getSqlPostfix() {
		return sqlPostfix;
	}

	public String getJfishPostfix() {
		return jfishPostfix;
	}

	/****
	 * 根据配置扫描文件
	 * @param conf
	 * @return
	 */
	@Override
	public Map> scanMatchSqlFiles(String dialectDir){
		try {
			Map> allSqlFileMap = Maps.newHashMap();
			
			Map> sqlFileMap = scanCommonSqlFiles();
			if(!LangUtils.isEmpty(sqlFileMap)){
				allSqlFileMap.putAll(sqlFileMap);
			}
			//scan dialectDir
			Map> dbsqlfileMap = scanDialetSqlFiles(dialectDir);
			
			if(!LangUtils.isEmpty(dbsqlfileMap)){
				allSqlFileMap.putAll(dbsqlfileMap);
			}
			
			return allSqlFileMap;
		} catch (Exception e) {
			throw new DbmException("scan sql file error!", e);
		}
	}

	protected Map> scanCommonSqlFiles() throws Exception {
		String sqldirPath = FileUtils.getResourcePath(classLoader, dir);

		String postfix = jfishPostfix;
		ResourceAdapter[] sqlfileArray = scanSqlFilesByDir(sqldirPath, postfix);
		if(logger.isInfoEnabled())
			logger.info("find {} [{}] named file in dir {}", sqlfileArray.length, postfix, sqldirPath);
		return map(Arrays.asList(sqlfileArray), postfix);
	}
	
	protected String getDialetSqlPostfix(String dialect){
		if(StringUtils.isBlank(dialect)){
			return jfishPostfix;
		}
		String postfix = "."+dialect+sqlPostfix;
		return postfix;
	}

	protected Map> scanDialetSqlFiles(String dialectDir)  throws Exception {
		if(StringUtils.isBlank(dialectDir)){
			return null;
		}

//		String postfix = "."+dialectDir+SQL_POSTFIX;
		String postfix = getDialetSqlPostfix(dialectDir);
		String sqldirPath = FileUtils.getResourcePath(classLoader, dir);
/*		String overridePath = sqldirPath+"/"+dialectDir;
		logger.info("scan database dialect dir : " + overridePath);
		ResourceAdapter[] dbsqlfiles = scanSqlFilesByDir(overridePath, postfix);
		if(logger.isInfoEnabled())
			logger.info("find {} sql named file in dir {}", dbsqlfiles.length, overridePath);*/
		ResourceAdapter[] dbsqlfiles = scanSqlFilesByDir(sqldirPath, postfix);
		if(logger.isInfoEnabled())
			logger.info("find {} [{}] named file in dir {}", dbsqlfiles.length, postfix, sqldirPath);
		return  map(Arrays.asList(dbsqlfiles), postfix);
	}
	
	protected ResourceAdapter[] scanSqlFilesByDir(String sqldirPath, String postfix){
		File[] sqlfileArray = FileUtils.listFiles(sqldirPath, postfix);
		return FileUtils.adapterResources(sqlfileArray, postfix);
	}
	
	protected Function, String> keyfunc(String postfix){
		return r->{
			String fileName = r.getName();
			fileName = fileName.substring(0, fileName.length()-postfix.length());
			return fileName;
		};
	}
	
	protected Map> map(Collection> resources, String postfix){
		return resources.stream().collect(Collectors.toMap(keyfunc(postfix), r->r));
	}
}