diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/puppet/network/client.rb | 6 | ||||
| -rw-r--r-- | lib/puppet/network/xmlrpc/client.rb | 6 |
2 files changed, 5 insertions, 7 deletions
diff --git a/lib/puppet/network/client.rb b/lib/puppet/network/client.rb index 7950abe3f..fa48ebfb5 100644 --- a/lib/puppet/network/client.rb +++ b/lib/puppet/network/client.rb @@ -85,9 +85,7 @@ class Puppet::Network::Client @driver = self.class.xmlrpc_client.new(args) - if self.read_cert - @driver.cert_setup(self) - end + self.read_cert # We have to start the HTTP connection manually before we start # sending it requests or keep-alive won't work. @@ -120,7 +118,7 @@ class Puppet::Network::Client # Make sure we set the driver up when we read the cert in. def read_cert if super - @driver.cert_setup(self) if @driver.respond_to?(:cert_setup) + @driver.recycle_connection(self) if @driver.respond_to?(:recycle_connection) return true else return false diff --git a/lib/puppet/network/xmlrpc/client.rb b/lib/puppet/network/xmlrpc/client.rb index e2417120b..83d4cf203 100644 --- a/lib/puppet/network/xmlrpc/client.rb +++ b/lib/puppet/network/xmlrpc/client.rb @@ -61,7 +61,7 @@ module Puppet::Network raise error rescue Errno::EPIPE Puppet.warning "Other end went away; restarting connection and retrying" - self.recycle_connection + self.recycle_connection(@cert_client) retry rescue => detail Puppet.err "Could not call %s.%s: %s" % @@ -157,12 +157,12 @@ module Puppet::Network end end - def recycle_connection + def recycle_connection(client) conn_key = "%s:%s" % [@host, @port] @@http_cache.delete(conn_key) initialize_connection - cert_setup(@cert_client) + cert_setup(client) end def start |