天天看點

初探BTrace腳本 - 看看持續內建哪個環節比較慢

需求

    這幾天發現持續內建環境的執行效率比較低,是以想到用BTrace來看看是什麼原因造成的,哪些環節比較慢。

    持續內建環境的服務端是hudson,裡面跑的主要是各個應用的單元測試,由maven管理執行。

擷取資源

    從BTrace官方下載下傳它的工具包 http://kenai.com/projects/btrace/downloads/directory/releases

    我這邊下載下傳的是 release-1.2 目錄下的1.2版本。

    簡單的使用說明,可以參考使用者手冊:http://kenai.com/projects/btrace/pages/UserGuide

hudson中運作的是

    hudson中跑的是單元測試代碼内容類似如下:

package com.hugehard......test;

......
public class SomeBizServiceTest extends JTester {

......

    @Test
    public void testXX() {
        ...
    }

    @Test
    public void testOO() {
        ...
    }
    ...
}
           

    是以我們需要監控的是單元測試類,這些類處于名稱包含hugehard的包路徑中。

    我們希望知道testXX, testOO這樣的方法的執行時間。

編寫監控腳本

package hudson;

import static com.sun.btrace.BTraceUtils.print;
import static com.sun.btrace.BTraceUtils.println;

import com.sun.btrace.BTraceUtils.Time;
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;

/**
 * BtraceHudson.java
 * 
 * @author caesar 2011-7-30 AM11:23:17
 */
@BTrace
public class BtraceHudson {

    @TLS
    static long starttime = 0;

    @OnMethod(clazz = "/.*hugehard.*/", method = "/test.*/")
    public static void startMethod(@ProbeClassName String probeClass, 
                                                 @ProbeMethodName String probeMethodName) {
        print("probeClass: ");
        print(probeClass);
        print("  ");
        print("probeMethodName: ");
        print(probeMethodName);
        print("  start time: ");
        starttime = Time.nanos();
        println(starttime);
    }

    @OnMethod(clazz = "/.*hugehard.*/", method = "/test.*/",  
                        location = @Location(Kind.RETURN))
    public static void collectTestMethodExecutionEnd(@ProbeClassName String probeClass,
                                                     @ProbeMethodName String probeMethodName) {
        print("probeClass: ");
        print(probeClass);
        print("  ");
        print("probeMethodName: ");
        print(probeMethodName);
        print("  execute time: ");
        long executeTime = Time.nanos() - starttime;
        println(executeTime);
    }
}
           

    上面這段代碼是參考博文 http://jarit.iteye.com/blog/1010908 中的内容,是以這裡就不贅述它的原理了。

maven不能btrace?

    我在自己的工程根目錄下,執行“mvn clean test”。

    通過“jps -l”拿到maven的PID。

    而後通過“btrace <PID> BtraceHudson.java”,期望能夠獲得每個測試方法的執行時間。

    但是沒有獲得任何的資訊。

    後來把标注 @OnMethod 中的clazz和method限定都放到最寬泛的程度,才發現,裡面拿到的方法執行資訊,根本與單元測試類毫無關聯,而都是來自maven的代碼類。

    這時候想起來,maven在執行每個工程的單元測試的時候,它會建立獨立的線程去執行它們pom.xml中定義的内容。

    是以我這裡拿到PID,僅僅是maven自身的線程資訊,根本拿不到具體工程的線程中的方法執行資訊。

javaagent模式

    根目錄的pom.xml中我們定義了jmockit的javaagent,用于執行test階段的單元測試mock代碼之用。配置如下:

<project>
    <dependencies>
          ......
    </dependencies>

    <build>
      <plugins>
        <plugin>
          <groupId>org.apache.maven.plugins</groupId>
          <artifactId>maven-surefire-plugin</artifactId>
          <version>2.4.2</version>
          <configuration>
            <argLine>-javaagent:"${settings.localRepository}/com/hugehard/external/test.jmockit/0.997/test.jmockit-0.997.jar"</argLine>
            <junitArtifactName>com.alibaba.external:test.junit</junitArtifactName>
            <testNGArtifactName>com.alibaba.external:test.testng.jdk15</testNGArtifactName>
          </configuration>
          <executions>
            <execution>
              <goals>
                <goal>test</goal>
              </goals>
            </execution>
          </executions>
        </plugin>
      </plugins>
    </build>
</project>      

    這讓我想起來,btrace也是有agent模式的,可以讓它和應用一起啟動。

    這樣每次maven建立新線程來執行單元測試的時候,會否也讓btrace一起啟動了呢?

    參考官方文檔,修改argLine配置如下:

<argLine>-javaagent:"${settings.localRepository}/com/hugehard/external/test.jmockit/0.997/test.jmockit-0.997.jar" -javaagent:/tmp/hudson-optimize/credit-shared-1/btrace-agent.jar=script=/tmp/hudson-optimize/BtraceHudson.class,scriptOutputFile=/tmp/hudson-optimize/btrace.${pom.artifactId}</argLine>      

    這裡需要說明一下:

1、javaagent模式下,需要預編譯btrace類

(詳細請參考 http://kenai.com/projects/btrace/pages/UserGuide#precompile)

  是以我們需要用如下指令先編譯咱們的BtraceHudson.java

/work/ide/btrace/btrace-bin/bin/btracec -cp . BtraceHudson.java      

而後會得到編譯好的類 BtraceHudson.class。

2、javaagent中,請確定給出的btrace-agent.jar所在的目錄中也包括了btrace-boot.jar, btrace-client.jar這兩個包。

3、我們可以在scriptOutputFile中指定btrace的輸出結果重定向到我們希望拿到報告的位置。

4、另外,這裡使用的${pom.artifactId} 是為了每個pom.xml對應的工程在執行單元測試的時候,它們所生成的btrace報告不會互相覆寫。

接着像我們往常一樣,執行“mvn clean test”指令,就能在咱們指定的目錄中檢視btrace報告,而後對它進行分析,找出執行效率低下的單元測試了。

== 全文完 ==