结合SpringBoot的actuator实现自定义k8s优雅停机插件

要在Spring Boot中实现一个适用于Kubernetes优雅停机的自定义actuator端点，你可以创建一个自定义的Endpoint来响应Kubernetes发送的停机信号。以下是一个简化的例子：

1. 创建一个自定义的Endpoint来处理Kubernetes发送的优雅停机信号。

import org.springframework.boot.actuate.endpoint.annotation.Endpoint;
import org.springframework.boot.actuate.endpoint.annotation.Selector;
import org.springframework.boot.actuate.endpoint.annotation.WriteOperation;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Component;
 
@Component
@Endpoint(id = "k8s-shutdown")
public class K8SShutdownEndpoint {
 
    private final ApplicationContext applicationContext;
 
    public K8SShutdownEndpoint(ApplicationContext applicationContext) {
        this.applicationContext = applicationContext;
    }
 
    @WriteOperation
    public void shutdown() {
        // 执行优雅停机逻辑
        System.out.println("Received request to shutdown gracefully");
        // 可以调用Spring Boot的停机钩子来触发优雅停机
        applicationContext.publishEvent(new ShutdownEvent());
    }
}

2. 确保Spring Boot的ShutdownEndpoint没有被自动配置，或者将其暴露出来，以避免非优雅关闭。

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.boot.actuate.autoconfigure.endpoint.EndpointAutoConfiguration;
import org.springframework.boot.actuate.autoconfigure.web.server.ManagementWebServerAutoConfiguration;
import org.springframework.boot.actuate.endpoint.ShutdownEndpoint;
 
@Configuration
public class ActuatorConfig {
 
    @Bean
    public ShutdownEndpoint shutdownEndpoint() {
        // 返回一个自定义的ShutdownEndpoint或者一个空的Bean来禁用自动配置的ShutdownEndpoint
        return new ShutdownEndpoint();
    }
}

3. 在你的application.properties或application.yml中配置actuator端点暴露：

management.endpoints.web.exposure.include=k8s-shutdown

确保你的应用程序暴露了k8s-shutdown端点，并且Kubernetes能够访问这个端点。当Kubernetes向你的应用程序发送优雅停机信号时，它会调用这个自定义k8s-shutdown端点，应用程序会执行优雅的关闭流程。