@@ -14,7 +14,6 @@ import {
1414 PreStopCauses ,
1515 EnvironmentType ,
1616} from "@trigger.dev/core/v3" ;
17- import { randomUUID } from "crypto" ;
1817import { TaskMonitor } from "./taskMonitor" ;
1918import { PodCleaner } from "./podCleaner" ;
2019import { UptimeHeartbeat } from "./uptimeHeartbeat" ;
@@ -41,7 +40,7 @@ type Namespace = {
4140 } ;
4241} ;
4342
44- type ComputeResources = {
43+ type ResourceQuantities = {
4544 [ K in "cpu" | "memory" | "ephemeral-storage" ] ?: string ;
4645} ;
4746
@@ -166,15 +165,7 @@ class KubernetesTaskOperations implements TaskOperations {
166165 containerPort : 8000 ,
167166 } ,
168167 ] ,
169- resources : {
170- requests : {
171- ...this . #defaultResourceRequests,
172- } ,
173- limits : {
174- ...this . #defaultResourceLimits,
175- ...this . #getResourcesFromMachineConfig( opts . machine ) ,
176- } ,
177- } ,
168+ resources : this . #getResourcesForMachine( opts . machine ) ,
178169 lifecycle : {
179170 preStop : {
180171 exec : {
@@ -265,15 +256,7 @@ class KubernetesTaskOperations implements TaskOperations {
265256 containerPort : 8000 ,
266257 } ,
267258 ] ,
268- resources : {
269- requests : {
270- ...this . #defaultResourceRequests,
271- } ,
272- limits : {
273- ...this . #defaultResourceLimits,
274- ...this . #getResourcesFromMachineConfig( opts . machine ) ,
275- } ,
276- } ,
259+ resources : this . #getResourcesForMachine( opts . machine ) ,
277260 lifecycle : {
278261 postStart : {
279262 exec : {
@@ -414,13 +397,13 @@ class KubernetesTaskOperations implements TaskOperations {
414397 } ;
415398 }
416399
417- get #defaultResourceRequests( ) : ComputeResources {
400+ get #defaultResourceRequests( ) : ResourceQuantities {
418401 return {
419402 "ephemeral-storage" : "2Gi" ,
420403 } ;
421404 }
422405
423- get #defaultResourceLimits( ) : ComputeResources {
406+ get #defaultResourceLimits( ) : ResourceQuantities {
424407 return {
425408 "ephemeral-storage" : "10Gi" ,
426409 } ;
@@ -486,13 +469,33 @@ class KubernetesTaskOperations implements TaskOperations {
486469 } ;
487470 }
488471
489- #getResourcesFromMachineConfig( preset : MachinePreset ) : ComputeResources {
472+ #getResourceRequestsForMachine( preset : MachinePreset ) : ResourceQuantities {
473+ return {
474+ cpu : `${ preset . cpu * 0.75 } ` ,
475+ memory : `${ preset . memory } G` ,
476+ } ;
477+ }
478+
479+ #getResourceLimitsForMachine( preset : MachinePreset ) : ResourceQuantities {
490480 return {
491481 cpu : `${ preset . cpu } ` ,
492482 memory : `${ preset . memory } G` ,
493483 } ;
494484 }
495485
486+ #getResourcesForMachine( preset : MachinePreset ) : k8s . V1ResourceRequirements {
487+ return {
488+ requests : {
489+ ...this . #defaultResourceRequests,
490+ ...this . #getResourceRequestsForMachine( preset ) ,
491+ } ,
492+ limits : {
493+ ...this . #defaultResourceLimits,
494+ ...this . #getResourceLimitsForMachine( preset ) ,
495+ } ,
496+ } ;
497+ }
498+
496499 #getLifecycleCommand< THookType extends "postStart" | "preStop" > (
497500 type : THookType ,
498501 cause : THookType extends "postStart" ? PostStartCauses : PreStopCauses
0 commit comments