当前位置：首页 > article >正文

PDF书籍《手写调用链监控APM系统-Java版》第10章插件与链路的结合：SpringBoot环境插件获取应用名

article 2024/12/28 1:15:49

本人阅读了 Skywalking 的大部分核心代码，也了解了相关的文献，对此深有感悟，特此借助巨人的思想自己手动用JAVA语言实现了一个 “调用链监控APM” 系统。本书采用边讲解实现原理边编写代码的方式，看本书时一定要跟着敲代码。

作者已经将过程写成一部书籍，奈何没有钱发表，如果您知道渠道可以联系本人。一定重谢。

本书涉及到的核心技术与思想

JavaAgent ， ByteBuddy，SPI服务，类加载器的命名空间，增强JDK类，kafka，插件思想，切面，链路栈等等。实际上远不止这么多，差不多贯通了整个java体系。

适用人群

自己公司要实现自己的调用链的；写架构的；深入java编程的；阅读Skywalking源码的；

版权

本书是作者呕心沥血亲自编写的代码，不经同意切勿拿出去商用，否则会追究其责任。

原版PDF+源码请见：

本章涉及到的工具类也在这里面：

PDF书籍《手写调用链监控APM系统-Java版》第1章开篇介绍-CSDN博客

第10章插件与链路的结合：SpringBoot环境插件获取应用名

由于我们上报给kafka的数据中serviceName为空，所以制作了springboot的环境插件，用来填充serviceName。

要想获取到springboot的环境配置，需要拦截的信息如下：

类名：org.springframework.boot.SpringApplication

方法：prepareEnvironment

非JDK类库

下面来开发这个插件，要记住前面说的插件开发四部曲。在插件模块下新增app-enviroment-plugin项目，hadluo-apm-plugin.def内容如下：

app-names=com.hadluo.apm.enviroment.AppNamesInstrumentation

AppNamesInstrumentation代码如下：

public class AppNamesInstrumentation extends AbstractClassEnhancePluginDefine {
    @Override
    public String enhanceClass() {
        return "org.springframework.boot.SpringApplication";
    }
    @Override
    public MethodsInterceptPoint[] configMethodsInterceptPoint() {
        return new MethodsInterceptPoint[]{
                new MethodsInterceptPoint() {
                    @Override
                    public ElementMatcher<MethodDescription> getMethodsMatcher() {
                        // 拦截 prepareEnvironment 方法
                        return ElementMatchers.named("prepareEnvironment");
                    }
                    @Override
                    public String getMethodsInterceptor() {
                        // 拦截逻辑执行的拦截器
                        return "com.hadluo.apm.enviroment.AppNamesInterceptor";
                    }
                    @Override
                    public boolean isOverrideArgs() {
                        return false;
                    }
                }
        };
    }
}

AppNamesInterceptor 代码如下：

public class AppNamesInterceptor implements InstanceMethodsAroundInterceptor {

    @Override
    public Object afterMethod(EnhancedInstance objInst, Method method, Object[] allArguments, Class<?>[] argumentsTypes, Object ret) throws Throwable {
        ConfigurableEnvironment environment = (ConfigurableEnvironment) ret;
        Config.Agent.serviceName = environment.getProperty("spring.application.name");
        return ret;
    }
}

SpringApplication#prepareEnvironment这个方法结束后，就已经完成了配置文件等的解析存储，信息都是存储到返回值ConfigurableEnvironment 对象上面。我们直接获取就可以了，然后设置到我们的配置Config.Agent.serviceName里面。

修改controller接口为一个空接口，主要是看serverName有没有获取到，打包测试：