A1 2.0.0:

[ric-plt/a1.git] / a1 / controller.py

diff --git a/a1/controller.py b/a1/controller.py
index 75d8456..cde7483 100644 (file)
--- a/a1/controller.py
+++ b/a1/controller.py
@@ -17,16 +17,15 @@ Main a1 controller
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
 # ==================================================================================
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
 # ==================================================================================

-import json

 from flask import Response
 from jsonschema import validate
 from jsonschema.exceptions import ValidationError
 import connexion
 from flask import Response
 from jsonschema import validate
 from jsonschema.exceptions import ValidationError
 import connexion

-from a1 import get_module_logger
+from mdclogpy import Logger

 from a1 import a1rmr, exceptions, data
 
 
 from a1 import a1rmr, exceptions, data
 
 

-logger = get_module_logger(__name__)
+mdc_logger = Logger(name=__name__)

 
 
 def _try_func_return(func):
 
 
 def _try_func_return(func):


@@ -41,22 +40,10 @@ def _try_func_return(func):
         return "", 404
     except BaseException as exc:
         # catch all, should never happen...
         return "", 404
     except BaseException as exc:
         # catch all, should never happen...

-        logger.exception(exc)
+        mdc_logger.exception(exc)

         return Response(status=500)
 
 
         return Response(status=500)
 
 

-def _gen_body_to_handler(operation, policy_type_id, policy_instance_id, payload=None):
-    """
-    used to create the payloads that get sent to downstream policy handlers
-    """
-    return {
-        "operation": operation,
-        "policy_type_id": policy_type_id,
-        "policy_instance_id": policy_instance_id,
-        "payload": payload,
-    }
-
-

 # Healthcheck
 
 
 # Healthcheck
 
 


@@ -142,14 +129,7 @@ def get_policy_instance_status(policy_type_id, policy_instance_id):
         3. "NOT IN EFFECT" otherwise (no statuses, or none are OK but not all are deleted)
     """
 
         3. "NOT IN EFFECT" otherwise (no statuses, or none are OK but not all are deleted)
     """
 

-    def get_status_handler():
-        vector = data.get_policy_instance_statuses(policy_type_id, policy_instance_id)
-        for i in vector:
-            if i == "OK":
-                return "IN EFFECT", 200
-        return "NOT IN EFFECT", 200
-
-    return _try_func_return(get_status_handler)
+    return _try_func_return(lambda: data.get_policy_instance_status(policy_type_id, policy_instance_id))

 
 
 def create_or_replace_policy_instance(policy_type_id, policy_instance_id):
 
 
 def create_or_replace_policy_instance(policy_type_id, policy_instance_id):


@@ -171,9 +151,8 @@ def create_or_replace_policy_instance(policy_type_id, policy_instance_id):
         # store the instance
         data.store_policy_instance(policy_type_id, policy_instance_id, instance)
 
         # store the instance
         data.store_policy_instance(policy_type_id, policy_instance_id, instance)
 

-        # send rmr (best effort)
-        body = _gen_body_to_handler("CREATE", policy_type_id, policy_instance_id, payload=instance)
-        a1rmr.queue_work({"payload": json.dumps(body), "msg type": policy_type_id})
+        # queue rmr send (best effort)
+        a1rmr.queue_instance_send(("CREATE", policy_type_id, policy_instance_id, instance))

 
         return "", 202
 
 
         return "", 202
 


@@ -188,12 +167,12 @@ def delete_policy_instance(policy_type_id, policy_instance_id):
     def delete_instance_handler():
         """
         here we send out the DELETEs but we don't delete the instance until a GET is called where we check the statuses
     def delete_instance_handler():
         """
         here we send out the DELETEs but we don't delete the instance until a GET is called where we check the statuses

+        We also set the status as deleted which would be reflected in a GET to ../status (before the DELETE completes)

         """
         """

-        data.instance_is_valid(policy_type_id, policy_instance_id)
+        data.delete_policy_instance(policy_type_id, policy_instance_id)

 
 

-        # send rmr (best effort)
-        body = _gen_body_to_handler("DELETE", policy_type_id, policy_instance_id)
-        a1rmr.queue_work({"payload": json.dumps(body), "msg type": policy_type_id})
+        # queue rmr send (best effort)
+        a1rmr.queue_instance_send(("DELETE", policy_type_id, policy_instance_id, ""))

 
         return "", 202
 
 
         return "", 202