Infinidat/infi.clickhouse_orm
1
1
Fork 0
mirror of https://github.com/Infinidat/infi.clickhouse_orm.git synced 2025-08-02 11:10:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

Support FINAL on distributed tables

Browse Source 
Support the FINAL modifier on distributed collapsing- and replacingmergetree tables.
This commit is contained in:
Christian Pedersen 2020-12-15 09:41:16 +01:00 committed by GitHub
parent 7c90c1e4c3
commit fedd6c6020
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/infi/clickhouse_orm/query.py
View File

@ -539,8 +539,20 @@ class QuerySet(object):
Adds a FINAL modifier to table, meaning data will be collapsed to final version.
Can be used with the `CollapsingMergeTree` and `ReplacingMergeTree` engines only.
"""
from .engines import CollapsingMergeTree, ReplacingMergeTree
if not isinstance(self._model_cls.engine, (CollapsingMergeTree, ReplacingMergeTree)):
from .engines import CollapsingMergeTree, ReplacingMergeTree, Distributed
_engine = self._model_cls.engine
if isinstance(_engine, (Distributed)):
# find the "real" engine for the Distributed table
children = self._model_cls.mro()
for child in children:
# we asume the first that is not Distributed is the child.
if isinstance(child.engine, (Distributed)): continue
_engine = child.engine
break
if not isinstance(_engine, (CollapsingMergeTree, ReplacingMergeTree)):
raise TypeError('final() method can be used only with the CollapsingMergeTree and ReplacingMergeTree engines')
qs = copy(self)