from jsonschema import validate
from jsonschema.exceptions import ValidationError
import connexion
+from prometheus_client import Counter
from mdclogpy import Logger
from ricsdl.exceptions import RejectedByBackend, NotConnected, BackendError
from a1 import a1rmr, exceptions, data
mdc_logger = Logger(name=__name__)
+request_counter = Counter('policy_requests', 'Policy type and instance requests', ['action', 'target'])
def _log_build_http_resp(exception, http_resp_code):
"""
Handles PUT /a1-p/policytypes/policy_type_id
"""
+ request_counter.labels(action='create', target='policy_type').inc()
def put_type_handler():
data.store_policy_type(policy_type_id, body)
"""
Handles DELETE /a1-p/policytypes/policy_type_id
"""
+ request_counter.labels(action='delete', target='policy_type').inc()
def delete_policy_type_handler():
data.delete_policy_type(policy_type_id)
2. if a1 has received at least one status and at least one is OK, we return "IN EFFECT"
3. "NOT IN EFFECT" otherwise (no statuses, or none are OK but not all are deleted)
"""
-
return _try_func_return(lambda: data.get_policy_instance_status(policy_type_id, policy_instance_id))
"""
Handles PUT /a1-p/policytypes/polidyid/policies/policy_instance_id
"""
+ request_counter.labels(action='create', target='policy_inst').inc()
instance = connexion.request.json
def put_instance_handler():
validate(instance=instance, schema=schema)
# store the instance
- data.store_policy_instance(policy_type_id, policy_instance_id, instance)
+ operation = data.store_policy_instance(policy_type_id, policy_instance_id, instance)
# queue rmr send (best effort)
- a1rmr.queue_instance_send(("CREATE", policy_type_id, policy_instance_id, instance))
+ a1rmr.queue_instance_send((operation, policy_type_id, policy_instance_id, instance))
return "", 202
"""
Handles DELETE /a1-p/policytypes/polidyid/policies/policy_instance_id
"""
+ request_counter.labels(action='delete', target='policy_inst').inc()
def delete_instance_handler():
data.delete_policy_instance(policy_type_id, policy_instance_id)
return "", 202
return _try_func_return(delete_instance_handler)
+
+
+# data delivery
+
+
+def data_delivery():
+ """
+ Handle data delivery /data-delivery
+ """
+
+ def data_delivery_handler():
+ mdc_logger.debug("data: {}".format(connexion.request.json))
+ ei_job_result_json = connexion.request.json
+ mdc_logger.debug("jobid: {}".format(ei_job_result_json.get("job")))
+ a1rmr.queue_ei_job_result((ei_job_result_json.get("job"), ei_job_result_json))
+ return "", 200
+
+ return _try_func_return(data_delivery_handler)