Пример #1
0
  @Override
  public void convertImpl() throws CannotCompileException, NotFoundException, IOException {
    CtClass ctClass = getCtClass();
    ClassPool pool = getClassPool();

    CtClass jobInfoIf = pool.get(MapReduceTaskAccessor.class.getName());
    ctClass.addInterface(jobInfoIf);
    CtField mapReduceJobId = CtField.make("java.lang.String mapReduceJobId_;", ctClass);
    ctClass.addField(mapReduceJobId);
    CtField mapReduceTaskIdMap =
        CtField.make(
            "java.util.Map mapReduceTaskIdMap_ = "
                + "new "
                + ConcurrentHashMap.class.getName()
                + "();",
            ctClass);
    ctClass.addField(mapReduceTaskIdMap);
    CtMethod getJobId =
        CtNewMethod.make("public String getJobId() {return mapReduceJobId_;}", ctClass);
    ctClass.addMethod(getJobId);
    CtMethod setJobId =
        CtNewMethod.make("public void setJobId(String jobId){mapReduceJobId_ = jobId;}", ctClass);
    ctClass.addMethod(setJobId);
    CtMethod getMapperName =
        CtNewMethod.make(
            "public String getMapperName(){" + "return mapper.getClass().getName();}", ctClass);
    ctClass.addMethod(getMapperName);
    CtMethod putTaskId =
        CtNewMethod.make(
            "public void putTaskId(String address, String taskId){"
                + "mapReduceTaskIdMap_.put(address, taskId);}",
            ctClass);
    ctClass.addMethod(putTaskId);
    CtMethod getSizeOfTaskIdMap =
        CtNewMethod.make(
            "public int getSizeOfTaskIdMap(){" + "return mapReduceTaskIdMap_.size();}", ctClass);
    ctClass.addMethod(getSizeOfTaskIdMap);

    List<CtBehavior> behaviorList = getMatcheDeclaredBehavior();
    for (CtBehavior ctBehavior : behaviorList) {
      convertMethod(ctBehavior);
    }

    setNewClassfileBuffer(ctClass.toBytecode());
  }
Пример #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;
    }
  }