使用CGLIB进行类生成和增强

时间:2015-11-07 16:54:38

标签: java scala cglib

代码在Scala中,但希望Java程序员也能理解。

我有以下课程:

class SampleClass {
  def test(in: String) = "Hello world!"
}

我有以下代码创建此类的增强实例:

val e = new Enhancer
e.setSuperclass(classOf[SampleClass])
e.setCallback(new Cb)
println(e.create.asInstanceOf[SampleClass].test("foo"))

回调如下:

class Cb extends MethodInterceptor {
  override def intercept(obj: Any, m: Method, args: Array[Object], proxy: MethodProxy): Object = {
    println(s"$m is called: ${m.getName}")
    proxy.invokeSuper(obj, args)
  }
}

这很好用。现在,如果我尝试创建然后创建它的新实例,则会直接调用所有方法,绕过回调:

val e = new Enhancer
e.setSuperclass(classOf[SampleClass])
e.setCallbackType(classOf[Cb])
println(e.createClass.newInstance.asInstanceOf[SampleClass].test("foo"))

为什么这样?如何创建java.lang.Class并使用其newInstance()方法生成增强对象?

1 个答案:

答案 0 :(得分:1)

解决了它。您需要实例化实例,但要避免在进程中调用其构造函数。然后通过调用其私有方法手动绑定回调:

val instance = instantiator.newInstance.asInstanceOf[SampleClass]

val bc  = subclass.getDeclaredMethod("CGLIB$BIND_CALLBACKS", classOf[Object])
val stc = subclass.getDeclaredMethod("CGLIB$SET_THREAD_CALLBACKS", classOf[Array[Callback]])
bc .setAccessible(true)
stc.setAccessible(true)

stc.invoke(null, Array[Callback](new Cb))
bc .invoke(null, instance)
stc.invoke(null, null)

instantiator此处的类型为ObjectInstantiator - 请使用Objenesis以避免调用构造函数。 subclass这里是由e.createClass生成的子类。

避免在进程中调用构造函数很重要,因为增强的构造函数将设置实例创建的标志,并且无法绑定回调。顺便说一句,您还应该在整个操作之后设置此标志,以保持一致性。它的名字是CGLIB$CONSTRUCTED

您可以看到增强型构造函数的工作原理here