com.hangum.tadpole.engine.initialize.JDBCDriverLoader.java Source code

Introduction

Here is the source code for com.hangum.tadpole.engine.initialize.JDBCDriverLoader.java

Source

/*******************************************************************************
 * Copyright (c) 2016 hangum.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser Public License v2.1
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * 
 * Contributors:
 *     hangum - initial API and implementation
 ******************************************************************************/
package com.hangum.tadpole.engine.initialize;

import java.io.File;
import java.lang.reflect.Method;
import java.net.URL;
import java.net.URLClassLoader;

import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;

/**
 * Jar Class loader
 * 
 * @author hangum
 *
 */
public class JDBCDriverLoader {
    private static final Logger logger = Logger.getLogger(JDBCDriverLoader.class);

    /**
     * add jar loader of file
     * 
     * @param strFile
     * @throws Exception
     */
    public static void addJarFile(String strFile) throws Exception {
        File file = new File(strFile);
        if (file.isFile()) {
            addJARLoader(new Object[] { file.toURI().toURL() });
        }
    }

    /**
     * add JAR Loader of dir
     * 
     * @param strDir
     * @throws Exception
     */
    public static void addJARDir(String strDir) throws Exception {
        //      if(logger.isDebugEnabled()) logger.debug("--> JAR path : " + strDir);

        File fileDir = new File(strDir);
        if (fileDir.isDirectory()) {
            File[] files = fileDir.listFiles();
            for (File file : files) {
                if (file.isDirectory()) {
                    addJARDir(file.getAbsolutePath());
                } else {
                    if (StringUtils.endsWithIgnoreCase(file.getName(), ".jar")) {
                        addJARLoader(new Object[] { file.toURI().toURL() });
                    }
                }
            }

        } else {
            if (StringUtils.endsWithIgnoreCase(fileDir.getName(), ".jar")) {
                addJARLoader(new Object[] { fileDir.toURI().toURL() });
            }
        }
    }

    /**
     * Add jar loader
     * 
     * @param jarArray
     * @throws Exception
     */
    private static void addJARLoader(Object[] jarArray) throws Exception {
        URLClassLoader systemClassLoader = (URLClassLoader) ClassLoader.getSystemClassLoader();
        try {
            Method method = URLClassLoader.class.getDeclaredMethod("addURL", new Class[] { URL.class });
            method.setAccessible(true);
            method.invoke(systemClassLoader, jarArray);
        } catch (Throwable t) {
            logger.error("jar loader", t);
        }
    }
}