Also check sequences aux table

loculus-project · Sep 19, 2024 · efdba92 · efdba92
1 parent faec6a9
commit efdba92
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/backend/src/main/kotlin/org/loculus/backend/controller/SubmissionController.kt b/backend/src/main/kotlin/org/loculus/backend/controller/SubmissionController.kt
@@ -38,6 +38,7 @@ import org.loculus.backend.model.ReleasedDataModel
 import org.loculus.backend.model.SubmissionParams
 import org.loculus.backend.model.SubmitModel
 import org.loculus.backend.service.submission.MetadataUploadAuxTable
+import org.loculus.backend.service.submission.SequenceUploadAuxTable
 import org.loculus.backend.service.submission.SubmissionDatabaseService
 import org.loculus.backend.utils.Accession
 import org.loculus.backend.utils.IteratorStreamer
@@ -376,8 +377,10 @@ class SubmissionController(
             headers.add(HttpHeaders.CONTENT_ENCODING, compression.compressionName)
         }
 
-        val metadataInAuxTable = MetadataUploadAuxTable.select(MetadataUploadAuxTable.accessionColumn).count() > 0
-        if (metadataInAuxTable) {
+        val metadataInAuxTable = MetadataUploadAuxTable.select(MetadataUploadAuxTable.submissionIdColumn).count() > 0
+        val sequencesInAuxTable =
+            SequenceUploadAuxTable.select(SequenceUploadAuxTable.sequenceSubmissionIdColumn).count() > 0
+        if (metadataInAuxTable || sequencesInAuxTable) {
             return ResponseEntity.status(HttpStatus.SERVICE_UNAVAILABLE).build()
         }