From ef6fc46c5c1e15071f1703f5ff19aad3b2a1dfad Mon Sep 17 00:00:00 2001 From: M1ha Date: Tue, 5 Feb 2019 15:13:15 +0500 Subject: [PATCH] Measure time for inserting tuples --- src/django_clickhouse/database.py | 5 +++-- tests/clickhouse_models.py | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/django_clickhouse/database.py b/src/django_clickhouse/database.py index 03d5d1b..69f9de3 100644 --- a/src/django_clickhouse/database.py +++ b/src/django_clickhouse/database.py @@ -113,10 +113,11 @@ class Database(InfiDatabase): # Return any remaining lines in partial batch if lines: - statsd.incr(statsd_key.format('insert_batch'), count=lines) + statsd.incr(statsd_key, lines) yield buf.getvalue() - self._send(gen()) + with statsd.timer(statsd_key): + self._send(next(gen())) class ConnectionProxy: diff --git a/tests/clickhouse_models.py b/tests/clickhouse_models.py index 32b5450..32212ba 100644 --- a/tests/clickhouse_models.py +++ b/tests/clickhouse_models.py @@ -27,7 +27,7 @@ class ClickHouseCollapseTestModel(ClickHouseModel): id = fields.Int32Field() created = fields.DateTimeField() value = fields.Int32Field() - sign = fields.Int8Field() + sign = fields.Int8Field(default=1) version = fields.Int8Field(default=1) engine = CollapsingMergeTree('created', ('id',), 'sign')