Compare commits

...

1 Commits

Author SHA1 Message Date
usbharu a54aadfe87
watchdog-be: bellにアラートを出せるように 2024-03-19 16:17:14 +09:00
3 changed files with 55 additions and 8 deletions

View File

@ -3,9 +3,8 @@ package dev.usbharu.unos.watchdog.be
import com.mongodb.ConnectionString
import com.mongodb.MongoClientSettings
import com.mongodb.kotlin.client.coroutine.MongoClient
import com.mongodb.kotlin.client.coroutine.MongoDatabase
import dev.usbharu.unos.watchdog.be.domain.MongoMetricsRepository
import io.grpc.ServerBuilder
import io.grpc.ManagedChannelBuilder
import io.ktor.http.*
import io.ktor.serialization.kotlinx.json.*
import io.ktor.server.application.*
@ -13,7 +12,10 @@ import io.ktor.server.plugins.callloging.*
import io.ktor.server.plugins.contentnegotiation.*
import io.ktor.server.plugins.cors.routing.*
import io.ktor.server.routing.*
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.asExecutor
import org.slf4j.event.Level
import watchdog.WatchdogBellServiceGrpcKt
import java.net.URI
fun main(args: Array<String>) {
@ -56,7 +58,22 @@ fun Application.module() {
val mongoMetricsRepository = MongoMetricsRepository(database)
grpc(grpcPort,PushMetricsService(mongoMetricsRepository))
val useBell = environment.config.property("watchdog.bell.use").getString().toBoolean()
val bell = if (useBell) {
val bellHost = environment.config.property("watchdog.bell.host").getString()
val bellPort = environment.config.property("watchdog.bell.port").getString().toInt()
val channel =
ManagedChannelBuilder.forAddress(bellHost, bellPort).usePlaintext().executor(Dispatchers.IO.asExecutor())
.build()
WatchdogBellServiceGrpcKt.WatchdogBellServiceCoroutineStub(channel)
} else {
null
}
grpc(grpcPort, PushMetricsService(mongoMetricsRepository, bell))
routing {
api(ApiController(mongoMetricsRepository))
}

View File

@ -6,18 +6,22 @@ import dev.usbharu.unos.watchdog.be.domain.Metric
import dev.usbharu.unos.watchdog.be.domain.MetricsRepository
import dev.usbharu.unos.watchdog.be.domain.Status
import kotlinx.datetime.toKotlinInstant
import watchdog.PushMetrics
import watchdog.PushMetricsServiceGrpcKt
import watchdog.*
import java.util.*
import kotlin.coroutines.CoroutineContext
import kotlin.coroutines.EmptyCoroutineContext
class PushMetricsService(
private val metricsRepository: MetricsRepository,
private val bell: WatchdogBellServiceGrpcKt.WatchdogBellServiceCoroutineStub?,
coroutineContext: CoroutineContext = EmptyCoroutineContext
) :
PushMetricsServiceGrpcKt.PushMetricsServiceCoroutineImplBase(coroutineContext) {
override suspend fun push(request: PushMetrics.Metric): Empty {
bell(request)
metricsRepository.save(
Metric(
id = UUID.randomUUID().toString(),
@ -31,4 +35,25 @@ class PushMetricsService(
)
return Empty.getDefaultInstance()
}
private suspend fun bell(request: PushMetrics.Metric) {
val level = when (request.status) {
watchdog.PushMetrics.Status.OK -> return
watchdog.PushMetrics.Status.DEAD -> PushBell.Bell.Level.FATAL
watchdog.PushMetrics.Status.UNKNOWN -> PushBell.Bell.Level.WARN
watchdog.PushMetrics.Status.NOT_GOOD -> PushBell.Bell.Level.WARN
watchdog.PushMetrics.Status.UNRECOGNIZED -> PushBell.Bell.Level.WARN
}
bell?.push(
bell {
this.level = level
this.title = "${request.name} is ${request.status}"
this.location = request.domain
this.timestamp = request.timestamp
this.message = request.message
}
)
}
}

View File

@ -9,6 +9,7 @@ ktor {
}
watchdog {
front-url = "http://localhost:8081"
db {
url = "mongodb://localhost:27017"
database = "watchdog"
@ -16,5 +17,9 @@ watchdog {
grpc {
port = 50051
}
bell {
use = "true"
host = "localhost"
port = 50051
}
}