net.morimekta.tiny.http.TinyHealth Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tiny-http Show documentation
Smallest footprint of HTTP server helper library.
There is a newer version: 0.7.1
/*
 * 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.http;

import com.fasterxml.jackson.annotation.JsonInclude;
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
    }

    /**
     * @return The OK result.
     */
    public static Result ok() {
        return new Result(Status.OK, null);
    }

    /**
     * @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.
     */
    @JsonInclude(JsonInclude.Include.NON_NULL)
    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 = message;
        }

        /**
         * @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() {}
}