From e628d07247f788d20c06e8d310070a5bc66917f7 Mon Sep 17 00:00:00 2001
From: Bin Yang <bin.yang@windriver.com>
Date: Tue, 25 Dec 2018 10:42:52 +0000
Subject: Refactor the AAI cache for multicloud plugin

Fix the cache issue of cloud region which result in failure of
cloud region registration

Change-Id: Id13a40124efb92bd818686e069c9335ecd0d07c1
Issue-ID: MULTICLOUD-431
Signed-off-by: Bin Yang <bin.yang@windriver.com>
---
 share/common/utils/aai_cache.py | 53 +++++++++++++++++++++++++++++++++++++++++
 share/common/utils/restcall.py  | 22 ++++++++++++++---
 2 files changed, 72 insertions(+), 3 deletions(-)
 create mode 100644 share/common/utils/aai_cache.py

(limited to 'share/common/utils')

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):
-- 
cgit 1.2.3-korg