Cookies_Github_mirror/AquaDX
4
0
Fork 1
mirror of https://github.com/MewoLab/AquaDX.git synced 2026-02-12 08:27:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

format

Browse Source
This commit is contained in:
Menci
2024-12-11 11:11:17 +08:00
parent ebafb4c05e
commit ee88be613c
1 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/main/java/icu/samnyan/aqua/net/db/Prometheus.kt
View File

@@ -23,7 +23,11 @@ class APICounter(val api: String, val metrics: APIMetrics) {
val start = TimeSource.Monotonic.markNow() val start = TimeSource.Monotonic.markNow()
try { return fn().also { +this } } try { return fn().also { +this } }
catch (e: Exception) { throw e.also { this % e } } catch (e: Exception) { throw e.also { this % e } }
finally { metrics.timer("api_latency", arr("api", api)).record(start.elapsedNow().toJavaDuration()) } finally {
metrics
.timer("api_latency", arr("api", api))
.record(start.elapsedNow().toJavaDuration())
}
} }
} }
@@ -35,11 +39,18 @@ class APIMetrics(val domain: String) {
get(name, pairs.flatMap { listOf(it.first, it.second.toString()) }.toTypedArray()) get(name, pairs.flatMap { listOf(it.first, it.second.toString()) }.toTypedArray())
operator fun get(name: String, tag: Array<String>) = cache.computeIfAbsent(tag) { operator fun get(name: String, tag: Array<String>) = cache.computeIfAbsent(tag) {
Counter.builder("aquadx_${domain}_$name").tags(*tag).register(reg) Counter
.builder("aquadx_${domain}_$name")
.tags(*tag)
.register(reg)
} as Counter } as Counter
fun timer(name: String, tag: Array<String>) = cache.computeIfAbsent(tag) { fun timer(name: String, tag: Array<String>) = cache.computeIfAbsent(tag) {
Timer.builder("aquadx_${domain}_$name").tags(*tag).publishPercentiles(0.5, 0.75, 0.90, 0.95, 0.99).register(reg) Timer
.builder("aquadx_${domain}_$name")
.tags(*tag)
.publishPercentiles(0.5, 0.75, 0.90, 0.95, 0.99)
.register(reg)
} as Timer } as Timer
operator fun get(api: String) = APICounter(api, this) operator fun get(api: String) = APICounter(api, this)