diff options
Diffstat (limited to 'share/common/utils')
| -rw-r--r-- | share/common/utils/aai_cache.py | 53 | ||||
| -rw-r--r-- | share/common/utils/restcall.py | 22 |
2 files changed, 72 insertions, 3 deletions
diff --git a/share/common/utils/aai_cache.py b/share/common/utils/aai_cache.py new file mode 100644 index 00000000..53298bb8 --- /dev/null +++ b/share/common/utils/aai_cache.py @@ -0,0 +1,53 @@ +# 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: + cache.delete("AAI_" + resource_url) + 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) + return json.loads(value) if value else None + else: + return None + except: + 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: + pass + +def filter_cache_by_url(resource_url): + # hardcoded filter: cloud region only + if resource_url.find(r"cloud-infrastructure/cloud-regions/cloud-region") > 0: + return True + else: + return False
\ No newline at end of file diff --git a/share/common/utils/restcall.py b/share/common/utils/restcall.py index ec3abb30..eb4cb008 100644 --- a/share/common/utils/restcall.py +++ b/share/common/utils/restcall.py @@ -25,6 +25,8 @@ import uuid from rest_framework import status from django.conf import settings +from common.utils import aai_cache + rest_no_auth, rest_oneway_auth, rest_bothway_auth = 0, 1, 2 HTTP_200_OK, HTTP_201_CREATED = '200', '201' HTTP_204_NO_CONTENT, HTTP_202_ACCEPTED = '204', '202' @@ -124,7 +126,7 @@ def req_to_vim(base_url, resource, method, extra_headers='', content=''): resource, method, extra_headers, content) -def req_to_aai(resource, method, content='', appid=settings.MULTICLOUD_APP_ID): +def req_to_aai(resource, method, content='', appid=settings.MULTICLOUD_APP_ID, nocache=False): tmp_trasaction_id = '9003' #str(uuid.uuid1()) headers = { 'X-FromAppId': appid, @@ -133,8 +135,22 @@ def req_to_aai(resource, method, content='', appid=settings.MULTICLOUD_APP_ID): 'accept': 'application/json' } - return _call_req(settings.AAI_BASE_URL, settings.AAI_USERNAME, settings.AAI_PASSWORD, rest_no_auth, - resource, method, content=json.dumps(content), extra_headers=headers) + # hook to flush cache + if method.upper() in ["PUT", "POST", "PATCH", "DELETE"]: + aai_cache.flush_cache_by_url(resource) + elif method.upper in ["GET"] and not nocache: + content = aai_cache.get_cache_by_url(resource) + if content: + return content + + ret, resp_body, resp_status = _call_req( + settings.AAI_BASE_URL, settings.AAI_USERNAME, settings.AAI_PASSWORD, rest_no_auth, + resource, method, content=json.dumps(content), extra_headers=headers) + + if method.upper() in ["GET"] and ret == 0 and not nocache: + aai_cache.set_cache_by_url(resource, [ret, resp_body, resp_status]) + + return [ret, resp_body, resp_status] def _combine_url(base_url, resource): |