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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
|
# frozen_string_literal: true
require 'net/http'
require 'json'
require 'yaml'
# Helper module to encapsulate custom fact injection
module EsFacts
# Add a fact to the catalog of host facts
def self.add_fact(prefix, key, value)
key = "#{prefix}_#{key}".to_sym
::Facter.add(key) do
setcode { value }
end
end
def self.ssl?(config)
tls_keys = [
'xpack.security.http.ssl.enabled'
]
tls_keys.any? { |key| (config.key? key) && (config[key] == true) }
end
# Helper to determine the instance http.port number
def self.get_httpport(config)
enabled = 'http.enabled'
httpport = 'http.port'
return false, false if !config[enabled].nil? && config[enabled] == 'false'
return config[httpport], ssl?(config) unless config[httpport].nil?
['9200', ssl?(config)]
end
# Entrypoint for custom fact populator
#
# This is a super old function but works; disable a bunch of checks.
def self.run
dir_prefix = '/etc/elasticsearch'
# httpports is a hash of port_number => ssl?
transportports = []
http_bound_addresses = []
transport_bound_addresses = []
transport_publish_addresses = []
nodes = {}
# only when the directory exists we need to process the stuff
return unless File.directory?(dir_prefix)
if File.readable?("#{dir_prefix}/elasticsearch.yml")
config_data = YAML.load_file("#{dir_prefix}/elasticsearch.yml")
return unless config_data
httpport, ssl = get_httpport(config_data)
end
begin
add_fact('elasticsearch', 'port', httpport)
unless ssl
key_prefix = 'elasticsearch'
# key_prefix = "elasticsearch_#{httpport}"
uri = URI("http://localhost:#{httpport}")
http = Net::HTTP.new(uri.host, uri.port)
http.read_timeout = 10
http.open_timeout = 2
response = http.get('/')
json_data = JSON.parse(response.body)
if response.code == '200' && json_data['version'] && json_data['name']
add_fact(key_prefix, 'name', json_data['name'])
add_fact(key_prefix, 'version', json_data['version']['number'])
uri2 = URI("http://localhost:#{httpport}/_nodes/#{json_data['name']}")
http2 = Net::HTTP.new(uri2.host, uri2.port)
http2.read_timeout = 10
http2.open_timeout = 2
response2 = http2.get(uri2.path)
json_data_node = JSON.parse(response2.body)
add_fact(key_prefix, 'cluster_name', json_data_node['cluster_name'])
node_data = json_data_node['nodes'].first
add_fact(key_prefix, 'node_id', node_data[0])
nodes_data = json_data_node['nodes'][node_data[0]]
process = nodes_data['process']
add_fact(key_prefix, 'mlockall', process['mlockall'])
plugins = nodes_data['plugins']
plugin_names = []
plugins.each do |plugin|
plugin_names << plugin['name']
plugin.each do |key, value|
prefix = "#{key_prefix}_plugin_#{plugin['name']}"
add_fact(prefix, key, value) unless key == 'name'
end
end
add_fact(key_prefix, 'plugins', plugin_names.join(','))
nodes_data['http']['bound_address'].each { |i| http_bound_addresses << i }
nodes_data['transport']['bound_address'].each { |i| transport_bound_addresses << i }
transport_publish_addresses << nodes_data['transport']['publish_address'] unless nodes_data['transport']['publish_address'].nil?
transportports << nodes_data['settings']['transport']['tcp']['port'] unless nodes_data['settings']['transport']['tcp'].nil? || nodes_data['settings']['transport']['tcp']['port'].nil?
node = {
'http_ports' => httpports.keys,
'transport_ports' => transportports,
'http_bound_addresses' => http_bound_addresses,
'transport_bound_addresses' => transport_bound_addresses,
'transport_publish_addresses' => transport_publish_addresses,
json_data['name'] => {
'settings' => nodes_data['settings'],
'http' => nodes_data['http'],
'transport' => nodes_data['transport']
}
}
nodes.merge! node
end
end
rescue StandardError
# ignore
end
Facter.add(:elasticsearch) do
setcode do
nodes unless nodes.empty?
end
end
end
end
EsFacts.run
|
