Module: EsFacts

Defined in:
lib/facter/es_facts.rb

Overview

Helper module to encapsulate custom fact injection

Class Method Summary collapse

Class Method Details

.add_fact(prefix, key, value) ⇒ Object

Add a fact to the catalog of host facts



10
11
12
13
14
15
 
# File 'lib/facter/es_facts.rb', line 10

def self.add_fact(prefix, key, value)
  key = "#{prefix}_#{key}".to_sym
  ::Facter.add(key) do
    setcode { value }
  end
end

.get_httpport(config) ⇒ Object

Helper to determine the instance http.port number



26
27
28
29
30
31
32
33
34
 
# File 'lib/facter/es_facts.rb', line 26

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

.run ⇒ Object

Entrypoint for custom fact populator

This is a super old function but works; disable a bunch of checks.



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
 
# File 'lib/facter/es_facts.rb', line 39

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

.ssl?(config) ⇒ Boolean

Returns:

  • (Boolean) 


17
18
19
20
21
22
23
 
# File 'lib/facter/es_facts.rb', line 17

def self.ssl?(config)
  tls_keys = [
    'xpack.security.http.ssl.enabled'
  ]

  tls_keys.any? { |key| (config.key? key) && (config[key] == true) }
end