Skip to content

Commit

Permalink
[fix] Fixing cache key inconsistencies
Browse files Browse the repository at this point in the history
  • Loading branch information
john-bodley committed Mar 28, 2020
1 parent 499f9c8 commit a96db61
Show file tree
Hide file tree
Showing 2 changed files with 99 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -0,0 +1,98 @@
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
"""cleanup_time_grainularity
Revision ID: f9a30386bd74
Revises: b5998378c225
Create Date: 2020-03-25 10:42:11.047328
"""

# revision identifiers, used by Alembic.
revision = "f9a30386bd74"
down_revision = "b5998378c225"

import json

from alembic import op
from sqlalchemy import Column, Integer, String, Text
from sqlalchemy.ext.declarative import declarative_base

from superset import db

Base = declarative_base()


class Slice(Base):
__tablename__ = "slices"

id = Column(Integer, primary_key=True)
params = Column(Text)
viz_type = Column(String(250))


def upgrade():
"""
Remove any erroneous time grainularity fields from slices foor those visualization
types which do not support time granularity.
:see: https://github.com/apache/incubator-superset/pull/8674
:see: https://github.com/apache/incubator-superset/pull/8764
:see: https://github.com/apache/incubator-superset/pull/8800
:see: https://github.com/apache/incubator-superset/pull/8825
"""

bind = op.get_bind()
session = db.Session(bind=bind)

# Visualization types which support time grainularity (hence negate).
viz_types = [
"area",
"bar",
"big_number",
"compare",
"dual_line",
"line",
"pivot_table",
"table",
"time_pivot",
"time_table",
]

# Erroneous time grainularity fields for either Druid NoSQL or SQL slices which do
# not support time grainularity.
erroneous = ["grainularity", "time_grain_sqla"]

for slc in session.query(Slice).filter(Slice.viz_type.notin_(viz_types)).all():
try:
params = json.loads(slc.params)

if any(field in params for field in erroneous):
for field in erroneous:
if field in params:
del params[field]

slc.params = json.dumps(params, sort_keys=True)
except Exception:
pass

session.commit()
session.close()


def downgrade():
pass
2 changes: 1 addition & 1 deletion superset/viz.py
Original file line number Diff line number Diff line change
Expand Up @@ -330,7 +330,7 @@ def query_obj(self) -> Dict[str, Any]:
"druid_time_origin": form_data.get("druid_time_origin", ""),
"having": form_data.get("having", ""),
"having_druid": form_data.get("having_filters", []),
"time_grain_sqla": form_data.get("time_grain_sqla", ""),
"time_grain_sqla": form_data.get("time_grain_sqla"),
"time_range_endpoints": form_data.get("time_range_endpoints"),
"where": form_data.get("where", ""),
}
Expand Down

0 comments on commit a96db61

Please sign in to comment.