summaryrefslogtreecommitdiffstats
path: root/share/common/utils/aai_cache.py
blob: c2e8e034802b1f588f44db8a7c9641f5e5c4b8ce (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
# Copyright (c) 2017-2018 Wind River Systems, Inc.
#
# Licensed 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.

import json
import logging
from django.core.cache import cache

logger = logging.getLogger(__name__)

# note: memcached key length should be < 250, the value < 1MB


def flush_cache_by_url(resource_url):
    try:
        # this is to invalidate similar cache data blindly
        resource_wo_query = resource_url.split("?", 1)[0]
        cache.delete("AAI_" + resource_wo_query)
    except:
        pass  # silently drop any exception


def get_cache_by_url(resource_url):
    try:
        if filter_cache_by_url(resource_url):
            value = cache.get("AAI_" + resource_url)
            # logger.debug("Find cache the resource: %s, %s" %( resource_url, value))
            return json.loads(value) if value else None
        else:
            return None
    except Exception as e:
        logger.error("get_cache_by_url exception: %s" % str(e))
        return None


def set_cache_by_url(resource_url, resource_in_json):
    try:
        # filter out unmanaged AAI resource
        if filter_cache_by_url(resource_url):
            # cache the resource for 24 hours
            # logger.debug("Cache the resource: "+ resource_url)
            cache.set("AAI_" + resource_url,
                      json.dumps(resource_in_json), 3600 * 24)
    except Exception as e:
        logger.error("get_cache_by_url exception: %s" % str(e))


def filter_cache_by_url(resource_url):
    # cannot cache url with query string, e.g. depth=all
    if resource_url.find(r"?") >= 0:
        return False
    # hardcoded filter: cloud region only
    if resource_url.find(r"cloud-infrastructure/cloud-regions/cloud-region") >= 0:
        return True
    else:
        return False