/**
* 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 ctMethod 埋め込み対象のメソッド
* @throws CannotCompileException
*/
private void insertProcessesForJob(final CtBehavior ctMethod) throws CannotCompileException {
ctMethod.insertBefore(MapReduceTaskMonitor.class.getName() + ".preProcess(this);");
ctMethod.insertAfter(MapReduceTaskMonitor.class.getName() + ".postProcess(this);");
ctMethod.addCatch(
"{"
+ MapReduceTaskMonitor.class.getName()
+ ".postProcessNG(this, $e); $e.printStackTrace(); throw $e;}",
throwableClass_);
}
/**
* StatsJavelinによる計測コードを埋め込む。
*
* @param ctClass クラス
* @param method メソッド
* @throws CannotCompileException メソッドの開始位置と終了位置にコードを埋め込めなかった場合
*/
public static void addRecordCode(final CtClass ctClass, final CtBehavior method)
throws CannotCompileException {
// 前処理ログコードを作る
StringBuffer callPreProcessCodeBuffer = new StringBuffer();
callPreProcessCodeBuffer.append(JAVELIN_RECORDER_NAME);
CtClass[] parameterTypes = null;
boolean paramZero = false;
try {
parameterTypes = method.getParameterTypes();
} catch (NotFoundException ex) {
SystemLogger.getInstance().warn("", ex);
}
if (parameterTypes != null && (parameterTypes.length > 0)) {
callPreProcessCodeBuffer.append(".preProcessParam(");
callPreProcessCodeBuffer.append("$0");
callPreProcessCodeBuffer.append(", $args);");
} else {
callPreProcessCodeBuffer.append(".preProcessSQLArgs(");
callPreProcessCodeBuffer.append("$0");
callPreProcessCodeBuffer.append(", this.getJdbcJavelinSql().toArray());");
paramZero = true;
}
String callPreProcessCode;
callPreProcessCode = callPreProcessCodeBuffer.toString();
// 後処理ログコードを作る
StringBuffer callPostProcessCodeBuffer = new StringBuffer();
callPostProcessCodeBuffer.append(JAVELIN_RECORDER_NAME);
// Recorderにて実行計画取得にStatemen、クラス名、メソッド名が必要
callPostProcessCodeBuffer.append(".postProcessOK($0");
if (paramZero == false) { // 引数の数をpostProcessOKに渡す
callPostProcessCodeBuffer.append(", 1"); // 引数1以上
} else {
callPostProcessCodeBuffer.append(", 0"); // 引数0
}
callPostProcessCodeBuffer.append(");");
String returnPostProcessCode = callPostProcessCodeBuffer.toString();
// JavelinErrorLogger.getInstance().log("modified class:" + className);
method.insertBefore(callPreProcessCode);
method.insertAfter(returnPostProcessCode);
}