ToB企服应用市场:ToB评测及商务社交产业平台

标题: Skywalking光会用可不行,必须的源码分析分析 - Skywalking Agent &插件解 [打印本页]

作者: 南飓风    时间: 2022-6-23 13:14
标题: Skywalking光会用可不行,必须的源码分析分析 - Skywalking Agent &插件解
3 Skywalking源码导入

接上文,已经学习了Skywalking的应用,接下来我们将剖析Skywalking源码,深度学习Skywalking Agent。
3.1 源码环境搭建

当前最新版本是8.3.0,我们首先找到8.3.0的版本,然后下载并导入到IDEA,下载地址
https://github.com/apache/skywalking.git,我们直接用git克隆到本地。
1、推荐大家将github仓库拷贝到码云上,以提升下载速度
2、为了避免clone过程出错,可以设置git的全局参数:git config --global core.longpaths true避免出现Filename too long的报错信息
1)下载工程


这个过程比较耗时间,需要大家耐心等待!
2)切换版本
将Skywalking工程加入到Maven工程中,我们用的是当前最新版本8.3.0,因此需要切换版本:

项目导入IDEA后,会从指定路径加载项目,我们需要在skywalking的pom.xml中配置项目的工路径,添加如下properties配置即可:
  1. <maven.multiModuleProjectDirectory><maven.multiModuleProjectDirectory>C:\developer\WorkSpace\skywalking</maven.multiModuleProjectDirectory></maven.multiModuleProjectDirectory>
复制代码
pom中有一个插件maven-enforcer-plugin要求的maven的版本是3.6以上,需要注意!!!
我们接下来获取skywalking子模块的源码,需要在工程中执行如下命令:
  1. git submodule init
  2. git submodule update
复制代码

该步骤非常重要,不完整执行成功,后续的编译会失败。git submodule update执行很慢,还可能中途中断
编译项目,此时会生成一些类skywalking\apm-protocol\apm-network\target\generated-sources\protobuf\java\org\apache\skywalking\apm\network\common\v3目录下的类如下图:

接下来把生成的文件添加到类路径下,如下图:

除了上面这里,还有很多个地方都需要这么操作,我们执行OAPServerStartUp的main方法启动Skywalking,只要执行找不到类,就找下有没有任何编译后生成的类没在类路径下,都把他们设置为类路径即可。
安装项目,Skywalking依赖的插件特别多,因此依赖的包也特别多,我们把Skywalking安装到本地,会耗费很长时间,但不要担心,因为迟早会安装完成,如下图:

如果通过以上方式实在构建不了源码,也可尝试通过如下方式来,
官方提供了关于如何构建的步骤可以参阅:
https://github.com/apache/skywalking/blob/master/docs/en/guides/How-to-build.md
社区中文版:https://skyapm.github.io/document-cn-translation-of-skywalking/zh/8.0.0/guides/How-to-build.html
1、通过命令拉取源码
  1. git clone -b v8.3.0 --recurse-submodules https://gitee.com/giteets/skywalking.git
复制代码
构建过程中遇到的最大问题是:git submodule 子模块的源码构建不出来,整体项目拉取下来后也可通过如下命令再次拉取子模块源码
  1. git submodule init
  2. git submodule update
复制代码
如果实在不行:在项目下有个.gitmodules文件,定义了子模块的仓库地址和应该安装到什么目录下
  1. [submodule "apm-protocol/apm-network/src/main/proto"]
  2.         path = apm-protocol/apm-network/src/main/proto
  3.         url = https://github.com/apache/skywalking-data-collect-protocol.git
  4. [submodule "oap-server/server-query-plugin/query-graphql-plugin/src/main/resources/query-protocol"]
  5.         path = oap-server/server-query-plugin/query-graphql-plugin/src/main/resources/query-protocol
  6.         url = https://github.com/apache/skywalking-query-protocol.git
  7. [submodule "skywalking-ui"]
  8.         path = skywalking-ui
  9.         url = https://github.com/apache/skywalking-rocketbot-ui.git
  10. [submodule "test/e2e/e2e-protocol/src/main/proto"]
  11.         path = test/e2e/e2e-protocol/src/main/proto
  12.         url = https://github.com/apache/skywalking-data-collect-protocol.git
复制代码
实在不行,就手动将这四个子模块分别手动下载到指定的path目录下,注意版本
2、将项目导入到idea,要求jkd8,maven3.6
3、在项目的pom.xml中添加properties
  1. <maven.multiModuleProjectDirectory><maven.multiModuleProjectDirectory>C:\developer\WorkSpace\skywalking</maven.multiModuleProjectDirectory></maven.multiModuleProjectDirectory>
复制代码
4、clean,package,install,注意跳过测试
5、参考社区文档,设置idea,将生成的源代标记成Sources Root
设置 生成的源代码(Generated Source Code)目录.
3.2 模块分析

apm-application-toolkit:常用的工具工程,例如:log4j、log4j2、logback 等常见日志框架的接入接口,Kafka轮询调用注解,apm-application-toolkit 模块类似于暴露 API 定义,对应的处理逻辑在 apm-sniffer/apm-toolkit-activation 模块中实现,如下图:

apm-commons:SkyWalking 的公共组件和工具类。如下图所示,其中包含两个子模块,apm-datacarrier 模块提供了一个生产者-消费者模式的缓存组件(DataCarrier),无论是在 Agent 端还是 OAP 端都依赖该组件。apm-util 模块则提供了一些常用的工具类,例如,字符串处理工具类(StringUtil)、占位符处理的工具类(PropertyPlaceholderHelper、PlaceholderConfigurerSupport)等等。
apache-skywalking-apm:SkyWalking 打包后使用的命令文件都在此目录中,例如,前文启动 OAP 和 SkyWalking Rocketbot 使用的 startup.sh 文件。
apm-protocol:该模块中只有一个 apm-network 模块,我们需要关注的是其中定义的 .proto 文件,定义 Agent 与后端 OAP 使用 gRPC 交互时的协议。
apm-sniffer:agent核心功能以及agent依赖插件,模块比较多:
  1. apm-agent:只有一个类SkyWalkingAgent,是Skywalking的agent入口。
  2. apm-agent-core:看名字我们就知道它是Skywalking agent核心模块。
  3. apm-sdk-plugin:该模块下包含了 SkyWalking Agent 的全部插件。
  4. apm-toolkit-activation:apm-application-toolkit 模块的具体实现。
  5. apm-test-tools:Skywalking的测试功能。
  6. bootstrap-plugins:该插件主要提供了Http和多线程相关的功能支持,它里面有2个子工程。
  7. optional-plugins:可选插件,例如对spring支持、对kotlin支持等,它下面有多个插件工程实现。
  8. optional-reporter-plugins:该工程插件主要提供一些数据报告,集成了Kafka功能。
复制代码
apm-webapp:SkyWalking Rocketbot 对应的后端。
oap-server:oap 主程序,该工程中有多个模块,我们对核心模块进行说明:
  1. analyzer:数据分析工程,例如对内存分析、仪表盘分析报告等,它下面有2个子工程。
  2. exporter:导出数据功能。
  3. oal-grammar:操作适配语法,例如SQL语法。
  4. oal-rt:操作解析器,上面提供了语法,该工程提供对操作解析功能。
  5. server-alarm-plugin:负责实现 SkyWalking 的告警功能。
  6. server-cluster-plugin:OAP集群管理功能,提供了很多第三方介入的组件。
  7. server-configuration:负责管理 OAP 的配置信息,也提供了接入多种配置管理组件的相关插件。
  8. server-core:SkyWalking OAP的核心实现都在该模块中。
  9. server-library:OAP 以及 OAP 各个插件依赖的公共模块,其中提供了双队列 Buffer、请求远端的 Client 等工具类,这些模块都是对立于 SkyWalking OAP 体系之外的类库,我们可以直接拿着使用。
  10. server-query-plugin:SkyWalking Rocketbot 发送的请求首先由该模块接收处理,目前该模块只支持 GraphQL 查询。
  11. server-receiver-plugin:SkyWalking Agent 发送来的 Metrics、Trace 以及 Register 等写入请求都是首先由该模块接收处理的,不仅如此,该模块还提供了多种接收其他格式写入请求的插件。
  12. server-starter:OAP 服务启动的入口。
  13. server-storage-plugin:OAP 服务底层可以使用多种存储来保存 Metrics 数据以及Trace 数据,该模块中包含了接入相关存储的插件。
  14. skywalking-agent:SkyWalking Agent 编译后生成的 jar 包都会放到该目录中。
  15. skywalking-ui:SkyWalking Rocketbot 的前端。
复制代码
4 Skywalking Agent 启动流程剖析

我们已经学习了Skywalking常用操作,并且讲解了Java Agent,而且Skywalking Agent就是基于Java Agent研发而来,我们接下来深入学习Skywalking Agent架构、原理、常用组件。
4.1 Skywalking Agent架构

我们在学习Skywalking之前,先了解一下微内核架构,如下图:

微内核架构(Microkernel Architecture),也被成为插件化架构(Plug-in Architecture),是一种面向功能进行拆分的可扩展性架构,通常用于实现基于产品(原文为product-based,指存在多个版本,需要下载安装才能使用,与web-based想对应)的应用。
微内核架构的好处:
  1. 1:测试成本下降。从软件工程的角度看,微内核架构将变化的部分和不变的部分拆分,降低了测试的成本,符合设计模式中的开放封闭原则。
  2. 2:稳定性。由于每个插件模块相对独立,即使其中一个插件有问题,也可以保证内核系统以及其他插件的稳定性。
  3. 3:可扩展性。在增加新功能或接入新业务的时候,只需要新增相应插件模块即可;在进行历史功能下线时,也只需删除相应插件模块即可。
复制代码
微内核的核心系统设计的关键技术有:插件管理,插件连接 和 插件通信。
SkyWalking Agent 采用了微内核架构(Microkernel Architecture),是一种面向功能进行拆分的可扩展性架构。
  1. apm-agent-core:是Skywalking Agent的核心模块
  2. apm-sdk-plugin:是Skywalking需要的各个插件模块
复制代码

4.2 Skywalking Agent启动流程

1)启动OAP
我们接下来启动Skywalking oap,我们在oap-server\server-starter或者oap-server\server-starter-es7中找到OAPServerStartUp类,执行该类的main方法即可启动,但默认用的是H2存储,如果希望用elasticsearch存储,需要修改被调用的服务server-bootstrap的配置文件application.yml配置elasticsearch位置:
  1. storage:
  2.   #selector: ${SW_STORAGE:h2}
  3.   selector: ${SW_STORAGE:elasticsearch7}
  4.   elasticsearch7:
  5.     nameSpace: ${SW_NAMESPACE:""}
  6.     #clusterNodes: ${SW_STORAGE_ES_CLUSTER_NODES:localhost:9200}
  7.     clusterNodes: ${SW_STORAGE_ES_CLUSTER_NODES:192.168.211.145:9200}
  8.     ........略
复制代码
存储直接使用上一次课准备好的es7的存储即可。
执行OAPServerStartUp的main方法不报错就没问题。
2)启动SkyWalking Rocketbot
apm-webapp 是 Spring Boot 的 Web项目,执行 ApplicationStartUp  中的 main() 方法。正常启动之后,
访问 http://localhost:8080,看到 SkyWalking Rocketbot 的 UI 界面即为启动成功。
如果修改启动端口,可以直接修改application.yml即可。
3)直接使用源码中的Agent
项目打包会生成skywalking-agent目录,里面有skywalking-agent.jar,如下图:

我们来使用一下前面源码工程中打包生成的skywalking-agent.jar,复制该jar包的路径
找到hailtaxi-parent项目,修改-javaagent参数如下
hailtaxi-gateway
  1. -javaagent:C:\developer\WorkSpace\sources\skywalking\skywalking-agent\skywalking-agent.jar
  2. -Dskywalking_config=C:\developer\WorkSpace\sources\skywalking\skywalking-agent\config\agent.config
  3. -Dskywalking.agent.service_name=hailtaxi-gateway
复制代码
hailtaxi-driver和hailtaxi-order进行相同配置即可!
全都启动后,查看Skywalking Rocketbot :本地启动,需要等待一定的时间
5 Skywalking Agent源码剖析

1、创建sw-agent-debugger项目:一个普通的springboot项目即可

2、添加启动-javaagent参数


启动的整个方法执行流程如下:
  1. public static void premain(String agentArgs, Instrumentation instrumentation) throws PluginException {
  2.     final PluginFinder pluginFinder;
  3.     try {
  4.         //初始化加载 agent.config 配置文件,其中会检测 Java Agent 参数以及环境变量是否覆盖了相应配置项
  5.         SnifferConfigInitializer.initializeCoreConfig(agentArgs);
  6.     } catch (Exception e) {
  7.         // try to resolve a new logger, and use the new logger to write the error log here
  8.         LogManager.getLogger(SkyWalkingAgent.class)
  9.                 .error(e, "SkyWalking agent initialized failure. Shutting down.");
  10.         return;
  11.     } finally {
  12.         // refresh logger again after initialization finishes
  13.         LOGGER = LogManager.getLogger(SkyWalkingAgent.class);
  14.     }
  15.     try {
  16.         //管理插件
  17.         pluginFinder = new PluginFinder(new PluginBootstrap().loadPlugins());
  18.     } catch (AgentPackageNotFoundException ape) {
  19.         LOGGER.error(ape, "Locate agent.jar failure. Shutting down.");
  20.         return;
  21.     } catch (Exception e) {
  22.         LOGGER.error(e, "SkyWalking agent initialized failure. Shutting down.");
  23.         return;
  24.     }
  25.     //使用ByteBuddy创建AgentBuilder
  26.     final ByteBuddy byteBuddy = new ByteBuddy().with(TypeValidation.of(Config.Agent.IS_OPEN_DEBUGGING_CLASS));
  27.     //忽略拦截配置
  28.     AgentBuilder agentBuilder = new AgentBuilder.Default(byteBuddy).ignore(
  29.             nameStartsWith("net.bytebuddy.")
  30.                     .or(nameStartsWith("org.slf4j."))
  31.                     .or(nameStartsWith("org.groovy."))
  32.                     .or(nameContains("javassist"))
  33.                     .or(nameContains(".asm."))
  34.                     .or(nameContains(".reflectasm."))
  35.                     .or(nameStartsWith("sun.reflect"))
  36.                     .or(allSkyWalkingAgentExcludeToolkit())
  37.                     .or(ElementMatchers.isSynthetic()));
  38.     JDK9ModuleExporter.EdgeClasses edgeClasses = new JDK9ModuleExporter.EdgeClasses();
  39.     try {
  40.         agentBuilder = BootstrapInstrumentBoost.inject(pluginFinder, instrumentation, agentBuilder, edgeClasses);
  41.     } catch (Exception e) {
  42.         LOGGER.error(e, "SkyWalking agent inject bootstrap instrumentation failure. Shutting down.");
  43.         return;
  44.     }
  45.     try {
  46.         agentBuilder = JDK9ModuleExporter.openReadEdge(instrumentation, agentBuilder, edgeClasses);
  47.     } catch (Exception e) {
  48.         LOGGER.error(e, "SkyWalking agent open read edge in JDK 9+ failure. Shutting down.");
  49.         return;
  50.     }
  51.     if (Config.Agent.IS_CACHE_ENHANCED_CLASS) {
  52.         try {
  53.             agentBuilder = agentBuilder.with(new CacheableTransformerDecorator(Config.Agent.CLASS_CACHE_MODE));
  54.             LOGGER.info("SkyWalking agent class cache [{}] activated.", Config.Agent.CLASS_CACHE_MODE);
  55.         } catch (Exception e) {
  56.             LOGGER.error(e, "SkyWalking agent can't active class cache.");
  57.         }
  58.     }
  59.     //Java Agent创建代理流程
  60.     agentBuilder.type(pluginFinder.buildMatch())
  61.                 .transform(new Transformer(pluginFinder))
  62.                 .with(AgentBuilder.RedefinitionStrategy.RETRANSFORMATION)
  63.                 .with(new Listener())
  64.                 .installOn(instrumentation);
  65.     try {
  66.         //使用 JDK SPI加载的方式并启动 BootService 服务。
  67.         ServiceManager.INSTANCE.boot();
  68.     } catch (Exception e) {
  69.         LOGGER.error(e, "Skywalking agent boot failure.");
  70.     }
  71.     //添加一个JVM钩子
  72.     Runtime.getRuntime()
  73.             .addShutdownHook(new Thread(ServiceManager.INSTANCE::shutdown, "skywalking service shutdown thread"));
  74. }
复制代码
我们总结一下Skywalking Agent启动流程:
  1. 1:初始化配置信息。该步骤中会加载 agent.config 配置文件,其中会检测 Java Agent 参数以及环境变量是否覆盖了相应配置项。
  2. 2:查找并解析 skywalking-plugin.def 插件文件。
  3. 3:AgentClassLoader 加载插件。
  4. 4:PluginFinder 对插件进行分类管理。
  5. 5:使用 Byte Buddy 库创建 AgentBuilder。这里会根据已加载的插件动态增强目标类,插入埋点逻辑。
  6. 6:使用 JDK SPI 加载并启动 BootService 服务。
  7. 7:添加一个 JVM 钩子,在 JVM 退出时关闭所有 BootService 服务。
复制代码
这是org.apache.skywalking.apm.agent.SkyWalkingAgent#premain的主体工作流程
5.1 配置初始化
  1. -javaagent:D:/project/skywalking/skywalking/apm-sniffer/apm-agent/target/skywalking-agent.jar
  2. -Dskywalking_config=D:/project/skywalking/hailtaxi-parent/hailtaxi-driver/src/main/resources/agent.config
  3. -Dskywalking.collector.backend_service=127.0.0.1:11800
复制代码
启动driver服务的时候,会指定skywalking-agent.jar路径,同时会指定agent.config配置文件路径,如上配置,此时需要初始化加载该文件,加载流程可以从启动类SkyWalkingAgent.premain()方法找答案。

加载解析文件的时候,permain()方法会调用initializeCoreConfig(String agentOptions)方法,并解析agent.config文件,并将文件内容存入到Properties中,此时加载是按照${配置项名称:默认值}的格式解析各个配置,如下图:

loadConfig() 方法会优先根据环境变量(skywalking_config)指定的 agent.config 文件路径加载。若环境变量未指定 skywalking_ config 配置,则到 skywalking-agent.jar 同级的 config 目录下查找 agent.confg 配置文件。

解析前后的数据也是不一致的,如下图:

overrideConfigBySystemProp() 方法中会遍历环境变量(即 System.getProperties() 集合),如果环境变 是以 "skywalking." 开头的,则认为是 SkyWalking 的配置,同样会填充到 Config 类中,以覆盖 agent.config 中的默认值。如下图:

ConfigInitializer 工具类,将配置信息填充到 Config 中的静态字段中,SkyWalking Agent 启动所需的全部配置都已经填充到 Config 中,后续使用配置信息时直接访问 Config 中的相应静态字段即可。

Config结构:

Config中Agent类的SERVICE_NAME对应agent.config中的agent.service_name=${xxx}
Config中Collector类的BACKEND_SERVICE对应agent.config中的agent.backend_service=${xxx}
5.2 插件加载

加载插件执行流程:
  1. 1:new PluginBootstrap()
  2. 2:PluginBootstrap().loadPlugins()
  3. 3:AgentClassLoader.initDefaultLoader();        没有指定类加载器的时候使用PluginBootstrap.ClassLoader
  4. 4:创建PluginResourcesResolver插件加载解析器
  5. 5:将解析的插件存到List<PluginDefine> pluginClassList,此时只存储了插件的名字和类路径
  6. 6:创建插件实例
  7. 7:将所有插件添加到Skywalking内核中
复制代码
插件加载流程如下:
在SkyWalkingAgent.premain()方法中会执行插件加载,如下代码:
  1. pluginFinder = new PluginFinder(new PluginBootstrap().loadPlugins());
复制代码
加载插件的全部详细代码如下:
  1. public class PluginBootstrap {
  2.     private static final ILog LOGGER = LogManager.getLogger(PluginBootstrap.class);
  3.     /**
  4.      * 加载所有插件
  5.      * load all plugins.
  6.      * @return plugin definition list.
  7.      */
  8.     public List<AbstractClassEnhancePluginDefine> loadPlugins() throws AgentPackageNotFoundException {
  9.         //初始化AgentClassLoader
  10.         AgentClassLoader.initDefaultLoader();
  11.         //创建PluginResourcesResolver插件加载解析器
  12.         PluginResourcesResolver resolver = new PluginResourcesResolver();
  13.         //获取插件路径
  14.         List<URL> resources = resolver.getResources();
  15.         if (resources == null || resources.size() == 0) {
  16.             LOGGER.info("no plugin files (skywalking-plugin.def) found, continue to start application.");
  17.             return new ArrayList<AbstractClassEnhancePluginDefine>();
  18.         }
  19.         //循环加载插件路径
  20.         for (URL pluginUrl : resources) {
  21.             try {
  22.                 //插件会存到List<PluginDefine> pluginClassList,PluginDefine中只有插件名字和插件类路径
  23.                 PluginCfg.INSTANCE.load(pluginUrl.openStream());
  24.             } catch (Throwable t) {
  25.                 LOGGER.error(t, "plugin file [{}] init failure.", pluginUrl);
  26.             }
  27.         }
  28.         //获取解析的插件集合
  29.         List<PluginDefine> pluginClassList = PluginCfg.INSTANCE.getPluginClassList();
  30.         List<AbstractClassEnhancePluginDefine> plugins = new ArrayList<AbstractClassEnhancePluginDefine>();
  31.         //循环所有插件
  32.         for (PluginDefine pluginDefine : pluginClassList) {
  33.             try {
  34.                 LOGGER.debug("loading plugin class {}.", pluginDefine.getDefineClass());
  35.                 //创建插件实例(加载插件)
  36.                 AbstractClassEnhancePluginDefine plugin = (AbstractClassEnhancePluginDefine) Class.forName(pluginDefine.getDefineClass(), true, AgentClassLoader
  37.                     .getDefault()).newInstance();
  38.                 plugins.add(plugin);
  39.             } catch (Throwable t) {
  40.                 LOGGER.error(t, "load plugin [{}] failure.", pluginDefine.getDefineClass());
  41.             }
  42.         }
  43.         //将插件添加到内核中
  44.         plugins.addAll(DynamicPluginLoader.INSTANCE.load(AgentClassLoader.getDefault()));
  45.         return plugins;
  46.     }
  47. }
复制代码
SkyWalking Agent 加载插件时使用到一个自定义的 ClassLoader —— AgentClassLoader,之所以自定义类加载器,目的是不在应用的 Classpath 中引入 SkyWalking 的插件 jar 包,这样就可以让应用无依赖、无感知的插件。
AgentClassLoader 作为一个类加载器,主要工作还是从其 Classpath 下加载类(或资源文件),对应的就是其 findClass() 方法和 findResource() 方法:
我们来看一下findClass,主要根据类名获取它的Class:
  1. @Override
  2. protected Class<?> findClass(String name) throws ClassNotFoundException {
  3.     //扫描classpath所有的jar包
  4.     List<Jar> allJars = getAllJars();
  5.     //把包替换成路径,最后加上.class
  6.     String path = name.replace('.', '/').concat(".class");
  7.     //循环查找所有的jar包
  8.     for (Jar jar : allJars) {
  9.         //加载jar包的信息
  10.         JarEntry entry = jar.jarFile.getJarEntry(path);
  11.         if (entry == null) {
  12.             continue;
  13.         }
  14.         try {
  15.             //定位当前jar包位置
  16.             URL classFileUrl = new URL("jar:file:" + jar.sourceFile.getAbsolutePath() + "!/" + path);
  17.             //加载jar包
  18.             byte[] data;
  19.             try (final BufferedInputStream is = new BufferedInputStream(
  20.                 classFileUrl.openStream()); final ByteArrayOutputStream baos = new ByteArrayOutputStream()) {
  21.                 int ch;
  22.                 while ((ch = is.read()) != -1) {
  23.                     baos.write(ch);
  24.                 }
  25.                 data = baos.toByteArray();
  26.             }
  27.             //返回当前对象的Class
  28.             return processLoadedClass(defineClass(name, data, 0, data.length));
  29.         } catch (IOException e) {
  30.             LOGGER.error(e, "find class fail.");
  31.         }
  32.     }
  33.     throw new ClassNotFoundException("Can't find " + name);
  34. }
复制代码
findResource()方法主要获取文件路径,换句话理解,就是获取插件路径,我们来看下方法:
  1. @Override
  2. protected URL findResource(String name) {
  3.     //扫描classpath所有的jar包
  4.     List<Jar> allJars = getAllJars();
  5.     //循环查找所有的jar包
  6.     for (Jar jar : allJars) {
  7.         //加载jar包的信息
  8.         JarEntry entry = jar.jarFile.getJarEntry(name);
  9.         if (entry != null) {
  10.             try {
  11.                 //获取jar包的路径
  12.                 return new URL("jar:file:" + jar.sourceFile.getAbsolutePath() + "!/" + name);
  13.             } catch (MalformedURLException ignored) {
  14.             }
  15.         }
  16.     }
  17.     return null;
  18. }
复制代码
5.3 解析插件

我们在学习插件解析之前,先看看插件是如何定义的。我们可以打开apm-sniffer/apm-sdk-plugin,它里面都是要用到的插件集合:

我们看看mysql-5.x-plugin,在resources(也就是classpath)中定义skywalking-plugin.def文件,在该文件中定义加载插件需要解析的类,而插件类以key=value形式定义,如下图:

5.3.1 PluginResourcesResolver

在loadPlugins()方法中使用了PluginResourcesResolver,PluginResourcesResolver是 Agent 插件的资源解析器,会通过 AgentClassLoader 中的 findResource() 方法读取所有 Agent 插件中的 skywalking-plugin.def 文件。

拿到全部插件的 skywalking-plugin.def 文件之后,PluginCfg 会逐行进行解析,转换成 PluginDefine 对象。PluginDefine 中有两个字段,分别对应skywalking-plugin.def中的key和value,解析流程如下:

接下来会遍历全部 PluginDefine 对象,通过反射将其中 defineClass 字段中记录的插件类实例化,核心逻辑如下:

AbstractClassEnhancePluginDefine 抽象类是所有 Agent 插件类的顶级父类,其中定义了四个核心方法,决定了一个插件类应该增强哪些目标类、应该如何增强、具体插入哪些逻辑,如下所示:

ClassMatch
enhanceClass() 方法决定了一个插件类要增强的目标类,返回值为 ClassMatch 类型对象。ClassMatch 类似于一个过滤器,可以通过多种方式匹配到目标类,ClassMatch 接口的实现如下:

我们来分析一下ClassAnnotationMatch的buildJunction()方法和isMatch()方法:
  1. @Override
  2. public ElementMatcher.Junction buildJunction() {
  3.     ElementMatcher.Junction junction = null;
  4.     //annotations:指定了该 ClassAnnotationMatch 对象需要检查的注解
  5.     //遍历该对象需要检查的所有注解
  6.     for (String annotation : annotations) {
  7.         if (junction == null) {
  8.             //检测类是否标注了指定注解
  9.             junction = buildEachAnnotation(annotation);
  10.         } else {
  11.             //使用 and 方式将所有Junction对象连接起来
  12.             junction = junction.and(buildEachAnnotation(annotation));
  13.         }
  14.     }
  15.     // 排除接口
  16.     junction = junction.and(not(isInterface()));
  17.     return junction;
  18. }
复制代码
isMatch()方法如下:
  1. @Override
  2. public boolean isMatch(TypeDescription typeDescription) {
  3.     List<String> annotationList = new ArrayList<String>(Arrays.asList(annotations));
  4.     // 获取该类上的注解
  5.     AnnotationList declaredAnnotations = typeDescription.getDeclaredAnnotations();
  6.     // 匹配一个删除一个
  7.     for (AnnotationDescription annotation : declaredAnnotations) {
  8.         annotationList.remove(annotation.getAnnotationType().getActualName());
  9.     }
  10.     // 如果全部删除,则匹配成功
  11.     return annotationList.isEmpty();
  12. }
复制代码
5.3.2 PluginFinder

PluginFinder 是 AbstractClassEnhancePluginDefine 查找器,可以根据给定的类查找用于增强的 AbstractClassEnhancePluginDefine 集合。
在 PluginFinder 的构造函数中会遍历前面课程已经实例化的 AbstractClassEnhancePluginDefine ,并根据 enhanceClass() 方法返回的 ClassMatcher 类型进行分类,得到如下两个集合:
  1. //定义了集合
  2. //pluginFinder将插件分类保存在两个集合中,分别是:按名字分类和按其他辅助信息分类
  3. private final Map<String, LinkedList<AbstractClassEnhancePluginDefine>> nameMatchDefine = new HashMap<String, LinkedList<AbstractClassEnhancePluginDefine>>();
  4.     private final List<AbstractClassEnhancePluginDefine> signatureMatchDefine = new ArrayList<AbstractClassEnhancePluginDefine>();
  5.     private final List<AbstractClassEnhancePluginDefine> bootstrapClassMatchDefine = new ArrayList<AbstractClassEnhancePluginDefine>();
  6.        
  7.         //构造方法
  8.     public PluginFinder(List<AbstractClassEnhancePluginDefine> plugins) {
  9.         for (AbstractClassEnhancePluginDefine plugin : plugins) {
  10.             //抽象方法enhanceClass方法定义在插件的抽象基类AbstractClassEnhancePluginDefine中,每一个插件必须去实现这个类中的方法
  11.             ClassMatch match = plugin.enhanceClass();  //故enhanceClass是每个插件都会自己去实现的方法,指定需要增强的类
  12.             if (match == null) {
  13.                 continue;
  14.             }
  15.             if (match instanceof NameMatch) {
  16.                 NameMatch nameMatch = (NameMatch) match;
  17.                 LinkedList<AbstractClassEnhancePluginDefine> pluginDefines = nameMatchDefine.get(nameMatch.getClassName());
  18.                 if (pluginDefines == null) {
  19.                     pluginDefines = new LinkedList<AbstractClassEnhancePluginDefine>();
  20.                     nameMatchDefine.put(nameMatch.getClassName(), pluginDefines);
  21.                 }
  22.                 pluginDefines.add(plugin);
  23.             } else {
  24.                 signatureMatchDefine.add(plugin);
  25.             }
  26.             if (plugin.isBootstrapInstrumentation()) {
  27.                 bootstrapClassMatchDefine.add(plugin);
  28.             }
  29.         }
  30.     }
  31. //typeDescription是bytebuddy的内置接口,是对类的完整描述,包含了类的全类名
  32. //传入typeDescription,返回可以运用于typeDescription的类的插件
  33. public List<AbstractClassEnhancePluginDefine> find(TypeDescription typeDescription) {
  34.         List<AbstractClassEnhancePluginDefine> matchedPlugins = new LinkedList<AbstractClassEnhancePluginDefine>();
  35.         String typeName = typeDescription.getTypeName();
  36.             //根据名字信息匹配查找
  37.         if (nameMatchDefine.containsKey(typeName)) {
  38.             matchedPlugins.addAll(nameMatchDefine.get(typeName));
  39.         }
  40.                 //通过除了名字之外的辅助信息,在signatureMatchDefine集合中查找
  41.         for (AbstractClassEnhancePluginDefine pluginDefine : signatureMatchDefine) {
  42.             IndirectMatch match = (IndirectMatch) pluginDefine.enhanceClass();
  43.             if (match.isMatch(typeDescription)) {
  44.                 matchedPlugins.add(pluginDefine);
  45.             }
  46.         }
  47.         return matchedPlugins;
  48.     }
  49. public ElementMatcher<? super TypeDescription> buildMatch() {
  50.                    //设置匹配的规则,名字是否相同,通过名字直接匹配
  51.         ElementMatcher.Junction judge = new AbstractJunction<NamedElement>() {
  52.             @Override
  53.             public boolean matches(NamedElement target) {
  54.                 return nameMatchDefine.containsKey(target.getActualName());
  55.             }
  56.         };
  57.         judge = judge.and(not(isInterface())); //接口不增强,排除掉
  58.             //如果无法确定类的全限定名,则通过注解、回调信息等辅助方法间接匹配
  59.         for (AbstractClassEnhancePluginDefine define : signatureMatchDefine) {
  60.             ClassMatch match = define.enhanceClass();
  61.             if (match instanceof IndirectMatch) {
  62.                 judge = judge.or(((IndirectMatch) match).buildJunction());
  63.             }
  64.         }
  65.         return new ProtectiveShieldMatcher(judge);
  66. }
复制代码
思考:
1:如何自定义Skywalking插件
2:如何使用插件
本文由传智教育博学谷 - 狂野架构师教研团队发布
如果本文对您有帮助,欢迎关注和点赞;如果您有任何建议也可留言评论或私信,您的支持是我坚持创作的动力
转载请注明出处!

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!




欢迎光临 ToB企服应用市场:ToB评测及商务社交产业平台 (https://dis.qidao123.com/) Powered by Discuz! X3.4