转载

冷门instrument包，功能d炸天

文中代码示例工程如下，更多参考btrace和arthas：

https://github.com/sayhiai/example-javaagent

原创文章，转载注明出处 (http://sayhiai.com)

5版本以后，jdk有一个包叫做 instrument ，能够实现一些 非常酷 的功能。市面上一些 APM 工具，就是通过它来进行的增强。

这是基础架构的必备技能，但对业务开发来说并不是。许多面试会问到这个知识点，并不是因为将来会用到，而是因为你说对 jdk 比较熟悉，他想杀杀你的威风。

不会用没问题，但你要说不知道，就过分了点。

javaagent介绍

我们通常的java入口都是一个 main 方法，而 javaagent 的入口方法叫做 premain ，表明是在main运行之前的一些操作。javaagent就是一个jar包，定义了一个标准的premain()方法，并不需要继承或者实现任何其他的类。

这是一个约定，并木有什么其他的理由。这个方法，无论是第一次加载，还是每次新的ClassLoader加载，都会执行。

我们可以在这个前置的方法里，对字节码进行一些修改，来增加功能或者改变代码的行为。这种方法没有侵入性，只需要在启动命令中加上-javaagent参数就可以。Java6以后，甚至可以通过attach的方式，动态的给运行中的程序设置加载代理类。

有经验的同学肯定要提出异议了。其实，instrument有两个main方法，一个是 premain ，一个是 agentmain ，在一个JVM中，只会调用一个；前者是main执行之前的修改，后者控制类运行时的行为。它们还是有一些区别的，agentmain因为比较危险，限制会更大一些。

有什么用

获取统计信息

许多apm产品，比如Pinpoint、SkyWalking等，就是使用javaagent对代码进行的增强。通过在方法执行前后动态加入的统计代码，进行监控信息的收集；通过兼容OpenTracing协议，可以实现分布式链路追踪的功能。

它的原理类似于aop，最终以字节码存在，性能损失取决于你的代码逻辑。

热部署

通过自定义的ClassLoader，可以实现代码的热替换。使用agentmain，实现热部署功能会更加便捷。通过agentmain获取到Instrumentation以后，就可以对类进行动态重定义。

诊断

配合 JVMTI 技术，可以 attach 到某个进程进行运行时统计和调试，比较流行的 btrace 和 arthas ，底层就是这种技术。

如何做

大体分为以下步骤：

构建agent jar包，编写增强代码
在manifest中指定Premain-Class/Agent-Class属性
使用参数加载或者attach方式使用

编写Agent

javaagent最终的体现方式是一个jar包。使用idea创建一个默认的maven工程即可。

创建一个普通java类，添加 premain 或者 agentmain 方法，它们的参数完全一样。

冷门instrument包，功能d炸天

编写Transformer

此部分，要借助额外jar包的功能。

实际的代码逻辑需要实现 ClassFileTransformer 接口。假如我们要统计某个方法的执行时间。我们使用 javaassist 来增强字节码，则可以通过以下代码来实现。

MainRun
hello
_begin

别忘了加入maven依赖

<dependency>
    <groupId>org.javassist</groupId>
    <artifactId>javassist</artifactId>
    <version>3.24.1-GA</version>
</dependency>

冷门instrument包，功能d炸天字节码增强也可以使用Cglib、asm等其他工具。

MANIFEST.MF文件

那么我们编写的代码是如何让外界知晓呢？那就是 MANIFEST.MF 文件。具体路径在

src/main/resources/META-INF/MANIFEST.MF

Manifest-Version: 1.0
premain-class: com.sayhiai.example.javaagent.AgentApp

一般的，maven打包会覆盖这个文件，所以我们需要指定需要哪一个。

<build><plugins><plugin>
<groupId>org.apache.maven.plugins</groupId>
    <artifactId>maven-jar-plugin</artifactId>
    <configuration>
        <archive>
            <manifestFile>src/main/resources/META-INF/MANIFEST.MF</manifestFile>
            </archive>
    </configuration></plugin></plugins></build>

然后，在命令行，执行 mvn install 安装到本地代码库，或者使用 mvn deploy 发布到私服上。

附，MANIFEST.MF参数清单：

Premain-Class

Agent-Class

Boot-Class-Path

Can-Redefine-Classes

Can-Retransform-Classes

Can-Set-Native-Method-Prefix