[JBoss-dev] CVS update: contrib/iiop/src/main/org/jboss/iiop WebCL.java

[Francisco Reverbel]

  User: reverbel
  Date: 02/03/10 17:43:14

  Added:       iiop/src/main/org/jboss/iiop WebCL.java
  Log:
  Initial version of WebCL, a subclass of org.jboss.web.WebClassLoader that
  dynamically generates IIOP stubs.
  
  Revision  Changes    Path
  1.1                  contrib/iiop/src/main/org/jboss/iiop/WebCL.java
  
  Index: WebCL.java
  ===================================================================
  package org.jboss.iiop;
  
  import java.net.URL;
  import java.util.Collections;
  import java.util.Map;
  import java.util.WeakHashMap;
  
  import org.jboss.ejb.Container;
  import org.jboss.logging.Logger;
  import org.jboss.proxy.compiler.IIOPStubCompiler;
  import org.jboss.system.UnifiedClassLoader;
  import org.jboss.web.WebClassLoader;
  
  /**
   * A subclass of WebClassLoader that does IIOP bytecode generation on the fly.
   *
   * @author  <a href="mailto:[EMAIL PROTECTED]";>Francisco Reverbel</a>
   * @version $Revision: 1.1 $
  */
  public class WebCL extends WebClassLoader
  {
      /** Logger for trace messages */
      static Logger logger = Logger.getLogger(WebCL.class);
  
      /** Map from stub classes into bytecode arrays (stub bytecode cache) */
      private Map loadedStubMap = Collections.synchronizedMap(new WeakHashMap());
  
      /** Creates new WebCL */
      public WebCL(Container container, UnifiedClassLoader parent)
      {
          super(container, parent);
      }
  
      /** Gets a string key used as the key into the WebServer's loaderMap. */
      public String getKey()
      {
          String className = getClass().getName();
          int dot = className.lastIndexOf('.');
          if( dot >= 0 )
              className = className.substring(dot+1);
          String jndiName = getContainer().getBeanMetaData().getJndiName();
          String key =  className + '@' + jndiName + '/';
          return key;
      }
  
      /** Gets the bytecodes for a given stub class. */
      public byte[] getBytes(Class clz) {
          byte[] code = (byte[])loadedStubMap.get(clz);
          return (code == null) ? null : (byte[])code.clone();
      }
     
      protected  Class findClass(String name) 
          throws ClassNotFoundException 
      {
          if (logger.isTraceEnabled()) {
              logger.trace("findClass(" + name + ") called");
          }
          if (name.endsWith("_Stub")) {
              int start = name.lastIndexOf('.') + 1;
              if (name.charAt(start) == '_') {
                  String pkg = name.substring(0, start);
                  String interfaceName = pkg + name.substring(start + 1, 
                                                              name.length() - 5);
                  Class intf = super.loadClass(interfaceName);
                  if (logger.isTraceEnabled()) {
                      logger.trace("loaded class " + interfaceName);
                  }
                  
                  try {
                      byte[] code = 
                          IIOPStubCompiler.compile(intf, name);
                 
                      if (logger.isTraceEnabled()) {
                          logger.trace("compiled stub class for " 
                                       + interfaceName);
                      }
                      Class clz = defineClass(name, code, 0, code.length);
                      if (logger.isTraceEnabled()) {
                          logger.trace("defined stub class for " 
                                       + interfaceName);
                      }
                      resolveClass(clz);
                      if (logger.isTraceEnabled()) {
                          logger.trace("resolved stub class for " 
                                       + interfaceName);
                      }
                      loadedStubMap.put(clz, code);
                      return clz;
                  }
                  catch (RuntimeException e) {
                      logger.error("failed finding class " + name, e);
                      //throw e;
                      return super.findClass(name);
                  }
              }
              else {
                  return super.findClass(name);
              }
          }
          else {
              return super.findClass(name);
          }
      }
  
  }
  
  
  

_______________________________________________
Jboss-development mailing list
[EMAIL PROTECTED]
https://lists.sourceforge.net/lists/listinfo/jboss-development