Merge pull request #141 from meaningfy-ws/feature/TED-405

added indexes to mongo

Author: Dragos0000

ted_sws/data_manager/adapters/mapping_suite_repository.py

@@ -4,7 +4,7 @@
 import shutil
 from typing import Iterator, List, Optional
 
-from pymongo import MongoClient
+from pymongo import MongoClient, ASCENDING
 
 from ted_sws import config
 from ted_sws.core.model.transform import MappingSuite, FileResource, TransformationRuleSet, SHACLTestSuite, \
@@ -66,7 +66,7 @@ def get(self, reference) -> MappingSuite:
         :param reference:
         :return: MappingSuite
         """
-        result_dict = self.collection.find_one({"identifier": reference})
+        result_dict = self.collection.find_one({"_id": reference})
         return MappingSuite(**result_dict) if result_dict else None
 
     def list(self) -> Iterator[MappingSuite]:

ted_sws/data_manager/adapters/notice_repository.py

@@ -3,7 +3,7 @@
 from typing import Iterator, Union, Optional
 
 import gridfs
-from pymongo import MongoClient
+from pymongo import MongoClient, ASCENDING
 from bson import ObjectId
 from ted_sws import config
 from ted_sws.core.model.manifestation import XMLManifestation, RDFManifestation, METSManifestation, Manifestation
@@ -28,6 +28,8 @@ def __init__(self, mongodb_client: MongoClient, database_name: str = _database_n
         notice_db = mongodb_client[self._database_name]
         self.file_storage = gridfs.GridFS(notice_db)
         self.collection = notice_db[self._collection_name]
+        self.file_storage_collection = notice_db["fs.files"]
+        self.file_storage_collection.create_index([("notice_id", ASCENDING)])
 
     def get_file_content_from_grid_fs(self, file_id: str) -> str:
         """
@@ -188,7 +190,7 @@ def get(self, reference) -> Optional[Notice]:
         :param reference:
         :return: Notice
         """
-        result_dict = self.collection.find_one({"ted_id": reference})
+        result_dict = self.collection.find_one({"_id": reference})
         if result_dict is not None:
             notice = NoticeRepository._create_notice_from_repository_result(result_dict)
             notice = self.load_notice_fields_from_grid_fs(notice)