58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
import logging
|
|
import sys
|
|
from threading import Thread
|
|
from django.conf import settings
|
|
import requests
|
|
|
|
|
|
class LokiHandler(logging.Handler):
|
|
def __init__(
|
|
self,
|
|
timeout=0.5,
|
|
# url="http://localhost:3100/loki/api/v1/push",
|
|
url=f'{settings.LOKI_BASE_PUBLIC_URL}/api/v1/push',
|
|
auth=None,
|
|
tags={},
|
|
mode="sync",
|
|
):
|
|
super(LokiHandler, self).__init__()
|
|
|
|
self._url = url
|
|
self._timeout = timeout
|
|
self._auth = auth
|
|
self._tags = tags
|
|
|
|
self._mode = mode
|
|
|
|
def emit(self, record):
|
|
try:
|
|
payload = self.formatter.format(record)
|
|
|
|
self._push_message(
|
|
self._url, json=payload, timeout=self._timeout, auth=self._auth
|
|
)
|
|
except requests.exceptions.ReadTimeout:
|
|
sys.stderr.write("Loki connection timed out\n")
|
|
except Exception as e:
|
|
sys.stderr.write(f"Loki connection failed with: {e}\n")
|
|
|
|
def setFormatter(self, fmt):
|
|
fmt.tags = self._tags
|
|
|
|
self.formatter = fmt
|
|
|
|
def _push_message(self, *args, **kwargs):
|
|
if self._mode == "sync":
|
|
return _push_message(*args, **kwargs)
|
|
|
|
if self._mode == "thread":
|
|
return Thread(target=_push_message, args=args, kwargs=kwargs).start()
|
|
|
|
|
|
def _push_message(*args, **kwargs):
|
|
response = requests.post(*args, **kwargs)
|
|
|
|
if response.status_code != 204:
|
|
sys.stderr.write(
|
|
f"Got status {response.status_code} from loki with message: {response.text}\n"
|
|
)
|