StakeSquid/dshackle
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

auto restart prometheus metrics server (#669)

Browse Source
This commit is contained in:
msizov
2025-06-02 18:52:30 +07:00
committed by GitHub
parent f3bb97d51a
commit c8454a8e71
1 changed files with 47 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
src/main/kotlin/io/emeraldpay/dshackle/monitoring/MonitoringSetup.kt
View File

@@ -32,7 +32,9 @@ import org.slf4j.LoggerFactory
import org.springframework.beans.factory.annotation.Autowired import org.springframework.beans.factory.annotation.Autowired
import org.springframework.stereotype.Service import org.springframework.stereotype.Service
import java.io.IOException import java.io.IOException
import java.net.InetAddress
import java.net.InetSocketAddress import java.net.InetSocketAddress
import java.net.ServerSocket
import java.util.zip.GZIPOutputStream import java.util.zip.GZIPOutputStream
import javax.annotation.PostConstruct import javax.annotation.PostConstruct
@@ -45,6 +47,20 @@ class MonitoringSetup(
private val log = LoggerFactory.getLogger(MonitoringSetup::class.java) private val log = LoggerFactory.getLogger(MonitoringSetup::class.java)
} }
private fun isTcpPortAvailable(host: String, port: Int): Boolean {
try {
ServerSocket().use { serverSocket ->
// setReuseAddress(false) is required only on macOS,
// otherwise the code will not work correctly on that platform
serverSocket.reuseAddress = false
serverSocket.bind(InetSocketAddress(InetAddress.getByName(host), port), 1)
return true
}
} catch (ex: java.lang.Exception) {
return false
}
}
@PostConstruct @PostConstruct
fun setup() { fun setup() {
val prometheusRegistry = PrometheusMeterRegistry(PrometheusConfig.DEFAULT) val prometheusRegistry = PrometheusMeterRegistry(PrometheusConfig.DEFAULT)
@@ -75,29 +91,40 @@ class MonitoringSetup(
if (monitoringConfig.prometheus.enabled) { if (monitoringConfig.prometheus.enabled) {
// use standard JVM server with a single thread blocking processing // use standard JVM server with a single thread blocking processing
// prometheus is a single thread periodic call, no reason to setup anything complex // prometheus is a single thread periodic call, no reason to setup anything complex
try { var started = false
log.info("Run Prometheus metrics on ${monitoringConfig.prometheus.host}:${monitoringConfig.prometheus.port}${monitoringConfig.prometheus.path}") while (true) {
val server = HttpServer.create( if (isTcpPortAvailable(monitoringConfig.prometheus.host, monitoringConfig.prometheus.port)) {
InetSocketAddress( started = true
monitoringConfig.prometheus.host, try {
monitoringConfig.prometheus.port, log.info("Run Prometheus metrics on ${monitoringConfig.prometheus.host}:${monitoringConfig.prometheus.port}${monitoringConfig.prometheus.path}")
), val server = HttpServer.create(
0, InetSocketAddress(
) monitoringConfig.prometheus.host,
server.createContext(monitoringConfig.prometheus.path) { httpExchange -> monitoringConfig.prometheus.port,
val response = prometheusRegistry.scrape() ),
httpExchange.responseHeaders.add("Content-Encoding", "gzip") 0,
httpExchange.responseHeaders.add("Content-Type", "text/plain") )
httpExchange.sendResponseHeaders(200, 0) server.createContext(monitoringConfig.prometheus.path) { httpExchange ->
httpExchange.responseBody.use { os -> val response = prometheusRegistry.scrape()
GZIPOutputStream(os).use { gzos -> httpExchange.responseHeaders.add("Content-Encoding", "gzip")
gzos.write(response.toByteArray()) httpExchange.responseHeaders.add("Content-Type", "text/plain")
httpExchange.sendResponseHeaders(200, 0)
httpExchange.responseBody.use { os ->
GZIPOutputStream(os).use { gzos ->
gzos.write(response.toByteArray())
}
}
} }
Thread(server::start).start()
} catch (e: IOException) {
log.error("Failed to start Prometheus Server", e)
} }
} else {
if (!started) {
log.error("Can't start prometheus metrics on ${monitoringConfig.prometheus.host}:${monitoringConfig.prometheus.port}${monitoringConfig.prometheus.path}")
}
Thread.sleep(1000)
} }
Thread(server::start).start()
} catch (e: IOException) {
log.error("Failed to start Prometheus Server", e)
} }
} }
} }