Измените метод, объявленный в суперклассе с помощью Javassist

Question

Я пытаюсь изменить метод с CtMethod#insertBefore, который объявлен в суперклассе. Однако с Джавасистом это кажется невозможным.

private class AbstractTestDataSource { 
    public Connection getConnection() throws SQLException { 
     return connection; 
    } 
} 

private class TestDataSource extends AbstractTestDataSource implements DataSource { 
    public Connection getConnection(String username, String password) throws SQLException { 
     return connection; 
    } 
    // other methods omitted 
} 

Это мой ClassFileTransformer

public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, 
            ProtectionDomain protectionDomain, byte[] classfileBuffer) 
     throws Exception { 
    if (!className.equals("org/example/TestDataSource")) { 
     return classfileBuffer; 
    } 
    final CtClass ctClass = createCtClass(loader, classfileBuffer); 
    for (CtMethod method : ctClass.getMethods()) { 
     if (method.getName().equals("getConnection")) { 
      System.out.print(method.getName()); 
      System.out.println(method.getSignature()); 
      method.insertBefore("System.out.println(\"foo\");"); 
     } 
    } 
    return ctClass.toBytecode(); 
} 

Когда я звоню метод getConnection(String, String), foo выводится на консоль, но если я называю getConnection() метод, объявленный в AbstractTestDataSource ничего не происходит. Что я делаю неправильно?

Редактировать

Я могу подтвердить, что оба метода инструментальными, потому что это то, что выводится на консоль:

getConnection(Ljava/lang/String;Ljava/lang/String;)Ljava/sql/Connection; 
getConnection()Ljava/sql/Connection; 

Answer 1

Мое решение - проверить, объявлен ли метод getConnection в классе, отличном от текущего.

if (!ctClass.equals(method.getDeclaringClass())) { 
    method = overrideMethod(ctClass, method); 
} 

Если да, то я создаю (и, таким образом переопределить) метод GetConnection и делегат суперкласса.

private CtMethod overrideMethod(CtClass ctClass, CtMethod getConnectionMethodOfSuperclass) 
     throws NotFoundException, CannotCompileException { 
    final CtMethod m = CtNewMethod.delegator(getConnectionMethodOfSuperclass, ctClass); 
    ctClass.addMethod(m); 
    return m; 
} 

Это не идеальное решение, но оно отлично работает.

Answer 2

Вы не перекрывая метод GetConnection() в родительском классе, этот метод имеет этот метод подписи:

public Connection getConnection() throws SQLException 

В вашем классе ребенка вам придется переопределить этот метод с тем же сигнатурой метода, иначе вы просто создаете новый метод в дочернем классе