com.cmeza.sdgenerator.util.CustomResourceLoader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-data-generator Show documentation
Spring Data Generator for JPA repositories and managers.
The newest version!
package com.cmeza.sdgenerator.util;

import org.apache.maven.artifact.DependencyResolutionRequiredException;
import org.apache.maven.project.MavenProject;
import org.springframework.core.io.Resource;
import org.springframework.core.io.ResourceLoader;

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

/**
 * Created by carlos on 22/04/17.
 */
public class CustomResourceLoader implements ResourceLoader {

    private URLClassLoader urlClassLoader;
    private String postfix;
    private boolean overwrite;
    private String repositoryPackage;
    private String repositoryPostfix;

    public CustomResourceLoader(MavenProject project){

        try {
            List runtimeClasspathElements = project.getRuntimeClasspathElements();
            URL[] runtimeUrls = new URL[runtimeClasspathElements.size()];
            for (int i = 0; i < runtimeClasspathElements.size(); i++) {
                String element = runtimeClasspathElements.get(i);
                runtimeUrls[i] = new File(element).toURI().toURL();
            }
            urlClassLoader = new URLClassLoader(runtimeUrls, Thread.currentThread().getContextClassLoader());
        } catch (DependencyResolutionRequiredException | MalformedURLException e) {
            e.printStackTrace();
        }
    }

    @Override
    public Resource getResource(String s) {
        return null;
    }

    @Override
    public ClassLoader getClassLoader() {
        return this.urlClassLoader;
    }

    public URLClassLoader getUrlClassLoader(){
        return this.urlClassLoader;
    }

    public String getPostfix() {
        return postfix;
    }

    public CustomResourceLoader setPostfix(String postfix) {
        this.postfix = postfix;
        return this;
    }

    public boolean isOverwrite() {
        return overwrite;
    }

    public CustomResourceLoader setOverwrite(boolean overwrite) {
        this.overwrite = overwrite;
        return this;
    }

    public String getRepositoryPackage() {
        return repositoryPackage;
    }

    public CustomResourceLoader setRepositoryPackage(String repositoryPackage) {
        this.repositoryPackage = repositoryPackage;
        return this;
    }

    public String getRepositoryPostfix() {
        return repositoryPostfix;
    }

    public CustomResourceLoader setRepositoryPostfix(String repositoryPostfix) {
        this.repositoryPostfix = repositoryPostfix;
        return this;
    }
}