/**
* PreparedStatement用にパターン別のコードを埋め込む
*
* @param ctClass クラス
* @param method メソッド
* @param bindValCode バインド変数取得用コード
* @param explainCodeTemplate 実行計画取得用コード
* @throws CannotCompileException コード埋め込みに失敗した場合
*/
public static void convertPreparedMethod(
final CtClass ctClass,
final CtBehavior method,
final String bindValCode,
final String explainCodeTemplate)
throws CannotCompileException {
String className = ctClass.getName();
String methodName = method.getName();
className = className.substring(className.lastIndexOf('.') + 1);
// setメソッドの引数の1番目がintのときのみ、実行計画取得用処理、バインド引数取得処理を追加する
try {
CtClass[] paramTypes;
paramTypes = method.getParameterTypes();
if (paramTypes.length >= 1 && "int".equals(paramTypes[0].getName())) {
// 実行計画取得用PreparedStatementのsetXXXを適したメソッド名に変更する
String explainCode =
explainCodeTemplate.replaceAll(REPLACETARGET_OF_PLANPREPARED, methodName);
// 前処理を埋め込む
String key = "javelin.jdbc.instrument.JdbcJavelinConverter.ModifiedMethodLabel";
String message = JdbcJavelinMessages.getMessage(key, className, methodName);
SystemLogger.getInstance().info(message);
method.insertBefore(bindValCode + explainCode);
}
} catch (NotFoundException e) {
SystemLogger.getInstance().warn(e);
}
}
/**
* 指定したクラスが指定したインターフェースを持つか判定する
*
* @param targetClass 判定する対象クラス
* @param interfaceClass 確認するインターフェース
* @return 指定したインターフェースを持つときtrue/そうでないときfalse
*/
public static boolean hasInterface(CtClass targetClass, CtClass interfaceClass) {
boolean hasInterface = false;
try {
for (CtClass stmtInterface : targetClass.getInterfaces()) {
if (stmtInterface.equals(interfaceClass)) {
hasInterface = true;
break;
}
}
} catch (NotFoundException ex) {
// 何もしない。
SystemLogger.getInstance().warn(ex);
}
return hasInterface;
}
@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());
}
/**
* PreparedStatement#close用にコードを埋め込む
*
* @param ctClass PreparedStatementを実装するクラス
* @param method closeメソッド
* @throws CannotCompileException コード埋め込みに失敗した場合
*/
public static void convertPreparedMethodClose(final CtClass ctClass, final CtBehavior method)
throws CannotCompileException {
String className = ctClass.getName();
className = className.substring(className.lastIndexOf('.') + 1);
// 前処理を埋め込む
String key = "javelin.jdbc.instrument.JdbcJavelinConverter.ModifiedMethodLabel";
String message = JdbcJavelinMessages.getMessage(key, className, method.getName());
SystemLogger.getInstance().info(message);
method.insertAfter(BCI_METHOD_PLANFORPREPARED_CLOSE);
}
/**
* 指定したクラスに指定したメソッドが存在するかどうかを調べます。
*
* @param targetClass メソッドの存在を調べるクラス
* @param methodName メソッド名
* @return メソッドが存在する場合は <code>true</code> 、存在しない場合は <code>false</code>
*/
public static boolean hasBehavior(final CtClass targetClass, final String methodName) {
boolean method = false;
CtBehavior[] declaredBehaviors = targetClass.getDeclaredBehaviors();
for (CtBehavior behavior : declaredBehaviors) {
String behaviorName = behavior.getName();
if (methodName.equals(behaviorName)) {
method = true;
break;
}
}
return method;
}
/**
* PreparedStatementのexecuteメソッドに、 バインド引数保存用ArrayList初期化処理を追加する。
*
* @param ctClass 変換対象のクラス。
* @param method メソッド。
* @throws CannotCompileException javassistがコンパイルに失敗した場合。
*/
private static void convertExecuteMethod(final CtClass ctClass, final CtBehavior method)
throws CannotCompileException {
String className = ctClass.getName();
className = className.substring(className.lastIndexOf('.') + 1);
// 前処理を埋め込む
String key = "javelin.jdbc.instrument.JdbcJavelinConverter.ModifiedMethodLabel";
String message = JdbcJavelinMessages.getMessage(key, className, method.getName());
SystemLogger.getInstance().info(message);
method.insertAfter(
"this.jdbcJavelinBindValIndex_ = 0;" + "this.flagForPlanStmt_ = false;", true);
}
private static void addSqlToFieldCon(final CtClass ctClass, final CtBehavior method)
throws CannotCompileException {
String addSqlCode =
JdbcJavelinRecorder.class.getName()
+ ".postPrepareStatement($0, $1, $_, \""
+ method.getName()
+ "\");";
method.insertAfter(addSqlCode);
String key = "javelin.jdbc.instrument.JdbcJavelinConverter.ModifiedMethodLabel";
String message = JdbcJavelinMessages.getMessage(key, ctClass.getName(), method.getName());
SystemLogger.getInstance().info(message);
if (SystemLogger.getInstance().isDebugEnabled()) {
String tegKey = "javelin.jdbc.instrument.JdbcJavelinConverter.JDBCJavelinTag";
String jdbcJavelinTag = JdbcJavelinMessages.getMessage(tegKey);
String messageKey = "javelin.jdbc.instrument.JdbcJavelinConverter.SQLAddedLabel1";
String logMessage = JdbcJavelinMessages.getMessage(messageKey, method.getName());
SystemLogger.getInstance().debug(jdbcJavelinTag + logMessage);
}
}
/**
* 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;
}
}