allow imagehandlers to provide per file overrides

UMN-LATIS · Sep 14, 2024 · 421408e · 421408e
1 parent 2a9414e
commit 421408e
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/application/models/filehandlers/Filehandlerbase.php b/application/models/filehandlers/Filehandlerbase.php
@@ -414,6 +414,10 @@ public function save() {
 
 	}
 
+	public function getCustomJobOverrides() {
+		return [];
+	}
+
 	public function queueBatchItem($asset, $sourceFile) {
 		$fileObjectId = $asset->getFileObjectId();
 		$fileSize = $sourceFile->getFileSize();
@@ -425,6 +429,10 @@ public function queueBatchItem($asset, $sourceFile) {
 		else {
 			$size = "large";
 		}
+
+		$overrides = $this->getCustomJobOverrides();
+
+
 		$jobDefinition = $this->config->item('awsQueueJobDefinition') . "-" . $size;
 
 		$batchClient = new BatchClient([
@@ -444,6 +452,7 @@ public function queueBatchItem($asset, $sourceFile) {
 			],
 			'containerOverrides' => [
         		'command' => ['bash', 'runJob.sh',  $fileObjectId],
+				'resourceRequirements' => $overrides
     		],
 		];
 		try {

diff --git a/application/models/filehandlers/Imagehandler.php b/application/models/filehandlers/Imagehandler.php
@@ -484,6 +484,15 @@ public function priority() {
 		return 0;
 	}
 
+	public function getCustomJobOverrides() {
+		//if the file type is czi bump the memory
+		if($this->sourceFile->getType() == "czi") {
+			return [
+				["type"=>"MEMORY",
+						"value"=>"4096"]
+					];
+		}
+	}
 
 
 }