CustomSqlSessionFactoryBean

import java.io.File;
import java.net.JarURLConnection;
import java.net.URL;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;

import org.apache.ibatis.type.Alias;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.mybatis.spring.SqlSessionFactoryBean;

import com.yb.common.system.string.StringUtils;

public class CustomSqlSessionFactoryBean extends SqlSessionFactoryBean {

    private static final Logger logger = LogManager.getLogger(CustomSqlSessionFactoryBean.class);

    @Override
    public void setTypeAliasesPackage(String typeAliasesPackage) {

        logger.debug("typeAliasesPackage = " + typeAliasesPackage);

        Set<String> typeAliasesPackages = CustomSqlSessionFactoryBean
                .getTypeAliasesPackageFromFileAndJar(typeAliasesPackage);

        String typeAliasesPackageTemp = StringUtils.join(typeAliasesPackages, ",");

        logger.debug("typeAliasesPackageTemp = " + typeAliasesPackageTemp);

        super.setTypeAliasesPackage(typeAliasesPackageTemp);
    }

    private static Set<String> getTypeAliasesPackageFromFileAndJar(String typeAliasesPackage) {

        Set<String> packageNames = new HashSet<String>();

        ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
        try {
            String typeAliasesPackagePath = typeAliasesPackage.replace(".", "/");

            Enumeration<URL> urls = classLoader.getResources(typeAliasesPackagePath);

            while (urls.hasMoreElements()) {
                URL url = urls.nextElement();

                logger.debug(url.getPath());

                if (url != null) {
                    String protocol = url.getProtocol();

                    switch (protocol) {
                    case "file":
                        packageNames.addAll(extractPackageFromFile(url, typeAliasesPackage));
                        break;
                    case "jar":
                        packageNames.addAll(extractPackageFromJar(url, typeAliasesPackage));
                        break;
                    default:
                        ;
                    }
                }
            }

        } catch (Exception e) {
            logger.error("CustomSqlSessionFactoryBean exception");
            e.printStackTrace();
        }

        return packageNames;
    }

    private static Set<String> extractPackageFromFile(URL url, String typeAliasesPackage) throws Exception {

        List<String> packageNames = new ArrayList<String>();
        Set<String> set = new HashSet<String>();

        String packagePath = typeAliasesPackage.replace(".", File.separator);

        String path = url.getPath();

        File file = new File(path);

        findClass(file, packageNames);

        for (int i = 0; i < packageNames.size(); i++) {
            String pkgName = packageNames.get(i);
            String classNamePath = pkgName.substring(pkgName.indexOf(packagePath), packageNames.get(i).length())
                    .replace(".class", "");
            String className = classNamePath.replace(File.separator, ".");

            Class<?> clazz = Class.forName(className);

            if (clazz.isAnnotationPresent(Alias.class)) {
                set.add(classNamePath.substring(0, classNamePath.lastIndexOf(File.separator)).replace(File.separator,
                        "."));
            }
        }

        logger.debug("extractPackageFromFile set = " + set);

        return set;
    }

    private static void findClass(File file, List<String> packageNames) {

        if (file.exists() && file.isDirectory()) {

            for (File f : file.listFiles()) {
                findClass(f, packageNames);
            }
        }

        if (file.getName().endsWith(".class")) {
            packageNames.add(file.getAbsolutePath());
        }

    }

    private static Set<String> extractPackageFromJar(URL url, String typeAliasesPackage) throws Exception {

        Set<String> packageNames = new HashSet<String>();

        JarURLConnection jarURLConnection = (JarURLConnection) url.openConnection();

        JarFile jarFile = jarURLConnection.getJarFile();
        Enumeration<JarEntry> entries = jarFile.entries();

        while (entries.hasMoreElements()) {
            JarEntry jarEntry = entries.nextElement();
            String jarEntryName = jarEntry.getName();

            if (jarEntryName.endsWith(".class")) {
                String className = jarEntryName.substring(0, jarEntryName.indexOf(".class")).replace("/", ".");
                Class<?> clazz = Class.forName(className);

                if (clazz.isAnnotationPresent(Alias.class)) {

                    String packageName = className.substring(0, className.lastIndexOf("."));

                    logger.debug("typeAliasesPackage = " + packageName);

                    packageNames.add(packageName);
                }
            }
        }

        logger.debug("extractPackageFromJar set = " + packageNames);

        return packageNames;
    }

}