package com.xdc.basic.api.btrace;

import static com.sun.btrace.BTraceUtils.str;

import java.sql.Connection;

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.OnEvent;

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.Self;

@BTrace(unsafe = true)

public class DataSourceTracer

{

private static Map<Object, Object> 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);

}

}

package com.xdc.basic.api.btrace;

import java.util.Random;

public class HelloWorld

{

public static void main(String[] args) throws Exception

{

while (true)

{

Random random = new Random();

execute(random.nextInt(4000));

}

}

public static void execute(int sleepTime)

{

System.out.println("Start to sleep, sleep time is=>" + sleepTime);

try

{

Thread.sleep(sleepTime);

}

catch (InterruptedException e)

{

e.printStackTrace();

}

System.out.println("End to sleep, sleep time is=>" + sleepTime);

}

}

package com.xdc.basic.api.btrace;

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.TLS;

@BTrace(unsafe = true)

public class HelloWorldTracer

{

// 注意：

// 这个方法输出在Btrace上： BTraceUtils.println("Log in Btrace start-------------------------------------------");

// 这个方法输出在java程序上：System.out.println("Log in target java start-------------------------------------------");

@TLS

private static long startTime = 0;

@OnMethod(clazz = "com.xdc.basic.api.btrace.HelloWorld", method = "execute")

public static void startMethod()

{

BTraceUtils.println("Log in Btrace start-------------------------------------------");

System.out.println("Log in target java start-------------------------------------------");

startTime = BTraceUtils.timeMillis();

}

@OnMethod(clazz = "com.xdc.basic.api.btrace.HelloWorld", method = "execute", location = @Location(Kind.RETURN))

public static void endMethod()

{

BTraceUtils.println(BTraceUtils.strcat("the class method execute time=>",

BTraceUtils.str(BTraceUtils.timeMillis() - startTime)));

System.out.println("Log in target java end-------------------------------------------");

BTraceUtils.println("Log in Btrace end-------------------------------------------");

BTraceUtils.println();

}

@OnMethod(clazz = "com.xdc.basic.api.btrace.HelloWorld", method = "execute", location = @Location(Kind.RETURN))

public static void traceExecute(@ProbeClassName String name, @ProbeMethodName String method, int sleepTime)

{

BTraceUtils.println(BTraceUtils.strcat("the class name=>", name));

BTraceUtils.println(BTraceUtils.strcat("the class method=>", method));

BTraceUtils.println(BTraceUtils.strcat("the class method params=>", BTraceUtils.str(sleepTime)));

}

}

介绍

Btrace (Byte Trace)是sun推出的一款java 动态、安全追踪（监控）工具，可以不停机的情况下监控线上情况，并且做到最少的侵入，占用最少的系统资源。

In a way, BTrace scripts are very similar to AOP's aspects, but can be attached to any existing Java code (or better, bytecode) at runtime and without any configuration or modification at development time.

背景

在生产环境中可能经常遇到各种问题，定位问题需要获取程序运行时的数据信息，如方法参数、返回值、全局变量、堆栈信息等。为了获取这些数据信息，我们可以 通过改写代码，增加日志信息的打印，再发布到生产环境。通过这种方式，一方面将增大定位问题的成本和周期，对于紧急问题无法做到及时响应；另一方面重新部 署后环境可能已被破坏，很难重新问题的场景。

BTrace功能

BTrace天生就为解决这类问题而来，它可以动态地跟踪java运行程序。通过hotswap技术，动态将跟踪字节码注入到运行类中，对运行代码侵入较小，对性能上的影响可以忽略不计。

BTrace在使用上有很多限制条件，如不能创建对象、数组、抛出和捕获异常、循环等，具体限制条件参考用户文档中的BTrace Restrictions。用户文档地址: http://kenai.com/projects/btrace/pages/UserGuide。

根据官方声明，不当地使用btrace可能导致jvm崩溃，如BTrace使用错误的.class文件，Hotspot JVM自身存在的hotswap bug等。可以先在本地验证BTrace脚本的正确性，再传到生产环境中定位问题。

工具的使用：存在两种使用方式

(1)单独的工具包：解压btrace-bin.zip即可用

btrace.bat <java_pid> HelloWorldTracer.java

(2)jvisualvm插件：安装指导：jvisualvm--工具--插件--可用插件--BTrace Workbench

右键左侧的应该java程序列表--Trace application...--打开之前编写好的java脚本。

常见错误：

/HelloWorldTracer.java:21:method calls are not allowed - only calls to BTraceUtils are allowed

这个错误说的是在BTrace脚本中只能使用BTraceUtils中的方法，这个主要是尽量减少对被调试的java程序的影响。

如果是开发环境而非生产环境，可以关闭安全校验，BTrace的功能将大大增强。

总共需要注意的是三点：

1、修改bin\btrace.bat：-Dcom.sun.btrace.unsafe=true 注意：修改只针对初次运行生效，二次运行请重启pid对应服务。因内存中仍为上次值，不推荐此方法。

2、修改脚本注解：@BTrace(unsafe=true)