pt/superset
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2024-09-12 08:39:45 -04:00
Code Issues Packages Projects Releases Wiki Activity

[druid] Adding cluster filter for refresh

Browse Source
This commit is contained in:
John Bodley 2018-03-12 12:54:32 -07:00
parent 2bc089ef8d
commit 9604e1dd14
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
superset/connectors/druid/models.py
View File

@ -157,16 +157,15 @@ class DruidCluster(Model, AuditMixinNullable, ImportMixin):
def refresh(self, datasource_names, merge_flag, refreshAll): def refresh(self, datasource_names, merge_flag, refreshAll):
""" """
Fetches metadata for the specified datasources andm Fetches metadata for the specified datasources and
merges to the Superset database merges to the Superset database
""" """
session = db.session session = db.session
ds_list = ( ds_list = (
session.query(DruidDatasource) session.query(DruidDatasource)
.filter(or_(DruidDatasource.datasource_name == name .filter(DruidDatasource.cluster_name == self.cluster_name)
for name in datasource_names)) .filter(DruidDatasource.datasource_name.in_(datasource_names))
) )
ds_map = {ds.name: ds for ds in ds_list} ds_map = {ds.name: ds for ds in ds_list}
for ds_name in datasource_names: for ds_name in datasource_names:
datasource = ds_map.get(ds_name, None) datasource = ds_map.get(ds_name, None)