spring cloud框架之健康检查（多数据源场景）

import org.springframework.boot.actuate.health.CompositeHealthIndicator;
import org.springframework.boot.actuate.health.Health;
import org.springframework.boot.actuate.health.HealthIndicator;
import org.springframework.boot.actuate.health.Status;
import org.springframework.stereotype.Component;
 
import javax.annotation.PostConstruct;
import java.util.HashMap;
import java.util.Map;
 
@Component
public class CustomHealthIndicator extends CompositeHealthIndicator {
 
    private final HealthIndicator db1HealthIndicator;
    private final HealthIndicator db2HealthIndicator;
 
    public CustomHealthIndicator(HealthIndicator db1HealthIndicator, HealthIndicator db2HealthIndicator) {
        this.db1HealthIndicator = db1HealthIndicator;
        this.db2HealthIndicator = db2HealthIndicator;
    }
 
    @PostConstruct
    private void initialize() {
        // 注册健康指标，可以通过这种方式为不同的数据源定义健康指标
        healthIndicators().put("db1", db1HealthIndicator);
        healthIndicators().put("db2", db2HealthIndicator);
    }
 
    @Override
    protected Map<String, HealthIndicator> getHealthIndicators() {
        return healthIndicators();
    }
 
    private Map<String, HealthIndicator> healthIndicators() {
        // 这里可以根据实际情况动态初始化健康指标
        return new HashMap<>();
    }
 
    @Override
    public Health health() {
        // 这里可以自定义返回的健康信息，比如返回所有数据源的状态
        int upCount = 0;
        Map<String, Health> healths = new HashMap<>();
        for (Map.Entry<String, HealthIndicator> entry : getHealthIndicators().entrySet()) {
            Health health = entry.getValue().health();
            healths.put(entry.getKey(), health);
            if (health.getStatus().equals(Status.UP)) {
                upCount++;
            }
        }
        // 假设如果任何一个数据源是DOWN，则整个服务状态是DOWN
        Status status = (upCount == 0) ? Status.DOWN : Status.UP;
        return new Health.Builder(status, healths).build();
    }
}

这个代码示例展示了如何在Spring Cloud应用中定义一个自定义的健康指标，用于监控多个数据源。在这个例子中，我们创建了一个CustomHealthIndicator类，它继承自CompositeHealthIndicator。在初始化方法中，我们注册了两个健康指标，分别对应两个数据源。在health方法中，我们自定义了返回的健康信息，通过遍历所有的健康指标，我们可以返回每个数据源的健康状态，并且可以设置整体服务的健康状态。