com.commercetools.sunrise.common.healthcheck.SunriseHealthCheckController Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common_2.11 Show documentation
common
The newest version!
package com.commercetools.sunrise.common.healthcheck;

import com.commercetools.sunrise.framework.controllers.SunriseController;
import io.sphere.sdk.client.SphereClient;
import io.sphere.sdk.products.ProductProjection;
import io.sphere.sdk.products.search.ProductProjectionSearch;
import io.sphere.sdk.search.PagedSearchResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import play.libs.concurrent.HttpExecution;
import play.mvc.Http;
import play.mvc.Result;

import java.io.IOException;
import java.util.concurrent.CompletionStage;

/**
 * Controller for health check report.
 */
public abstract class SunriseHealthCheckController extends SunriseController {

    private final Logger logger = LoggerFactory.getLogger(getClass());
    private final SphereClient sphereClient;

    protected SunriseHealthCheckController(final SphereClient sphereClient) {
        this.sphereClient = sphereClient;
    }

    public CompletionStage show() throws IOException {
        final ProductProjectionSearch productRequest = ProductProjectionSearch.ofCurrent()
                .withMarkingMatchingVariants(false)
                .withLimit(1);
        return sphereClient.execute(productRequest)
                .thenApplyAsync(this::renderGoodHealthStatus, HttpExecution.defaultContext())
                .exceptionally(this::renderBadHealthStatus)
                .thenApply(result -> result.as(Http.MimeTypes.JSON));
    }

    private Result renderGoodHealthStatus(final PagedSearchResult productResult) {
        final boolean containsNoProducts = productResult.getResults().isEmpty();
        if (containsNoProducts) {
            throw new RuntimeException("Cannot find any product!");
        }
        return ok(healthReportAsJson(true));
    }

    private Result renderBadHealthStatus(final Throwable throwable) {
        logger.error("Could not fetch products", throwable);
        return status(Http.Status.SERVICE_UNAVAILABLE, healthReportAsJson(false));
    }

    private String healthReportAsJson(final boolean healthy) {
        return "{\n" +
                "  \"self\" : {\n" +
                "    \"healthy\" : " + healthy + "\n" +
                "  }\n" +
                "}";
    }
}