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

Chaned statsd key for inserted tuples

Browse Source
This commit is contained in:
M1ha 2019-02-04 12:52:21 +05:00
parent 2128dd0efc
commit a205b451a0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/django_clickhouse/database.py
View File

@ -85,7 +85,7 @@ class Database(InfiDatabase):
fields_list = ','.join('`%s`' % name for name in first_tuple._fields) fields_list = ','.join('`%s`' % name for name in first_tuple._fields)
fields_dict = model_class.fields(writable=True) fields_dict = model_class.fields(writable=True)
fields = [fields_dict[name] for name in first_tuple._fields] fields = [fields_dict[name] for name in first_tuple._fields]
statsd_key = "%s.inserted_tuples.%s.{0}" % (config.STATSD_PREFIX, model_class.__name__) statsd_key = "%s.inserted_tuples.%s" % (config.STATSD_PREFIX, model_class.__name__)
def tuple_to_csv(tup): def tuple_to_csv(tup):
return '\t'.join(field.to_db_string(val, quote=False) for field, val in zip(fields, tup)) + '\n' return '\t'.join(field.to_db_string(val, quote=False) for field, val in zip(fields, tup)) + '\n'
@ -105,7 +105,7 @@ class Database(InfiDatabase):
lines += 1 lines += 1
if batch_size is not None and lines >= batch_size: if batch_size is not None and lines >= batch_size:
# Return the current batch of lines # Return the current batch of lines
statsd.incr(statsd_key.format('insert_batch'), lines) statsd.incr(statsd_key, lines)
yield buf.getvalue() yield buf.getvalue()
# Start a new batch # Start a new batch
buf = BytesIO() buf = BytesIO()