Examples with SessionFactoryProxy org.hotswap.agent.plugin.hibernate3.session.proxy.SessionFactoryProxy used on opensource projects

Example 1 with SessionFactoryProxy

use of org.hotswap.agent.plugin.hibernate3.session.proxy.SessionFactoryProxy in project HotswapAgent by HotswapProjects.

the class Hibernate3Transformers method proxySessionFactory.

/**
 * Patch org.hibernate.cfg.Configuration with ReInitializable features. When
 * java8+ is supprted, then make methods default in ReInitializable
 *
 * @param classLoader
 *            the class loader
 * @param classPool
 *            the class pool
 * @param clazz
 *            the clazz
 * @throws Exception
 *             the exception
 */
@OnClassLoadEvent(classNameRegexp = "org.hibernate.cfg.Configuration")
public static void proxySessionFactory(ClassLoader classLoader, ClassPool classPool, CtClass clazz) throws Exception {
    LOGGER.debug("Adding interface o.h.a.p.h.s.p.ReInitializable to org.hibernate.cfg.Configuration.");
    clazz.addInterface(classPool.get("org.hotswap.agent.plugin.hibernate3.session.proxy.ReInitializable"));
    CtField field = CtField.make("private org.hotswap.agent.plugin.hibernate3.session.proxy.OverrideConfig $$override  = new org.hotswap.agent.plugin.hibernate3.session.proxy.OverrideConfig();", clazz);
    clazz.addField(field);
    LOGGER.debug("Patching org.hibernate.cfg.Configuration#buildSessionFactory to create a SessionFactoryProxy proxy.");
    CtMethod oldMethod = clazz.getDeclaredMethod("buildSessionFactory");
    oldMethod.setName("_buildSessionFactory");
    CtMethod newMethod = // 
    CtNewMethod.make(// 
    "public org.hibernate.SessionFactory buildSessionFactory() throws org.hibernate.HibernateException {" + "  return " + // 
    SessionFactoryProxy.class.getName() + // 
    "       .getWrapper(this)" + // 
    "       .proxy(_buildSessionFactory()); " + "}", clazz);
    clazz.addMethod(newMethod);
    LOGGER.debug("Adding org.hibernate.cfg.Configuration.reInitialize() method");
    CtMethod reInitMethod = // 
    CtNewMethod.make(// 
    "public void reInitialize(){" + // 
    "  this.settingsFactory = new org.hibernate.cfg.SettingsFactory();" + // 
    "  this.reset();" + "}", clazz);
    clazz.addMethod(reInitMethod);
    LOGGER.debug("Adding org.hibernate.cfg.Configuration.getOverrideConfig() method");
    CtMethod internalPropsMethod = // 
    CtNewMethod.make(// 
    "public org.hotswap.agent.plugin.hibernate3.session.proxy.OverrideConfig getOverrideConfig(){" + // 
    "  return $$override;" + "}", clazz);
    clazz.addMethod(internalPropsMethod);
    CtConstructor con = clazz.getDeclaredConstructor(new CtClass[] {});
    LOGGER.debug("Patching org.hibernate.cfg.Configuration.<init>");
    // 
    con.insertAfter(// 
    "java.lang.ClassLoader $$cl = Thread.currentThread().getContextClassLoader();" + // 
    PluginManagerInvoker.buildInitializePlugin(Hibernate3Plugin.class, "$$cl") + // 
    "java.lang.String $$version = org.hibernate.Version.getVersionString();" + // 
    PluginManagerInvoker.buildCallPluginMethod("$$cl", Hibernate3Plugin.class, "setVersion", "$$version", "java.lang.String"));
    ProxyUtil.addMethod(classLoader, classPool, clazz, "void", "hotSwap", null);
    ProxyUtil.addMethod(classLoader, classPool, clazz, "org.hibernate.cfg.Configuration", "setProperty", new String[] { "java.lang.String", "java.lang.String" });
    ProxyUtil.addMethod(classLoader, classPool, clazz, "org.hibernate.cfg.Configuration", "configure", new String[] { "java.lang.String" });
    ProxyUtil.addMethod(classLoader, classPool, clazz, "org.hibernate.cfg.Configuration", "configure", new String[] { "java.net.URL" });
    ProxyUtil.addMethod(classLoader, classPool, clazz, "org.hibernate.cfg.Configuration", "configure", new String[] { "java.io.File" });
    ProxyUtil.addMethod(classLoader, classPool, clazz, "org.hibernate.cfg.Configuration", "configure", new String[] { "org.w3c.dom.Document" });
    ProxyUtil.addMethod(classLoader, classPool, clazz, "org.hibernate.cfg.Configuration", "configure", null);
    LOGGER.info("Hibernate3Plugin, patched org.hibernate.cfg.Configuration");
}