carrotquest/django-clickhouse
1
1
Fork 0
mirror of https://github.com/carrotquest/django-clickhouse.git synced 2024-11-15 05:46:41 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added query time statsd timer

Browse Source
This commit is contained in:
M1ha 2019-01-23 10:32:11 +05:00
parent 780b13a42e
commit 882930a3a8
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/django_clickhouse/database.py
View File

@ -1,5 +1,5 @@
from infi.clickhouse_orm.database import Database as InfiDatabase, ServerError from infi.clickhouse_orm.database import Database as InfiDatabase
from infi.clickhouse_orm.migrations import MigrationHistory from statsd.defaults.django import statsd
from .configuration import config from .configuration import config
from .exceptions import DBAliasError from .exceptions import DBAliasError
@ -26,6 +26,11 @@ class Database(InfiDatabase):
def _get_applied_migrations(self, migrations_package_name): def _get_applied_migrations(self, migrations_package_name):
raise NotImplementedError("This method is not supported by django_clickhouse.") raise NotImplementedError("This method is not supported by django_clickhouse.")
def _send(self, data, settings=None, stream=False):
statsd_key = "%s.query" % (config.STATSD_PREFIX)
with statsd.timer(statsd_key):
return super(Database, self)._send(data, settings=settings, stream=stream)
class ConnectionProxy: class ConnectionProxy:
_connections = {} _connections = {}

2
src/django_clickhouse/models.py
View File

@ -9,7 +9,7 @@ import six
from django.db import transaction from django.db import transaction
from django.db.models.signals import post_save, post_delete from django.db.models.signals import post_save, post_delete
from django.dispatch import receiver from django.dispatch import receiver
from django.db.models import QuerySet as DjangoQuerySet, Manager as DjangoManager, Model as DjangoModel from django.db.models import QuesrySet as DjangoQuerySet, Manager as DjangoManager, Model as DjangoModel
from .configuration import config from .configuration import config
from .storages import Storage from .storages import Storage