【SkyWalking】如何在业务系统中控制SkyWalkingAgent的生命周期

写在前面

文章基于SkyWalking-oap-9.7.0版本,agent版本为9.1.0。

使用场景

在业务系统使用SkyWalking时,希望通过统一的应用来管理agent信息,例如运维系统 / 流水线系统等,使用SkyWalking的ui来管理,就会多出一套系统维护的成本,我们可以通过对SkyWalking的代码进行改造,实现agent数据统一化管理。

改造方案

在agent启动和停止阶段,加入一个grpc的调用,通过agent给oap服务传递的信息,由oap进行对业务系统的接口调用,上报agent的生命周期信息及心跳数据。

改造代码

agent部分

上线

前往apm-sniffer -> apm-agent-core -> remote目录下,找到ServiceManagementClient文件

在run方法中加入自定义的grpc请求代码,其中blockingStud是grpc-proto生成的文件名,online方法和Message类是proto文件中定义的方法名及实体类。

加入成功后,在agent启动时,就会将agent定义的serviceName和instanceName参数传给oap后端

下线

在ServiceManagementClient文件中,找到shutdown部分,加入下面的代码,在服务停止时,agent会先进行下线消息通知,再进行agent自身的停止逻辑。 需要特别注意的是,修改shutdown的方法后,需要调整BootService接口的执行顺序,否则会出现serviceBlockingStud为null的情况,导致下线通知失败。

ServiceManager中需要调整的部分如下,先通过反射获取到上面修改的ServiceManagementClient,先进行下线通知,然后一次对其他的BootService进行停止。

oap部分

在SkyWalking代码中,找到oap-server -> server-receiver-plugin目录,在下面新建一个用于接受agent上线/下线通知的新模块,核心代码为继承Grpc生成的类文件,并实现GRPCHandler接口,将Grpc服务交给oap统一管理。 在online和offline方法中,接收到agent上报的信息,经过自定义整合后,通过http或其他请求方式,传递给业务系统,进行持久化或其他处理。

相关推荐
王码码203522 分钟前
Go语言的测试:从单元测试到集成测试
后端·golang·go·接口
王码码203527 分钟前
Go语言中的测试:从单元测试到集成测试
后端·golang·go·接口
嵌入式×边缘AI:打怪升级日志1 小时前
使用JsonRPC实现前后台
前端·后端
小码哥_常2 小时前
从0到1:Spring Boot 中WebSocket实战揭秘,开启实时通信新时代
后端
lolo大魔王2 小时前
Go语言的异常处理
开发语言·后端·golang
IT_陈寒4 小时前
Python多进程共享变量那个坑,我差点没爬出来
前端·人工智能·后端
码事漫谈4 小时前
2026软考高级·系统架构设计师备考指南
后端
AI茶水间管理员5 小时前
如何让LLM稳定输出 JSON 格式结果?
前端·人工智能·后端
其实是白羊6 小时前
我用 Vibe Coding 搓了一个 IDEA 插件,复制URI 再也不用手动拼了
后端·intellij idea
用户8356290780516 小时前
Python 操作 Word 文档节与页面设置
后端·python