import static com.sun.btrace.BTraceUtils.println;
import static com.sun.btrace.BTraceUtils.str;
import static com.sun.btrace.BTraceUtils.strcat;
import static com.sun.btrace.BTraceUtils.timeMillis;
import java.sql.Connection;
import java.util.HashMap;
import java.util.Map;
import com.sun.btrace.BTraceUtils;
import com.sun.btrace.annotations.BTrace;
import com.sun.btrace.annotations.Kind;
import com.sun.btrace.annotations.Location;
import com.sun.btrace.annotations.OnMethod;
import com.sun.btrace.annotations.ProbeClassName;
import com.sun.btrace.annotations.ProbeMethodName;
import com.sun.btrace.annotations.Return;
import com.sun.btrace.annotations.OnEvent;
import com.sun.btrace.annotations.Self;
import com.sun.btrace.annotations.TLS;
import java.lang.Appendable;
@BTrace public class DataSourceTrace {
private static Map map = BTraceUtils.newHashMap();
@OnMethod(clazz = "org.apache.commons.dbcp.BasicDataSource", method = "getConnection", location = @Location(Kind.RETURN))
public static void traceExecute(@ProbeClassName String name,@ProbeMethodName String method, @Return Connection conn) {
//BTraceUtils.println(strcat("获取连接:",BTraceUtils.str(conn)));
Appendable buffer=BTraceUtils.Strings.newStringBuilder();
BTraceUtils.Strings.append(buffer,BTraceUtils.timestamp("yyyy-MM-dd HH:mm:ss"));
BTraceUtils.Strings.append(buffer," - ");
BTraceUtils.Strings.append(buffer,BTraceUtils.jstackStr());
BTraceUtils.put(map, conn,str(buffer));
}
@OnMethod(clazz = "org.apache.commons.dbcp.PoolingDataSource$PoolGuardConnectionWrapper", method = "close")
public static void traceExecute2(@ProbeClassName String name,@ProbeMethodName String method, @Self Connection conn) {
//BTraceUtils.println(strcat("释放连接:",BTraceUtils.str(conn)));
BTraceUtils.remove(map, conn);
}
@OnEvent
public static void exit(){
//这里打印泄漏连接的方法堆栈，运行一段时间后通过Ctrl+C,选择2，发送事件打印
BTraceUtils.printMap(map);
}
}