【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或其他请求方式,传递给业务系统,进行持久化或其他处理。

相关推荐
Marktowin5 小时前
Mybatis-Plus更新操作时的一个坑
java·后端
赵文宇6 小时前
CNCF Dragonfly 毕业啦!基于P2P的镜像和文件分发系统快速入门,在线体验
后端
程序员爱钓鱼6 小时前
Node.js 编程实战:即时聊天应用 —— WebSocket 实现实时通信
前端·后端·node.js
Libby博仙7 小时前
Spring Boot 条件化注解深度解析
java·spring boot·后端
源代码•宸7 小时前
Golang原理剖析(Map 源码梳理)
经验分享·后端·算法·leetcode·golang·map
小周在成长7 小时前
动态SQL与MyBatis动态SQL最佳实践
后端
瓦尔登湖懒羊羊7 小时前
TCP的自我介绍
后端
小周在成长7 小时前
MyBatis 动态SQL学习
后端
子非鱼9217 小时前
SpringBoot快速上手
java·spring boot·后端