1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
# frozen_string_literal: true
require 'socket'
require 'timeout'
module Puppet
# Namespace for miscellaneous tools
module Util
# Helper class to assist with talking to the Elasticsearch service ports.
class EsInstanceValidator
attr_reader :instance_server, :instance_port
def initialize(instance_server, instance_port)
@instance_server = instance_server
@instance_port = instance_port
# Avoid deprecation warnings in Puppet versions < 4
@timeout = if Facter.value(:puppetversion).split('.').first.to_i < 4
Puppet[:configtimeout]
else
Puppet[:http_connect_timeout]
end
end
# Utility method; attempts to make an https connection to the Elasticsearch instance.
# This is abstracted out into a method so that it can be called multiple times
# for retry attempts.
#
# @return true if the connection is successful, false otherwise.
def attempt_connection
Timeout.timeout(@timeout) do
TCPSocket.new(@instance_server, @instance_port).close
true
rescue Errno::EADDRNOTAVAIL, Errno::ECONNREFUSED, Errno::EHOSTUNREACH => e
Puppet.debug "Unable to connect to Elasticsearch instance (#{@instance_server}:#{@instance_port}): #{e.message}"
false
end
rescue Timeout::Error
false
end
end
end
end
|
