add prometheus exporter http server

Author: mingtsay

.idea/artifacts/moztw_space_bot_jar.xml

@@ -36,6 +36,10 @@ dependencies {
     implementation("com.squareup.okhttp3", "okhttp", "4.10.0")
     implementation("com.beust", "klaxon", "5.5")
     implementation("io.prometheus", "simpleclient", "0.16.0")
+    implementation("io.prometheus", "simpleclient_hotspot", "0.16.0")
+    implementation("io.prometheus", "simpleclient_httpserver", "0.16.0")
+    implementation("org.slf4j", "slf4j-api", "2.0.0")
+    implementation("org.slf4j", "slf4j-simple", "2.0.0")
     testImplementation("org.junit.jupiter", "junit-jupiter-api", "5.8.2")
 }
 

build.gradle.kts

@@ -1,6 +1,7 @@
 package org.moztw.bot.telegram.space
 
 import io.prometheus.client.Counter
+import io.prometheus.client.Gauge
 import okhttp3.OkHttpClient
 import okhttp3.Request
 import org.moztw.bot.telegram.space.co2.SpaceCo2
@@ -24,10 +25,11 @@ internal class Bot(val username: String, val token: String) : TelegramLongPollin
         updates.inc()
         println("* [$dateTime] Update received: $update")
 
-        if (!(update.hasMessage() && onMessageReceived(update.message) || update.hasCallbackQuery() && onCallbackQueryReceived(update.callbackQuery))) {
-            updatesUnhandled.inc()
-            println("* [$dateTime] Update not handled: $update")
-        }
+        if (update.hasMessage() && onMessageReceived(update.message)) return
+        if (update.hasCallbackQuery() && onCallbackQueryReceived((update.callbackQuery))) return
+
+        updatesUnhandled.inc()
+        println("* [$dateTime] Update not handled: $update")
     }
 
     private fun onCallbackQueryReceived(callbackQuery: CallbackQuery): Boolean {
@@ -104,9 +106,12 @@ internal class Bot(val username: String, val token: String) : TelegramLongPollin
                         }
                     }
                 }
+                return true
             } else if (isAdminChat(message.chat)) {
                 if (isCommandOpen(message.text)) {
                     commandCalls.labels("space_open").inc()
+                    spaceDoorState.set(1.0)
+                    spaceDoorStateTime.setToCurrentTime()
                     if (tryExecute(Caption().getCaptionOpened(chatId = generalChatId))
                             && tryExecute(Reply().getGeneralMessageOpen(chatId = generalChatId, operator = message.from))
                             && tryExecute(Reply().getMessageOpen(message = message))) {
@@ -118,6 +123,8 @@ internal class Bot(val username: String, val token: String) : TelegramLongPollin
                     }
                 } else if (isCommandClose(message.text)) {
                     commandCalls.labels("space_close").inc()
+                    spaceDoorState.set(0.0)
+                    spaceDoorStateTime.setToCurrentTime()
                     if (tryExecute(Caption().getCaptionClosed(chatId = generalChatId))
                             && tryExecute(Reply().getGeneralMessageClose(chatId = generalChatId, operator = message.from))
                             && tryExecute(Reply().getMessageClose(message = message))) {
@@ -218,5 +225,15 @@ internal class Bot(val username: String, val token: String) : TelegramLongPollin
             .help("Total Telegram Bots API calls.")
             .labelNames("method")
             .register()
+
+        val spaceDoorState: Gauge = Gauge.build()
+            .name("moztw_space_door_state")
+            .help("The state of MozTW Space Door. 0 = closed, 1 = opened")
+            .register()
+
+        val spaceDoorStateTime: Gauge = Gauge.build()
+            .name("moztw_space_door_state_time")
+            .help("The update time for the state of MozTW Space Door.")
+            .register()
     }
 }

src/main/kotlin/org/moztw/bot/telegram/space/Bot.kt

@@ -1,5 +1,6 @@
 package org.moztw.bot.telegram.space
 
+import io.prometheus.client.exporter.HTTPServer
 import org.apache.commons.cli.*
 import org.telegram.telegrambots.meta.TelegramBotsApi
 import org.telegram.telegrambots.meta.exceptions.TelegramApiRequestException
@@ -30,8 +31,10 @@ fun main(args: Array<String>) {
         val botToken = cmd.getOptionValue("token")
         val exporterListenPort = cmd.getOptionValue("port", "")
 
-        if (exporterListenPort.isNotBlank()){
-            Exporter.run(exporterListenPort.toInt())
+        if (exporterListenPort.isNotEmpty()) {
+            HTTPServer.Builder()
+                .withPort(exporterListenPort.toInt())
+                .build()
         }
 
         try {