001/*
002 * Copyright 2012-2017 the original author or authors.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 *      http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016
017package org.springframework.boot.actuate.redis;
018
019import java.util.Properties;
020
021import reactor.core.publisher.Mono;
022
023import org.springframework.boot.actuate.health.AbstractReactiveHealthIndicator;
024import org.springframework.boot.actuate.health.Health;
025import org.springframework.boot.actuate.health.ReactiveHealthIndicator;
026import org.springframework.data.redis.connection.ReactiveRedisConnection;
027import org.springframework.data.redis.connection.ReactiveRedisConnectionFactory;
028
029/**
030 * A {@link ReactiveHealthIndicator} for Redis.
031 *
032 * @author Stephane Nicoll
033 * @author Mark Paluch
034 * @since 2.0.0
035 */
036public class RedisReactiveHealthIndicator extends AbstractReactiveHealthIndicator {
037
038        private final ReactiveRedisConnectionFactory connectionFactory;
039
040        public RedisReactiveHealthIndicator(
041                        ReactiveRedisConnectionFactory connectionFactory) {
042                this.connectionFactory = connectionFactory;
043        }
044
045        @Override
046        protected Mono<Health> doHealthCheck(Health.Builder builder) {
047                ReactiveRedisConnection connection = this.connectionFactory
048                                .getReactiveConnection();
049                return connection.serverCommands().info().map((info) -> up(builder, info))
050                                .doFinally((signal) -> connection.close());
051        }
052
053        private Health up(Health.Builder builder, Properties info) {
054                return builder.up().withDetail(RedisHealthIndicator.VERSION,
055                                info.getProperty(RedisHealthIndicator.REDIS_VERSION)).build();
056        }
057
058}