github/FlareSolverr
1
1
Fork 0
mirror of https://github.com/FlareSolverr/FlareSolverr.git synced 2025-06-10 05:17:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix URL domain in Prometheus exporter

Browse Source
This commit is contained in:
ngosang 2023-08-03 09:02:46 +02:00
parent 644a843d89
commit b5eef32615
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/bottle_plugins/prometheus_plugin.py
View File

@ -2,7 +2,8 @@ import logging
import os import os
import urllib.parse import urllib.parse
from dtos import V1ResponseBase from bottle import request
from dtos import V1RequestBase, V1ResponseBase
from metrics import start_metrics_http_server, REQUEST_COUNTER, REQUEST_DURATION from metrics import start_metrics_http_server, REQUEST_COUNTER, REQUEST_DURATION
PROMETHEUS_ENABLED = os.environ.get('PROMETHEUS_ENABLED', 'false').lower() == 'true' PROMETHEUS_ENABLED = os.environ.get('PROMETHEUS_ENABLED', 'false').lower() == 'true'
@ -39,8 +40,12 @@ def prometheus_plugin(callback):
domain = "unknown" domain = "unknown"
if res.solution and res.solution.url: if res.solution and res.solution.url:
parsed_url = urllib.parse.urlparse(res.solution.url) domain = parse_domain_url(res.solution.url)
domain = parsed_url.hostname else:
# timeout error
req = V1RequestBase(request.json)
if req.url:
domain = parse_domain_url(req.url)
run_time = (res.endTimestamp - res.startTimestamp) / 1000 run_time = (res.endTimestamp - res.startTimestamp) / 1000
REQUEST_DURATION.labels(domain=domain).observe(run_time) REQUEST_DURATION.labels(domain=domain).observe(run_time)
@ -54,4 +59,8 @@ def prometheus_plugin(callback):
result = "error" result = "error"
REQUEST_COUNTER.labels(domain=domain, result=result).inc() REQUEST_COUNTER.labels(domain=domain, result=result).inc()
def parse_domain_url(url):
parsed_url = urllib.parse.urlparse(url)
return parsed_url.hostname
return wrapper return wrapper