Пример #1
0
  /**
   * ログ取得コードをThrowableのcatch節として追加する
   *
   * @param pool Statementを含むプール
   * @param ctClass PreparedStatementを実装するクラス
   * @param behaviour behaviour
   * @throws CannotCompileException コンパイルができないときの例外
   */
  public static void convertCatch(
      final ClassPool pool, final CtClass ctClass, final CtBehavior behaviour)
      throws CannotCompileException {
    try {
      CtClass throwable = pool.get("java.lang.Throwable");

      // メソッドの定義がない場合は実行しない。
      final int MODIFIER = behaviour.getModifiers();
      if (Modifier.isAbstract(MODIFIER)) {
        return;
      }

      // JavelinLogger#writeExceptionLogの呼び出しコードを作成する。
      StringBuffer code = new StringBuffer();

      // 後処理(例外場合)
      code.append(JAVELIN_RECORDER_NAME);
      code.append(".postProcessNG(");
      code.append("$e");
      code.append(");");

      // 例外を再throwする。
      code.append("throw $e;");

      // ログ取得コードをThrowableのcatch節として追加する。
      behaviour.addCatch(code.toString(), throwable);
    } catch (NotFoundException nfe) {
      SystemLogger.getInstance().warn(nfe);
    }
  }
Пример #2
0
  /**
   * Statementのメソッドに対して計測コードを埋め込む。
   *
   * @param pool Statementを含むプール。
   * @param ctClass Statementを実装するクラス
   * @return 変換結果
   */
  public static CtClass convertConnection(final ClassPool pool, final CtClass ctClass) {
    CtClass jvnConnction;
    try {
      jvnConnction = pool.get(JdbcJavelinConnection.class.getCanonicalName());
      boolean hasInterface = hasInterface(ctClass, jvnConnction);
      if (hasInterface == false) {
        ctClass.addInterface(jvnConnction);
      }

      // すでにメソッドを追加している場合は処理を行わない
      if (hasBehavior(ctClass, "getJdbcJavelinProcessor")) {
        return ctClass;
      }

      CtField procField =
          CtField.make(
              "private " + DBProcessor.class.getCanonicalName() + " dbProcessor_;", ctClass);
      ctClass.addField(procField);
      CtMethod procGetMethod =
          CtMethod.make(
              "public "
                  + DBProcessor.class.getCanonicalName()
                  + " getJdbcJavelinProcessor(){ return dbProcessor_; }",
              ctClass);
      ctClass.addMethod(procGetMethod);
      CtMethod procSetMethod =
          CtMethod.make(
              "public void setJdbcJavelinProcessor("
                  + DBProcessor.class.getCanonicalName()
                  + " dbProcessor){ dbProcessor_ = dbProcessor; }",
              ctClass);
      ctClass.addMethod(procSetMethod);

      CtField urlField = CtField.make("private String jdbcUrl_;", ctClass);
      ctClass.addField(urlField);
      CtMethod urlMethod = CtMethod.make("public String getJdbcUrl(){ return jdbcUrl_; }", ctClass);
      ctClass.addMethod(urlMethod);
      CtMethod urlSetMethod =
          CtMethod.make(
              "public void " + "setJdbcUrl(String jdbcUrl)" + "{ jdbcUrl_ = jdbcUrl; }", ctClass);
      ctClass.addMethod(urlSetMethod);
    } catch (NotFoundException ex) {
      SystemLogger.getInstance().warn(ex);
      return null;
    } catch (CannotCompileException ex) {
      SystemLogger.getInstance().warn(ex);
      return null;
    }

    CtBehavior[] behaviors = ctClass.getDeclaredBehaviors();
    for (int index = 0; index < behaviors.length; index++) {
      CtBehavior method = behaviors[index];
      // メソッドの定義がない場合、あるいはpublicでない
      // (->インターフェースに定義されていない)場合は実行しない。
      final int MODIFIER = method.getModifiers();
      if (Modifier.isAbstract(MODIFIER) || !Modifier.isPublic(MODIFIER)) {
        continue;
      }
      // BCI対象クラス「java.sql.Connection」に対して、コード転換を行う
      String methodName = method.getName();
      try {
        if ("prepareStatement".equals(methodName)) {
          addSqlToFieldCon(ctClass, method);
        } else if ("prepareCall".equals(methodName)) {
          addSqlToFieldCon(ctClass, method);
        }
      } catch (CannotCompileException ex) {
        SystemLogger.getInstance().warn(ex);
      }
    }

    try {
      return ctClass;
    } catch (Exception ex) {
      SystemLogger.getInstance().warn(ex);
      return null;
    }
  }