dalbodeule/chibot-chzzk-bot
1
0
Fork 0
mirror of https://github.com/dalbodeule/chibot-chzzk-bot.git synced 2025-08-07 04:41:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

[feature] add some api, etc..

Browse Source 
- add some api
- add /metrics routing
This commit is contained in:
dalbodeule
2025-07-15 21:52:41 +09:00
parent 9e3a79a613
commit 66df771cb7
8 changed files with 143 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
chatbot/src/main/kotlin/space/mori/chzzk_bot/chatbot/chzzk/ChzzkHandler.kt
View File

@@ -9,6 +9,7 @@ import space.mori.chzzk_bot.chatbot.chzzk.Connector.getChannel
import space.mori.chzzk_bot.chatbot.discord.Discord
import space.mori.chzzk_bot.chatbot.utils.refreshAccessToken
import space.mori.chzzk_bot.common.events.*
import space.mori.chzzk_bot.common.metrics.Metrics
import space.mori.chzzk_bot.common.models.User
import space.mori.chzzk_bot.common.services.LiveStatusService
import space.mori.chzzk_bot.common.services.TimerConfigService
@@ -293,6 +294,7 @@ class UserHandler(
logger.info("ChzzkChat connecting... ${channel.channelName} - ${channel.channelId}")
streamStartTime = LocalDateTime.now()
Metrics.streaming++
if(!_isActive) {
_isActive = true
@@ -329,6 +331,7 @@ class UserHandler(
listener?.unsubscribeAsync(ChzzkSessionSubscriptionType.CHAT)?.join()
listener?.disconnectAsync()?.join()
_isActive = false
Metrics.streaming--;
CoroutineScope(Dispatchers.Default).launch {
val events = listOf(

5
common/build.gradle.kts
View File

@@ -37,6 +37,11 @@ dependencies {
// https://mvnrepository.com/artifact/com.google.code.gson/gson
implementation("com.google.code.gson:gson:2.11.0")
api("io.micrometer:micrometer-registry-prometheus:1.15.1")
// https://mvnrepository.com/artifact/io.insert-koin/koin-core
api("io.insert-koin:koin-core:4.0.0")
testImplementation(kotlin("test"))
}

25
common/src/main/kotlin/space/mori/chzzk_bot/common/metrics/Metrics.kt Normal file
View File

@@ -0,0 +1,25 @@
package space.mori.chzzk_bot.common.metrics
import io.micrometer.core.instrument.Gauge
import io.micrometer.prometheusmetrics.PrometheusConfig
import io.micrometer.prometheusmetrics.PrometheusMeterRegistry
import space.mori.chzzk_bot.common.services.UserService
object Metrics {
val registry = PrometheusMeterRegistry(PrometheusConfig.DEFAULT)
var streamer = UserService.getAllUsers().size.toDouble()
val streamerGauge: Gauge = Gauge.builder("streamer_gauge", this) { streamer }
.description("Current All Streamer Count")
.register(registry)
var activeStreamer = UserService.getAllUsers().filter { !it.isDisabled }.size.toDouble()
val activateGauge: Gauge = Gauge.builder("active_streamer_gauge", this) { streamer }
.description("Current Active Streamer Count")
.register(registry)
var streaming: Double = 0.0
val streamingGauge: Gauge = Gauge.builder("streaming_gauge", this) { streaming }
.description("Current Streaming Streamer Count")
.register(registry)
}

2
src/main/kotlin/space/mori/chzzk_bot/Main.kt
View File

@@ -12,6 +12,7 @@ import space.mori.chzzk_bot.chatbot.discord.Discord
import space.mori.chzzk_bot.chatbot.chzzk.Connector as ChzzkConnector
import space.mori.chzzk_bot.common.Connector
import space.mori.chzzk_bot.common.events.CoroutinesEventBus
import space.mori.chzzk_bot.common.metrics.Metrics
import space.mori.chzzk_bot.webserver.start
import space.mori.chzzk_bot.webserver.stop
import java.util.concurrent.TimeUnit
@@ -25,6 +26,7 @@ val logger: Logger = LoggerFactory.getLogger("main")
fun main(args: Array<String>) {
val dispatcher = module {
single { CoroutinesEventBus() }
single { Metrics.registry }
}
startKoin {
modules(dispatcher)

6
webserver/build.gradle.kts
View File

@@ -34,16 +34,16 @@ dependencies {
// https://mvnrepository.com/artifact/org.jetbrains.kotlin/kotlin-reflect
implementation("org.jetbrains.kotlin:kotlin-reflect:2.0.21")
// https://mvnrepository.com/artifact/io.insert-koin/koin-core
implementation("io.insert-koin:koin-core:4.0.0")
// https://mvnrepository.com/artifact/ch.qos.logback/logback-classic
implementation("ch.qos.logback:logback-classic:1.5.12")
// https://mvnrepository.com/artifact/io.github.cdimascio/dotenv-kotlin
implementation("io.github.cdimascio:dotenv-kotlin:6.4.2")
implementation("io.ktor:ktor-server-metrics-micrometer:$ktorVersion")
implementation(project(":common"))
implementation("io.ktor:ktor-server-metrics-micrometer:3.1.3")
testImplementation(kotlin("test"))
}

22
webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/Main.kt
View File

@@ -11,6 +11,7 @@ import io.ktor.serialization.kotlinx.json.*
import io.ktor.server.application.*
import io.ktor.server.auth.*
import io.ktor.server.engine.*
import io.ktor.server.metrics.micrometer.MicrometerMetrics
import io.ktor.server.netty.*
import io.ktor.server.plugins.contentnegotiation.*
import io.ktor.server.plugins.cors.routing.*
@@ -19,9 +20,16 @@ import io.ktor.server.response.*
import io.ktor.server.routing.*
import io.ktor.server.sessions.*
import io.ktor.server.websocket.*
import io.micrometer.core.instrument.binder.jvm.JvmGcMetrics
import io.micrometer.core.instrument.binder.jvm.JvmMemoryMetrics
import io.micrometer.core.instrument.binder.system.ProcessorMetrics
import io.micrometer.prometheusmetrics.PrometheusConfig
import io.micrometer.prometheusmetrics.PrometheusMeterRegistry
import kotlinx.coroutines.delay
import kotlinx.serialization.Serializable
import kotlinx.serialization.json.Json
import org.koin.core.context.startKoin
import org.koin.dsl.module
import org.koin.java.KoinJavaComponent.inject
import space.mori.chzzk_bot.common.events.CoroutinesEventBus
import space.mori.chzzk_bot.common.events.UserRegisterEvent
@@ -232,6 +240,18 @@ val server = embeddedServer(Netty, port = 8080, ) {
}
}
val appMicrometerRegistry: PrometheusMeterRegistry by inject(PrometheusMeterRegistry::class.java)
install(MicrometerMetrics) {
registry = appMicrometerRegistry
meterBinders = listOf(
JvmMemoryMetrics(),
JvmGcMetrics(),
ProcessorMetrics()
)
}
apiRoutes()
apiSongRoutes()
apiCommandRoutes()
@@ -242,6 +262,8 @@ val server = embeddedServer(Netty, port = 8080, ) {
wsSongRoutes()
wsSongListRoutes()
metricRoutes()
swaggerUI("swagger-ui/index.html", "openapi/documentation.yaml") {
options {
version = "1.2.0"

48
webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/routes/ApiRoutes.kt
View File

@@ -1,6 +1,7 @@
package space.mori.chzzk_bot.webserver.routes
import io.ktor.http.*
import io.ktor.server.request.receive
import io.ktor.server.response.*
import io.ktor.server.routing.*
import io.ktor.server.sessions.*
@@ -15,6 +16,7 @@ import kotlinx.coroutines.CompletableDeferred
import kotlinx.coroutines.withTimeoutOrNull
import space.mori.chzzk_bot.common.events.ChzzkUserFindEvent
import space.mori.chzzk_bot.common.events.ChzzkUserReceiveEvent
import space.mori.chzzk_bot.webserver.routes.GuildSettings
@Serializable
data class GetUserDTO(
@@ -146,4 +148,48 @@ fun Routing.apiRoutes() {
call.respond(HttpStatusCode.OK, returnUsers)
}
}
}
route("/settings") {
get {
val session = call.sessions.get<UserSession>()
if(session == null) {
call.respondText("No session found", status = HttpStatusCode.Unauthorized)
return@get
}
val user = UserService.getUser(session.id)
if(user == null) {
call.respondText("No user found", status = HttpStatusCode.NotFound)
return@get
}
call.respond(HttpStatusCode.OK, IUserSettingsDTO(
user.isDisabled,
user.isDisableStartupMsg
))
}
post {
val session = call.sessions.get<UserSession>()
val body: IUserSettingsDTO = call.receive()
if(session == null) {
call.respondText("No session found", status = HttpStatusCode.Unauthorized)
return@post
}
val user = UserService.getUser(session.id)
if(user == null) {
call.respondText("No user found", status = HttpStatusCode.NotFound)
return@post
}
UserService.setIsDisabled(user, body.isBotDisabled)
UserService.setIsStartupDisabled(user, body.isBotMsgDisabled)
call.respond(HttpStatusCode.OK, body)
}
}
}
data class IUserSettingsDTO(
val isBotDisabled: Boolean,
val isBotMsgDisabled: Boolean
)

36
webserver/src/main/kotlin/space/mori/chzzk_bot/webserver/routes/MetricRoutes.kt Normal file
View File

@@ -0,0 +1,36 @@
package space.mori.chzzk_bot.webserver.routes
import io.ktor.server.application.ApplicationStopped
import io.ktor.server.response.respondText
import io.ktor.server.routing.Routing
import io.ktor.server.routing.get
import io.micrometer.prometheusmetrics.PrometheusMeterRegistry
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.SupervisorJob
import kotlinx.coroutines.cancel
import org.koin.java.KoinJavaComponent.inject
import space.mori.chzzk_bot.common.events.CoroutinesEventBus
import space.mori.chzzk_bot.common.events.UserRegisterEvent
import space.mori.chzzk_bot.common.metrics.Metrics
import kotlin.getValue
val metricScope = CoroutineScope(SupervisorJob() + Dispatchers.Default)
fun Routing.metricRoutes() {
environment.monitor.subscribe(ApplicationStopped) {
metricScope.cancel()
}
val dispatcher: CoroutinesEventBus by inject(CoroutinesEventBus::class.java)
val registry: PrometheusMeterRegistry by inject(PrometheusMeterRegistry::class.java)
dispatcher.subscribe(UserRegisterEvent::class) {
Metrics.streamer++
}
get("/metrics") {
call.respondText(registry.scrape())
}
}