Open Source Repository

Home /hibernate/hibernate-3.2.4.ga | Repository Home


org/hibernate/bytecode/javassist/TransformingClassLoader.java
package org.hibernate.bytecode.javassist;

import javassist.ClassPool;
import javassist.NotFoundException;
import javassist.CtClass;
import javassist.CannotCompileException;
import org.hibernate.HibernateException;

import java.io.IOException;
import java.util.ArrayList;

/**
 @author Steve Ebersole
 */
public class TransformingClassLoader extends ClassLoader {
  private ClassLoader parent;
  private ClassPool classPool;

  /*package*/ TransformingClassLoader(ClassLoader parent, String[] classpath) {
    this.parent = parent;
    classPool = new ClassPooltrue );
    for int i = 0; i < classpath.length; i++ ) {
      try {
        classPool.appendClassPathclasspath[i] );
      }
      catch NotFoundException e ) {
        throw new HibernateException(
            "Unable to resolve requested classpath for transformation [" +
            classpath[i"] : " + e.getMessage()
        );
      }
    }
  }

  protected Class findClass(String namethrows ClassNotFoundException {
        try {
            CtClass cc = classPool.getname );
          // todo : modify the class definition if not already transformed...
            byte[] b = cc.toBytecode();
            return defineClassname, b, 0, b.length );
        }
        catch NotFoundException e ) {
            throw new ClassNotFoundException();
        }
        catch IOException e ) {
            throw new ClassNotFoundException();
        }
        catch CannotCompileException e ) {
            throw new ClassNotFoundException();
        }
    }

  public void release() {
    classPool = null;
    parent = null;
  }
}