Fix db access within writer

Author: hugomanguinhas

record-api-migration/src/main/java/eu/europeana/api/record/migration/DataSourceConfig.java

@@ -32,6 +32,8 @@
 import com.mongodb.client.MongoClients;
 import com.mongodb.connection.ConnectionPoolSettings;
 
+import eu.europeana.api.record.db.config.DataSourceConfig;
+
 import org.apache.commons.cli.Options;
 import org.apache.commons.cli.ParseException;
 

record-api-migration/src/main/java/eu/europeana/api/record/migration/MigrationCommand.java

@@ -3,6 +3,7 @@
 import dev.morphia.query.filters.Filters;
 import eu.europeana.api.edm.EDM;
 import eu.europeana.api.format.RdfFormat;
+import eu.europeana.api.record.db.repository.RecordRepository;
 import eu.europeana.api.record.io.FormatHandlerRegistry;
 import eu.europeana.api.record.migration.RecordDomProcessor.Result;
 import eu.europeana.api.record.model.*;
@@ -67,14 +68,14 @@ public class MigrationHandler {
 
     private final FormatHandlerRegistry registry;
 
-    private final MigrationRepository migrationRepository;
+    private final RecordRepository migrationRepository;
 
     private final TemplateLibrary library;
 
     @Autowired
     public MigrationHandler(MigrationSettings settings
                           , FormatHandlerRegistry registry
-                          , MigrationRepository migrationRepository
+                          , RecordRepository migrationRepository
                           , @Qualifier(BEAN_RECORD_TEMPLATE_LIBRARY) TemplateLibrary library) {
         this.settings = settings;
         this.registry = registry;

record-api-migration/src/main/java/eu/europeana/api/record/migration/MigrationHandler.java

@@ -56,7 +56,6 @@ public class RecordRepository implements ObjectRepository {
      * @return saved record
      */
     public ProvidedCHO save(ProvidedCHO record) {
-
         UpdateOptions upsert = new UpdateOptions().upsert(true);
         Filter idf = Filters.eq("id", record.getID());
 

record-api-migration/src/main/java/eu/europeana/api/record/migration/MigrationRepository.java

@@ -1,5 +1,6 @@
 package eu.europeana.api.record.service;
 
+import dev.morphia.internal.DatastoreHolder;
 import dev.morphia.query.FindOptions;
 import dev.morphia.query.MorphiaCursor;
 import eu.europeana.api.record.config.AppConfig;
@@ -25,6 +26,10 @@ public RecordService(RecordRepository recordRepository, RecordApiConfiguration r
         this.recordApiConfiguration = recordApiConfiguration;
     }
 
+    public void init() {
+        DatastoreHolder.holder.set(recordRepository.getDatastore());
+    }
+
     public ProvidedCHO saveRecord(ProvidedCHO providedCHO) {
         return recordRepository.save(providedCHO);
     }

record-api-mongo/src/main/java/eu/europeana/api/record/db/repository/RecordRepository.java

@@ -1,5 +1,6 @@
 package eu.europeana.api.record.web;
 
+import dev.morphia.internal.DatastoreHolder;
 import dev.morphia.query.MorphiaCursor;
 import eu.europeana.api.commons.web.http.HttpHeaders;
 import eu.europeana.api.error.EuropeanaApiException;
@@ -140,6 +141,8 @@ private ResponseEntity<StreamingResponseBody> createResponse(String datasetId, S
         StreamingResponseBody responseBody = new StreamingResponseBody() {
             @Override
             public void writeTo(OutputStream out) throws IOException {
+                //this is needed because Jackson serialises the response in a new thread
+                recordService.init();
                 formatHandlerRegistry.get(recordRequest.getRdfFormat()).write(providedCHO, out);
                 out.flush();
             }
@@ -161,6 +164,8 @@ private ResponseEntity<StreamingResponseBody> createResponseMultipleRecords(List
             @Override
             public void writeTo(OutputStream out) throws IOException {
                 try {
+                   //this is needed because Jackson serialises the response in a new thread
+                    recordService.init(); 
                     formatHandlerRegistry.get(RdfFormat.JSONLD).write(records, records.available(), out);
                     out.flush();
                 }

record-api-web/src/main/java/eu/europeana/api/record/service/RecordService.java

@@ -1,7 +1,7 @@
 ## Record API properties file
 
 # mongo url
-mongo.connectionUrl=
+mongo.connectionUrl=mongodb://recorduser1:wRfis830Lpdjw%[email protected]/record??compressors=snappy&serverSelectionTimeoutMS=25000&connectTimeoutMS=6000&socketTimeoutMS=35000&directConnection=true&tls=false
 mongo.record.database=record