Skip to content

Commit 725d24f

Browse files
committed
limit file size on ingest to 500MB
1 parent 0b56a49 commit 725d24f

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

image-loader/app/controllers/ImageLoaderController.scala

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,18 @@ class ImageLoaderController(auth: Authentication,
141141

142142
val approximateReceiveCount = getApproximateReceiveCount(sqsMessage)
143143

144-
if (approximateReceiveCount > 2) {
144+
if(s3IngestObject.contentLength > 500000000){ // 500MB
145+
val errorMessage = s"File size exceeds the maximum allowed size (500MB). Moving to fail bucket."
146+
logger.warn(logMarker, errorMessage)
147+
store.moveObjectToFailedBucket(s3IngestObject.key)
148+
s3IngestObject.maybeMediaIdFromUiUpload foreach { imageId =>
149+
uploadStatusTable.updateStatus( // fire & forget, since there's nothing else we can do
150+
imageId, UploadStatus(StatusType.Failed, Some(errorMessage))
151+
)
152+
}
153+
Future.unit
154+
}
155+
else if (approximateReceiveCount > 2) {
145156
metrics.abandonedMessagesFromQueue.incrementBothWithAndWithoutDimensions(metricDimensions)
146157
val errorMessage = s"File processing has been attempted $approximateReceiveCount times. Moving to fail bucket."
147158
logger.warn(logMarker, errorMessage)

0 commit comments

Comments
 (0)