K8s环境Spring Boot健康组状态聚合后响应前执行自定义逻辑

你需要在健康检查聚合逻辑完成后、响应返回前插入自定义逻辑，不要在单个FooHealthIndicator/BarHealthIndicator中实现相关逻辑——这两个组件仅负责返回单个依赖的健康状态，执行时机早于组聚合流程，无法拿到foo、bar两组聚合后的最终结果。

前提确认 #

先确认K8s存活探测指向的健康组配置：如果你已经在management.endpoint.health.group下把foo、bar加入了存活探测对应的组（比如默认的liveness组，或是自定义的专门给K8s探测用的组），确保探测路径和组名对应即可。

实现方式 #

根据你用的Spring Boot版本，二选一即可：

方式1：官方原生扩展（推荐，适用于Spring Boot 2.7+及3.x版本） #

Spring Boot从2.7版本开始提供了HealthEndpointResponseCustomizer扩展点，专门用于在健康端点响应返回前做自定义处理，能直接拿到当前请求所属的健康组、以及聚合完成的最终健康结果，不需要额外依赖AOP。
直接注册自定义实现类即可：

import org.springframework.boot.actuate.health.Health;
import org.springframework.boot.actuate.health.HealthEndpointResponseContext;
import org.springframework.boot.actuate.health.HealthEndpointResponseCustomizer;
import org.springframework.boot.actuate.health.Status;
import org.springframework.stereotype.Component;
import java.util.Set;

@Component
public class LivenessHealthCallbackCustomizer implements HealthEndpointResponseCustomizer<Health> {
    // 配置需要拦截的健康组名，即K8s存活探测对应的组，按实际配置修改
    private static final Set<String> TARGET_GROUPS = Set.of("liveness");

    @Override
    public void customize(HealthEndpointResponseContext<Health> context) {
        // 仅处理目标存活探测组的请求
        if (!TARGET_GROUPS.contains(context.getGroup().getName())) {
            return;
        }
        // 拿到聚合完成的最终健康结果
        Health finalHealth = context.getBody();
        Status finalStatus = finalHealth.getStatus();
        
        // 在这里写你的自定义逻辑即可
        // 可通过finalStatus判断最终是UP/DOWN/UNKNOWN状态
        // 可通过finalHealth.getDetails()拿到foo、bar两个指示器的单独检查结果
    }
}

方式2：AOP拦截（适用于Spring Boot 2.7以下版本） #

如果你的Spring Boot版本低于2.7，没有上述扩展点，可以用AOP直接拦截健康端点Web扩展类的执行方法，在原聚合逻辑执行完成后拿到结果做处理：
首先确保项目引入了AOP依赖（spring-boot-starter-aop），然后编写切面：

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.boot.actuate.health.Health;
import org.springframework.boot.actuate.health.HealthEndpointWebExtension;
import org.springframework.boot.actuate.health.Status;
import org.springframework.stereotype.Component;
import java.util.Set;

@Aspect
@Component
public class LivenessHealthCallbackAspect {
    private static final Set<String> TARGET_GROUPS = Set.of("liveness");

    // Servlet栈切点，如果是WebFlux响应式栈，把切点类换成HealthReactiveEndpointWebExtension即可
    @Around("execution(* org.springframework.boot.actuate.health.HealthEndpointWebExtension.health(..))")
    public Object afterHealthAggregated(ProceedingJoinPoint joinPoint) throws Throwable {
        // 执行原有健康检查+聚合逻辑，拿到返回结果
        Object response = joinPoint.proceed();
        // 从方法入参获取当前请求对应的健康组路径
        Object[] args = joinPoint.getArgs();
        String currentGroup = args.length > 0 ? String.valueOf(args[0]) : "";

        if (TARGET_GROUPS.contains(currentGroup) && response instanceof Health finalHealth) {
            Status finalStatus = finalHealth.getStatus();
            // 在这里写自定义逻辑，和上述customizer中的用法一致
        }
        // 原样返回结果，不影响K8s探测的正常响应
        return response;
    }
}

注意事项 #

• 自定义逻辑中不要随意修改拿到的finalHealth对象的状态和详情，除非你明确需要变更返回给K8s的探测结果，否则可能导致不符合预期的Pod重启/流量摘除。
• 自定义逻辑不要做耗时过长的操作，避免阻塞健康检查接口导致K8s判定探测超时。
• 如果你是把foo、bar两个组直接作为独立路径给K8s探测，只需要把TARGET_GROUPS里的组名改成"foo"、"bar"即可。

内容的提问来源于stack exchange，提问作者Hayi