com.graphaware.integration.es.test.CustomClassLoader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of neo4j-elasticsearch-tests Show documentation

Test Utilities for GraphAware Neo4j-Elasticsearch Integration

The newest version!

/*
 * Copyright (c) 2013-2016 GraphAware
 *
 * This file is part of the GraphAware Framework.
 *
 * GraphAware Framework is free software: you can redistribute it and/or modify it under the terms
 * of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License,
 * or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of the
 * GNU General Public License along with this program.  If not, see .
 */

package com.graphaware.integration.es.test;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.List;

/**
 * Custom class loader. The ultimate purpose of this class is to avoid clashes between Neo4j and Elasticsearch
 * Lucene versions.
 */
public class CustomClassLoader {
    private static final Logger LOG = LoggerFactory.getLogger(CustomClassLoader.class);

    private final ClassLoader classloader;

    public CustomClassLoader(String libPath, ClassLoader parent) throws MalformedURLException {
        List urls = new ArrayList<>();
        File directory = new File(libPath);
        if (!directory.exists()) {
            throw new RuntimeException("Path : " + libPath + " doesn't exist");
        }
        for (File f : directory.listFiles()) {
            final URL toURL = f.toURI().toURL();
            urls.add(toURL);
            LOG.info(toURL.getPath());
        }
        this.classloader = new URLClassLoader(urls.toArray(new URL[urls.size()]), parent);
    }

    public CustomClassLoader(String libPath) throws MalformedURLException {
        this(libPath, ClassLoader.getSystemClassLoader().getParent());
    }

    public Class loadClass(String classname) throws ClassNotFoundException {
        return classloader.loadClass(classname);
    }
}