mirror of
https://github.com/apache/superset.git
synced 2026-04-18 23:55:00 +00:00
c552c125d7
* Update wording * nit update for api endpoint url * move metadata cache one layer up * refactor cache * fix flake8 and DatabaseTablesAsync * nit * remove logging for cache * only fetch for all tables that allows cross schema fetch * default allow_multi_schema_metadata_fetch to False * address comments * remove unused defaultdict * flake 8
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
# pylint: disable=C,R,W
|
|
from flask import request
|
|
|
|
from superset import tables_cache
|
|
|
|
|
|
def view_cache_key(*unused_args, **unused_kwargs):
|
|
args_hash = hash(frozenset(request.args.items()))
|
|
return 'view/{}/{}'.format(request.path, args_hash)
|
|
|
|
|
|
def memoized_func(key=view_cache_key, attribute_in_key=None):
|
|
"""Use this decorator to cache functions that have predefined first arg.
|
|
|
|
enable_cache is treated as True by default,
|
|
except enable_cache = False is passed to the decorated function.
|
|
|
|
force means whether to force refresh the cache and is treated as False by default,
|
|
except force = True is passed to the decorated function.
|
|
|
|
timeout of cache is set to 600 seconds by default,
|
|
except cache_timeout = {timeout in seconds} is passed to the decorated function.
|
|
|
|
memoized_func uses simple_cache and stored the data in memory.
|
|
Key is a callable function that takes function arguments and
|
|
returns the caching key.
|
|
"""
|
|
def wrap(f):
|
|
if tables_cache:
|
|
def wrapped_f(self, *args, **kwargs):
|
|
if not kwargs.get('cache', True):
|
|
return f(self, *args, **kwargs)
|
|
|
|
if attribute_in_key:
|
|
cache_key = key(*args, **kwargs).format(
|
|
getattr(self, attribute_in_key))
|
|
else:
|
|
cache_key = key(*args, **kwargs)
|
|
o = tables_cache.get(cache_key)
|
|
if not kwargs.get('force') and o is not None:
|
|
return o
|
|
o = f(self, *args, **kwargs)
|
|
tables_cache.set(cache_key, o,
|
|
timeout=kwargs.get('cache_timeout'))
|
|
return o
|
|
else:
|
|
# noop
|
|
def wrapped_f(self, *args, **kwargs):
|
|
return f(self, *args, **kwargs)
|
|
return wrapped_f
|
|
return wrap
|