TinyHealth.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package net.morimekta.tiny.server.http;

import com.fasterxml.jackson.annotation.JsonProperty;

import java.util.Objects;

/**
 * Handle health and readiness for Tiny Server.
 */
public final class TinyHealth {
    /**
     * Interface for checking the healthiness of a service.
     */
    @FunctionalInterface
    public interface HealthCheck {
        /**
         * @return The healthiness result of the service.
         */
        Result healthCheck();
    }

    /**
     * Interface for checking the readiness of a service.
     */
    @FunctionalInterface
    public interface ReadyCheck {
        /**
         * @return The readiness result of the service.
         */
        Result readyCheck();
    }

    /**
     * The actual healthiness status.
     */
    public enum Status {
        /** Service is OK. */
        OK,
        /** Service is unhealthy. */
        UNHEALTHY,
        /** Error getting status. */
        ERROR
    }

    /**
     * @param message Healthy service message. Optional.
     * @return The OK result.
     */
    public static Result ok(String message) {
        return new Result(Status.OK, String.valueOf(message));
    }

    /**
     * @param message The unhealthy service message. Required.
     * @return The UNHEALTHY service.
     */
    public static Result unhealthy(String message) {
        return new Result(Status.UNHEALTHY, message);
    }

    /**
     * The healthiness result.
     */
    public final static class Result {
        private final Status status;
        private final String message;

        /**
         * @param status  The result status.
         * @param message The result status message.
         */
        public Result(Status status, String message) {
            this.status = Objects.requireNonNull(status, "status == null");
            this.message = Objects.requireNonNull(message, "message == null");
        }

        /**
         * @return The healthiness status.
         */
        @JsonProperty("status")
        public Status getStatus() {
            return status;
        }

        /**
         * @return The healthiness status message.
         */
        @JsonProperty("message")
        public String getMessage() {
            return message;
        }

        @Override
        public String toString() {
            return "Result{" +
                   "status=" + status +
                   ", message='" + message + '\'' +
                   '}';
        }

        @Override
        public boolean equals(Object o) {
            if (this == o) {
                return true;
            }
            if (o == null || getClass() != o.getClass()) {
                return false;
            }
            Result result = (Result) o;
            return status == result.status && message.equals(result.message);
        }

        @Override
        public int hashCode() {
            return Objects.hash(status, message);
        }
    }

    // ----- PRIVATE -----
    private TinyHealth() {}
}