mirror of
https://github.com/PostHog/posthog.git
synced 2024-11-24 18:07:17 +01:00
5c9baf6c72
* chore: basic retry logic for booting up the application and waiting for unit * finally block for close
89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
import http.client
|
|
import json
|
|
import time
|
|
import logging
|
|
from prometheus_client import CollectorRegistry, Gauge, multiprocess, generate_latest
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
UNIT_CONNECTIONS_ACCEPTED_TOTAL = Gauge(
|
|
"unit_connections_accepted_total",
|
|
"",
|
|
multiprocess_mode="livesum",
|
|
)
|
|
UNIT_CONNECTIONS_ACTIVE = Gauge(
|
|
"unit_connections_active",
|
|
"",
|
|
multiprocess_mode="livesum",
|
|
)
|
|
UNIT_CONNECTIONS_CLOSED = Gauge(
|
|
"unit_connections_closed",
|
|
"",
|
|
multiprocess_mode="livesum",
|
|
)
|
|
UNIT_CONNECTIONS_IDLE = Gauge(
|
|
"unit_connections_idle",
|
|
"",
|
|
multiprocess_mode="livesum",
|
|
)
|
|
UNIT_CONNECTIONS_TOTAL = Gauge(
|
|
"unit_requests_total",
|
|
"",
|
|
multiprocess_mode="livesum",
|
|
)
|
|
UNIT_PROCESSES_RUNNING_GAUGE = Gauge(
|
|
"unit_application_processes_running", "", multiprocess_mode="livesum", labelnames=["application"]
|
|
)
|
|
UNIT_PROCESSES_STARTING_GAUGE = Gauge(
|
|
"unit_application_processes_starting", "", multiprocess_mode="livesum", labelnames=["application"]
|
|
)
|
|
UNIT_PROCESSES_IDLE_GAUGE = Gauge(
|
|
"unit_application_processes_idle", "", multiprocess_mode="livesum", labelnames=["application"]
|
|
)
|
|
UNIT_REQUESTS_ACTIVE_GAUGE = Gauge(
|
|
"unit_application_requests_active", "", multiprocess_mode="livesum", labelnames=["application"]
|
|
)
|
|
|
|
|
|
def application(environ, start_response):
|
|
retries = 5
|
|
try:
|
|
connection = http.client.HTTPConnection("localhost:8181")
|
|
connection.request("GET", "/status")
|
|
response = connection.getresponse()
|
|
|
|
statj = json.loads(response.read())
|
|
except Exception as e:
|
|
if retries > 0:
|
|
retries -= 1
|
|
time.sleep(1)
|
|
return application(environ, start_response)
|
|
else:
|
|
raise e
|
|
finally:
|
|
try:
|
|
connection.close()
|
|
except Exception as e:
|
|
logger.error("Failed to close connection to unit: ", e)
|
|
|
|
UNIT_CONNECTIONS_ACCEPTED_TOTAL.set(statj["connections"]["accepted"])
|
|
UNIT_CONNECTIONS_ACTIVE.set(statj["connections"]["active"])
|
|
UNIT_CONNECTIONS_IDLE.set(statj["connections"]["idle"])
|
|
UNIT_CONNECTIONS_CLOSED.set(statj["connections"]["closed"])
|
|
UNIT_CONNECTIONS_TOTAL.set(statj["requests"]["total"])
|
|
|
|
for app in statj["applications"].keys():
|
|
UNIT_PROCESSES_RUNNING_GAUGE.labels(application=app).set(statj["applications"][app]["processes"]["running"])
|
|
UNIT_PROCESSES_STARTING_GAUGE.labels(application=app).set(statj["applications"][app]["processes"]["starting"])
|
|
UNIT_PROCESSES_IDLE_GAUGE.labels(application=app).set(statj["applications"][app]["processes"]["idle"])
|
|
UNIT_REQUESTS_ACTIVE_GAUGE.labels(application=app).set(statj["applications"][app]["requests"]["active"])
|
|
|
|
start_response("200 OK", [("Content-Type", "text/plain")])
|
|
# Create the prometheus multi-process metric registry here
|
|
# This will aggregate metrics we send from the Django app
|
|
# We prepend our unit metrics here.
|
|
registry = CollectorRegistry()
|
|
multiprocess.MultiProcessCollector(registry)
|
|
yield generate_latest(registry)
|