lib/capistrano/consul.rb in capistrano3-consul-0.3.4 vs lib/capistrano/consul.rb in capistrano3-consul-0.3.5

@@ -26,19 +26,19 @@
 
   module DSL
     def consul_all_nodes(properties = {})
       Consul.setup
       Diplomat::Node.get_all.each_with_index do |node, index|
-        properties[:primary] = true if index == 0
-        server(node['Address'], properties)
+        extra_properties = index.zero? ? { primary: true } : {}
+        server(node['Address'], properties.merge(extra_properties))
       end
     end
 
     def consul_service(service_name, properties = {})
       Consul.setup
       Diplomat::Service.get(service_name, :all).each_with_index do |node, index|
-        properties[:primary] = true if index == 0
-        server(node['Address'], properties)
+        extra_properties = index.zero? ? { primary: true } : {}
+        server(node['Address'], properties.merge(extra_properties))
       end
     end
   end
 end