com.avaje.ebean.enhance.agent.ClassPathClassBytesReader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of avaje-ebeanorm-agent Show documentation

There is a newer version: 8.1.1

package com.avaje.ebean.enhance.agent;

import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.net.URLClassLoader;

/**
 * Implementation of ClassBytesReader based on URLClassLoader.
 */
public class ClassPathClassBytesReader implements ClassBytesReader {
	

	private final URL[] urls;
	
	public ClassPathClassBytesReader(URL[] urls) {
		this.urls = urls == null ? new URL[0]: urls;
	}
	
	public byte[] getClassBytes(String className, ClassLoader classLoader) {

		URLClassLoader cl = new URLClassLoader(urls, classLoader);

		String resource = className.replace('.', '/') + ".class";

		InputStream is = null;
		try {

			// read the class bytes, and define the class
			URL url = cl.getResource(resource);
			if (url == null) {
				return null;
			}
	
			is = url.openStream();
			return InputStreamTransform.readBytes(is);
			
		} catch (IOException e){
			throw new RuntimeException("IOException reading bytes for "+className, e);
			
		} finally {
			if (is != null){
				try {
					is.close();
				} catch (IOException e) {
					throw new RuntimeException("Error closing InputStream for "+className, e);
				}
			}
      try {
        cl.close();
      } catch (NoSuchMethodError e) {
        // ignoring as running in JDK6 
      } catch (IOException e) {
        throw new RuntimeException("Error closing URLClassLoader for "+className, e);
      }
		}
	}
	
}