Merge branch 'MEMO-669-improve-mimetype-handling' into 'master'

Memo 669: improve mimetype handling

See merge request !11

src/integrationTest/kotlin/ServiceTest.kt

@@ -17,6 +17,7 @@
  */
 package org.memobase
 
+import com.beust.klaxon.Klaxon
 import java.io.File
 import java.io.FileInputStream
 import java.time.Duration
@@ -35,6 +36,7 @@ import org.apache.kafka.common.serialization.StringDeserializer
 import org.apache.kafka.common.serialization.StringSerializer
 import org.apache.logging.log4j.LogManager
 import org.assertj.core.api.Assertions.assertThat
+import org.junit.jupiter.api.Assertions.assertNotNull
 import org.junit.jupiter.api.TestInstance
 import org.junit.jupiter.api.extension.ExtendWith
 import org.junit.jupiter.params.ParameterizedTest
@@ -122,7 +124,9 @@ class ServiceTest {
         assertThat(totalConsumerRecords)
             .size().isEqualTo(expectedRecordCount)
 
-        assertThat(totalConsumerRecords[0].value()).isEqualTo(params.expectedIngestReports[0].toJson())
+        val receivedReport = Klaxon().parse<Report>(totalConsumerRecords[0].value())
+        assertNotNull(receivedReport)
+        assertThat(receivedReport).isEqualTo(params.expectedIngestReports[0])
     }
 
     private fun kafkaTests() = Stream.of(

src/main/kotlin/Ingester.kt

@@ -90,10 +90,8 @@ class Ingester(
                 it.second.let { path ->
                     if (path != null) {
                         sftpClient.open(File(path)).use { stream ->
-                            val binaryUri = "${it.first}/${Service.BINARY_FILE_URI_PATH}"
-                            val mimeType = rdfHandler.getMimeType().first { mT ->
-                                mT.first == digitalInstantiationUrl
-                            }.second
+                            val binaryUri = "$digitalInstantiationUrl/${Service.BINARY_FILE_URI_PATH}"
+                            val mimeType = rdfHandler.getMimeType(digitalInstantiationUrl)
                             log.info("Ingesting binary $binaryUri with mime type $mimeType.")
                             transaction.createOrUpdateBinaryResource(URI(binaryUri), stream, mimeType)
                         }

src/main/kotlin/RdfHandler.kt

@@ -9,6 +9,7 @@ import org.apache.jena.rdf.model.impl.StatementImpl
 import org.apache.jena.riot.Lang
 import org.apache.jena.riot.RDFDataMgr
 import org.apache.log4j.LogManager
+import org.memobase.exceptions.MissingMimeTypeException
 import org.memobase.rdf.EBUCORE
 import org.memobase.rdf.RDF
 import org.memobase.rdf.RICO
@@ -109,11 +110,14 @@ class RdfHandler(data: String, private val externalBaseUrl: String) {
         }.toList()
     }
 
-    fun getMimeType(): List<Pair<String, String?>> {
-        return model.listSubjectsWithProperty(RICO.type, Service.DIGITAL_OBJECT_TYPE).filterKeep {
-            it.hasProperty(EBUCORE.hasMimeType)
-        }.mapWith {
-            Pair(it.uri, it.getProperty(EBUCORE.hasMimeType).`object`.asLiteral().string)
+    fun getMimeType(uri: String): String {
+        val mimeTypes = model.getResource(uri).listProperties(EBUCORE.hasMimeType).mapWith {
+            it.`object`.asLiteral().string
         }.toList()
+        if (mimeTypes.size == 1) {
+            return mimeTypes[0]
+        } else {
+            throw MissingMimeTypeException("No MimeType found for resource $uri")
+        }
     }
 }

src/main/kotlin/Service.kt

@@ -24,6 +24,9 @@ import java.util.Properties
 import org.apache.kafka.clients.consumer.ConsumerRecord
 import org.apache.logging.log4j.LogManager
 import org.apache.logging.log4j.Logger
+import org.fcrepo.client.FcrepoOperationFailedException
+import org.memobase.exceptions.MissingMimeTypeException
+import org.memobase.exceptions.SftpClientException
 import org.memobase.fedora.FedoraClient
 import org.memobase.fedora.FedoraClientImpl
 import org.memobase.settings.SettingsLoader
@@ -80,32 +83,57 @@ class Service(fileName: String = "app.yml") : Closeable {
 
     fun processRecords() {
         for (record in consumer.fetchRecords()) {
-            val ingestReport = processRecord(record)
-            producer.sendReport(record.headers(), ingestReport)
+            processRecord(record)
         }
     }
 
-    private fun processRecord(record: ConsumerRecord<String, String>): Report {
+    private fun processRecord(record: ConsumerRecord<String, String>) {
         val ingester = Ingester(
                 settings.sftpSettings,
                 createFedoraClient(settings.appSettings),
                 settings.appSettings.getProperty("$FEDORA_PROPERTIES_PREFIX.externalBaseUrl")
         )
 
-        return try {
+        try {
             ingester.ingest(record.key(), record.value())
-            Report(
+            val report = Report(
                     id = record.key(),
                     status = ReportStatus.success,
                     message = ReportMessages.ingestedRecord(record.key())
             )
-        } catch (ex: Exception) {
-            log.error("Ingestion of record ${record.key()} failed: ${ex.localizedMessage}", ex)
-            Report(
+            producer.sendReport(record.headers(), report)
+        } catch (ex: FcrepoOperationFailedException) {
+            log.error("${ex.javaClass.canonicalName}:: ${ex.localizedMessage}", ex)
+            val report = Report(
                     id = record.key(),
                     status = ReportStatus.failure,
-                    message = ReportMessages.ingestFailed(record.key())
+                    message = "Fedora Exception: ${ex.localizedMessage}"
             )
+            producer.sendReport(record.headers(), report)
+        } catch (ex: MissingMimeTypeException) {
+            log.error("${ex.javaClass.canonicalName}:: ${ex.localizedMessage}", ex)
+            val report = Report(
+                    id = record.key(),
+                    status = ReportStatus.failure,
+                    message = "Missing MimeType Exception: ${ex.localizedMessage}"
+                )
+            producer.sendReport(record.headers(), report)
+        } catch (ex: SftpClientException) {
+            log.error("${ex.javaClass.canonicalName}: ${ex.localizedMessage}", ex)
+            val report = Report(
+                    id = record.key(),
+                    status = ReportStatus.failure,
+                    message = "SFTP Exception: ${ex.localizedMessage}"
+                )
+            producer.sendReport(record.headers(), report)
+        } catch (ex: Exception) {
+            log.error("${ex.javaClass.canonicalName}: ${ex.localizedMessage}", ex)
+            val report = Report(
+                    id = record.key(),
+                    status = ReportStatus.failure,
+                    message = "Unknown Exception: ${ex.localizedMessage}"
+                )
+            producer.sendReport(record.headers(), report)
         }
     }
 

src/main/kotlin/exceptions/MissingMimeTypeException.kt

+package org.memobase.exceptions
+
+class MissingMimeTypeException(message: String) : Exception(message)

src/test/kotlin/TestRdfHandler.kt

@@ -127,7 +127,8 @@ class TestRdfHandler {
     @Test
     fun `test mime type extraction for binaries`() {
         val rdfHandler = RdfHandler(readFile("input.nt"), "https://memobase.ch/")
-        assertThat(rdfHandler.getMimeType().first().second).isEqualTo("video/mpeg")
+        val mimeType = rdfHandler.getMimeType("https://memobase.ch/instantiation/digital/AFZ-IB_Becker_Audiovisuals_63-1")
+        assertThat(mimeType).isEqualTo("video/mpeg")
     }
 
     private val uris2 = listOf(