add constraint

theonestack · Jul 22, 2024 · 5fa1afc · 5fa1afc
1 parent 4d94332
commit 5fa1afc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/ecs-task.cfhighlander.rb b/ecs-task.cfhighlander.rb
@@ -7,6 +7,7 @@
     Parameters do
       ComponentParam 'EnvironmentName', 'dev', isGlobal: true
       ComponentParam 'EnvironmentType', 'development', allowedValues: ['development','production'], isGlobal: true
+      ComponentParam 'EbsAZ', ''
 
       task_definition.each do |task_def, task|
         if task.has_key?('tag_param')

diff --git a/ecs-task.cfndsl.rb b/ecs-task.cfndsl.rb
@@ -126,7 +126,10 @@
         EC2_Volume(ebs_volume['name']) do
           Size 100
           VolumeType "gp3"
+          AvailabilityZone Ref(:EbsAZ)
         end
+
+        task_constraints << {Type: "memberOf", Expression: "attribute:ecs.availability-zone in #{Ref(:EbsAZ)}"}
         mount_points << { ContainerPath: ebs_volume['container_path'], SourceVolume: Ref(ebs_volume['name']), ReadOnly: false}
         task_volumes << { Name: Ref(ebs_volume['name']), ConfiguredAtLaunch: true }
         task_def.merge!({MountPoints: mount_points })
@@ -312,8 +315,6 @@
       end
     end
 
-    print("Task Volumes: #{task_volumes}")
-
     task_type = external_parameters.fetch(:task_type, 'EC2')
     unless task_definition.empty?