grpcstarter.server.feature.health.Health Maven / Gradle / Ivy
package grpcstarter.server.feature.health;
import grpcstarter.server.GrpcServerProperties;
import grpcstarter.server.feature.health.datasource.DataSourceHealthChecker;
import grpcstarter.server.feature.health.redis.RedisHealthChecker;
import io.grpc.health.v1.HealthGrpc;
import io.grpc.protobuf.services.HealthStatusManager;
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.connection.RedisConnectionFactory;
import org.springframework.jdbc.core.JdbcTemplate;
/**
* @author Freeman
*/
@Configuration(proxyBeanMethods = false)
@ConditionalOnClass(HealthGrpc.HealthImplBase.class)
@ConditionalOnProperty(prefix = GrpcServerProperties.Health.PREFIX, name = "enabled", matchIfMissing = true)
public class Health {
@Bean
@ConditionalOnMissingBean
public HealthStatusManager grpcHealthStatusManager() {
return new HealthStatusManager();
}
@Bean
@ConditionalOnMissingBean
public HealthGrpc.HealthImplBase grpcHealthService(HealthStatusManager healthManager) {
return (HealthGrpc.HealthImplBase) healthManager.getHealthService();
}
@Bean
@ConditionalOnMissingBean
public HealthServerInterceptor grpcHealthServerInterceptor(
HealthStatusManager healthManager, ObjectProvider healthCheckers) {
return new HealthServerInterceptor(healthManager, healthCheckers);
}
@Configuration(proxyBeanMethods = false)
@ConditionalOnProperty(
prefix = GrpcServerProperties.Health.DataSource.PREFIX,
name = "enabled",
matchIfMissing = true)
@ConditionalOnClass(JdbcTemplate.class)
static class DataSource {
@Bean
@ConditionalOnMissingBean
public DataSourceHealthChecker grpcDataSourceHealthChecker(GrpcServerProperties properties) {
return new DataSourceHealthChecker(properties.getHealth().getDatasource());
}
}
@Configuration(proxyBeanMethods = false)
@ConditionalOnProperty(prefix = GrpcServerProperties.Health.Redis.PREFIX, name = "enabled", matchIfMissing = true)
@ConditionalOnClass(RedisConnectionFactory.class)
static class Redis {
@Bean
@ConditionalOnMissingBean
public RedisHealthChecker grpcRedisHealthChecker(GrpcServerProperties properties) {
return new RedisHealthChecker(properties.getHealth().getRedis());
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy