3

好的，我想做的是做一个可以监视应用程序的 java 代理。所以，我试图在 PreparedStatements 中注入代码来测量 SQL 查询的执行时间。为此，我开发了一个实现 ClassFileTransformer 的类。它看起来像这样：

public class JDBCClassTransformer implements ClassFileTransformer {
    @Override
    public byte[] transform(ClassLoader loader, String className, Class classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {

        ClassPool pool = ClassPool.getDefault();
        CtClass currentClass = null;
        CtClass statement = null;
        try {
            currentClass = pool.makeClass(new java.io.ByteArrayInputStream(classfileBuffer));
            statement = pool.get("java.sql.PreparedStatement");
            if (currentClass.subtypeOf(statement) && !currentClass.isInterface()) {
                probeStatement(currentClass);
            }
            classfileBuffer = currentClass.toBytecode();
        } catch (Exception e) {
            e.printStackTrace();

        } finally {
            if (currentClass != null) {
                currentClass.detach();
            }
        }
        return classfileBuffer;

    }

    private void probeStatement(CtClass currentClass) throws NotFoundException, CannotCompileException {
        CtField field = CtField.make("private fr.mael.package.SQLProbe $$probe;", currentClass);
        currentClass.addField(field);
        CtMethod setter = CtNewMethod.make("public void set$$probe(fr.mael.package.SQLProbe probe){ this.$$probe = probe;}", currentClass);
        currentClass.addMethod(setter);

        CtMethod executeQuery = currentClass.getMethod("executeQuery", "()Ljava/sql/ResultSet;");

        executeQuery.insertBefore("this.$$probe.start();");
        executeQuery.insertAfter("this.$$probe.stop();");
    }

}

所以，我想做的是在实现 java.sql.PreparedStatement 的每个类中（在“executeQuery()”方法中）注入代码。为了测试它，我只是在 MySQL 数据库上运行一个基本的“select * from a_table”，代理连接到 JVM。但我收到以下异常：

javassist.CannotCompileException: [source error] no such field: $$probe
    at javassist.CtBehavior.insertBefore(CtBehavior.java:725)
    at javassist.CtBehavior.insertBefore(CtBehavior.java:685)

它发生就行了executeQuery.insertBefore("this.$$probe.start();");。奇怪的是，每次执行“probeStatement”方法时都不会发生这种情况：首先为类调用该方法com.mysql.jdbc.PreparedStatement，并且没有抛出异常。然后，为类调用该方法com.mysql.jdbc.ServerPreparedStatement。抛出异常。该方法也被类调用，com.mysql.jdbc.JDBC4PreparedStatement异常也被抛出。两者都ServerPreparedStatement和JDBC4PreparedStatement扩展com.mysql.jdbc.PreparedStatement所以也许它是某种相关的......

我是 javassist 和 java 代理的新手，所以答案可能很明显，但我不明白为什么会抛出这个异常。

4

1 回答 1

3

我认为问题是这样的：您在 currentClass
中添加字段（您的，不是直接实现）但您是覆盖方法（来自 javadoc：CtClass.getMethod() - 返回的方法可能在超类中声明。）并且您没有t 注入字段，这可能会导致编译错误异常（当您检测时不会抛出错误，因为您将字段注入实现类）。 你遇到这种情况（只是一个想法）：$$probepublic class com.mysql.jdbc.ServerPreparedStatement extends com.mysql.jdbc.PreparedStatementjava.sql.PreparedStatementcom.mysql.jdbc.PreparedStatement.executeQuery()com.mysql.jdbc.PreparedStatementcom.mysql.jdbc.PreparedStatementjava.sql.PreparedStatement.executeQuery()

class com.mysql.jdbc.ServerPreparedStatement$$Probed {
  private fr.mael.package.SQLProbe $$probe;

  public void set$$probe(fr.mael.package.SQLProbe probe){ this.$$probe = probe;}
}
class com.mysql.jdbc.PreparedStatement$$Probed {
  public java.sql.ResultSet executeQuery() {
    // this.$$probe in this class doesn't exists! 
    this.$$probe.start();
    ...
    this.$$probe.end();
  }
}

从 currentClass 中提取第一个实现 (or @Override)的超类executeQuery()并注入$$probe到@Override executeQuery()当前类中

敌人例如，executeQuery()在 currentClass 中覆盖你将拥有

class com.mysql.jdbc.ServerPreparedStatement$$Probed {
  private fr.mael.package.SQLProbe $$probe;

  public void set$$probe(fr.mael.package.SQLProbe probe){ this.$$probe = probe;}

  @Override
  public java.sql.ResultSet executeQuery() {
    this.$$probe.start();
    super.executeQuery();
    this.$$probe.end();
  }
}

我希望清楚，英语不是我的母语。

于 2013-08-22T22:23:37.077 回答