lib/elasticity/job_flow.rb in elasticity-6.0.7 vs lib/elasticity/job_flow.rb in elasticity-6.0.8

@@ -27,10 +27,12 @@
     attr_accessor :job_flow_role
     attr_accessor :service_role
     attr_accessor :jobflow_id
     attr_accessor :aws_applications
     attr_accessor :additional_info
+    attr_accessor :additional_master_security_groups
+    attr_accessor :additional_slave_security_groups
 
     def initialize
       @action_on_failure = 'TERMINATE_JOB_FLOW'
       @name = 'Elasticity Job Flow'
       @keep_job_flow_alive_when_no_steps = false
@@ -244,9 +246,11 @@
       if @ec2_subnet_id
         preamble[:instances].merge!(:ec2_subnet_id => @ec2_subnet_id)
         preamble[:instances].delete(:placement)
       end
       preamble[:instances].merge!(:ec2_key_name => @ec2_key_name) if @ec2_key_name
+      preamble[:instances].merge!(:additional_master_security_groups => @additional_master_security_groups) if @additional_master_security_groups
+      preamble[:instances].merge!(:additional_slave_security_groups => @additional_slave_security_groups) if @additional_slave_security_groups
       preamble
     end
 
     def jobflow_steps
       steps = []