X-Git-Url: https://osm.etsi.org/gitweb/?p=osm%2FNBI.git;a=blobdiff_plain;f=osm_nbi%2Fengine.py;h=b90f713426f0a7b1791b3c52f0bd3a54836dbd47;hp=38cc08e0b327600f062a06e7f180a4de404a2d57;hb=f0637057dc1be391f068c7e2b9c8f81b16f0921e;hpb=55ba2e6f758ae650a2dbbc20638c0108df726809 diff --git a/osm_nbi/engine.py b/osm_nbi/engine.py index 38cc08e..b90f713 100644 --- a/osm_nbi/engine.py +++ b/osm_nbi/engine.py @@ -25,9 +25,10 @@ from descriptor_topics import VnfdTopic, NsdTopic, PduTopic, NstTopic from instance_topics import NsrTopic, VnfrTopic, NsLcmOpTopic, NsiTopic, NsiLcmOpTopic from base64 import b64encode from os import urandom +from threading import Lock __author__ = "Alfonso Tierno " -min_common_version = "0.1.8" +min_common_version = "0.1.16" class Engine(object): @@ -56,6 +57,7 @@ class Engine(object): self.config = None self.logger = logging.getLogger("nbi.engine") self.map_topic = {} + self.write_lock = None def start(self, config): """ @@ -96,8 +98,9 @@ class Engine(object): self.msg.connect(config["message"]) else: raise EngineException("Invalid configuration param '{}' at '[message]':'driver'".format( - config["storage"]["driver"])) + config["message"]["driver"])) + self.write_lock = Lock() # create one class per topic for topic, topic_class in self.map_from_topic_to_class.items(): self.map_topic[topic] = topic_class(self.db, self.fs, self.msg) @@ -110,8 +113,9 @@ class Engine(object): self.db.db_disconnect() if self.fs: self.fs.fs_disconnect() - if self.fs: - self.fs.fs_disconnect() + if self.msg: + self.msg.disconnect() + self.write_lock = None except (DbException, FsException, MsgException) as e: raise EngineException(str(e), http_code=e.http_code) @@ -130,7 +134,8 @@ class Engine(object): """ if topic not in self.map_topic: raise EngineException("Unknown topic {}!!!".format(topic), HTTPStatus.INTERNAL_SERVER_ERROR) - return self.map_topic[topic].new(rollback, session, indata, kwargs, headers, force) + with self.write_lock: + return self.map_topic[topic].new(rollback, session, indata, kwargs, headers, force) def upload_content(self, session, topic, _id, indata, kwargs, headers, force=False): """ @@ -146,7 +151,8 @@ class Engine(object): """ if topic not in self.map_topic: raise EngineException("Unknown topic {}!!!".format(topic), HTTPStatus.INTERNAL_SERVER_ERROR) - return self.map_topic[topic].upload_content(session, _id, indata, kwargs, headers, force) + with self.write_lock: + return self.map_topic[topic].upload_content(session, _id, indata, kwargs, headers, force) def get_item_list(self, session, topic, filter_q=None): """ @@ -196,7 +202,8 @@ class Engine(object): """ if topic not in self.map_topic: raise EngineException("Unknown topic {}!!!".format(topic), HTTPStatus.INTERNAL_SERVER_ERROR) - return self.map_topic[topic].delete_list(session, _filter) + with self.write_lock: + return self.map_topic[topic].delete_list(session, _filter) def del_item(self, session, topic, _id, force=False): """ @@ -209,7 +216,8 @@ class Engine(object): """ if topic not in self.map_topic: raise EngineException("Unknown topic {}!!!".format(topic), HTTPStatus.INTERNAL_SERVER_ERROR) - return self.map_topic[topic].delete(session, _id, force) + with self.write_lock: + return self.map_topic[topic].delete(session, _id, force) def edit_item(self, session, topic, _id, indata=None, kwargs=None, force=False): """ @@ -224,7 +232,8 @@ class Engine(object): """ if topic not in self.map_topic: raise EngineException("Unknown topic {}!!!".format(topic), HTTPStatus.INTERNAL_SERVER_ERROR) - return self.map_topic[topic].edit(session, _id, indata, kwargs, force) + with self.write_lock: + return self.map_topic[topic].edit(session, _id, indata, kwargs, force) def create_admin(self): """ @@ -259,6 +268,7 @@ class Engine(object): } self.db.create("admin", version_data) self.db.set_secret_key(serial) + return # TODO add future migrations here raise EngineException("Wrong database version '{}'. Expected '{}'"