feat: allow to configure resource limits and reservation with stack (#…

…698) (#699)
swiftwave-org · Jun 2, 2024 · fbb560d · fbb560d
1 parent 47bce42
commit fbb560d
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 11 deletions.
diff --git a/swiftwave_service/graphql/graphql_object_mapper.go b/swiftwave_service/graphql/graphql_object_mapper.go
@@ -551,15 +551,21 @@ func stackToApplicationsInput(stackName string, record *stack_parser.Stack, db g
 		image := service.Image
 		replicas := service.Deploy.Replicas
 		app := model.ApplicationInput{
-			Name:                         serviceName,
-			EnvironmentVariables:         environmentVariables,
-			PersistentVolumeBindings:     persistentVolumeBindings,
-			Capabilities:                 service.CapAdd,
-			Sysctls:                      sysctls,
-			Dockerfile:                   nil,
-			BuildArgs:                    []*model.BuildArgInput{},
-			DeploymentMode:               model.DeploymentMode(service.Deploy.Mode),
-			Replicas:                     &replicas,
+			Name:                     serviceName,
+			EnvironmentVariables:     environmentVariables,
+			PersistentVolumeBindings: persistentVolumeBindings,
+			Capabilities:             service.CapAdd,
+			Sysctls:                  sysctls,
+			Dockerfile:               nil,
+			BuildArgs:                []*model.BuildArgInput{},
+			DeploymentMode:           model.DeploymentMode(service.Deploy.Mode),
+			Replicas:                 &replicas,
+			ResourceLimit: &model.ResourceLimitInput{
+				MemoryMb: service.Deploy.Resources.Limits.MemoryMB,
+			},
+			ReservedResource: &model.ReservedResourceInput{
+				MemoryMb: service.Deploy.Resources.Reservations.MemoryMB,
+			},
 			UpstreamType:                 model.UpstreamTypeImage,
 			DockerImage:                  &image,
 			ImageRegistryCredentialID:    nil,

diff --git a/swiftwave_service/stack_parser/types.go b/swiftwave_service/stack_parser/types.go
@@ -160,8 +160,23 @@ const (
 )
 
 type Deploy struct {
-	Mode     DeploymentMode `yaml:"mode"`
-	Replicas uint           `yaml:"replicas"`
+	Mode      DeploymentMode `yaml:"mode"`
+	Replicas  uint           `yaml:"replicas"`
+	Resources Resources      `yaml:"resources"`
+}
+
+// Resources : Resources for the service
+type Resources struct {
+	Limits       ResourcesLimits       `yaml:"limits"`
+	Reservations ResourcesReservations `yaml:"reservations"`
+}
+
+type ResourcesLimits struct {
+	MemoryMB int `yaml:"memory"`
+}
+
+type ResourcesReservations struct {
+	MemoryMB int `yaml:"memory"`
 }
 
 // Docs : Documentation for the stack