orbit.server.service.HealthService.kt Maven / Gradle / Ivy

Go to download
/*
 Copyright (C) 2015 - 2019 Electronic Arts Inc.  All rights reserved.
 This file is part of the Orbit Project .
 See license in LICENSE.
 */

package orbit.server.service

import grpc.health.v1.HealthImplBase
import grpc.health.v1.HealthOuterClass
import kotlinx.coroutines.channels.ReceiveChannel

class HealthService(private val checks: HealthCheckList) : HealthImplBase() {
    override suspend fun check(request: HealthOuterClass.HealthCheckRequest): HealthOuterClass.HealthCheckResponse {
        return HealthOuterClass.HealthCheckResponse.newBuilder()
            .setStatus(
                if (this.isHealthy()) {
                    HealthOuterClass.HealthCheckResponse.ServingStatus.SERVING
                } else {
                    HealthOuterClass.HealthCheckResponse.ServingStatus.NOT_SERVING
                }
            ).build()
    }

    override fun watch(request: HealthOuterClass.HealthCheckRequest): ReceiveChannel {
        return super.watch(request)
    }

    suspend fun isHealthy(): Boolean {
        return checks.getChecks().all { check -> check.isHealthy() }
    }
}