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



8
9
10
11
12
13
 
# File 'lib/facter/es_facts.rb', line 8

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



24
25
26
27
28
29
30
31
 
# File 'lib/facter/es_facts.rb', line 24

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. rubocop:disable Lint/HandleExceptions rubocop:disable Metrics/CyclomaticComplexity rubocop:disable Metrics/PerceivedComplexity



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
 
# 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")
    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 json_data['status'] && json_data['status'] == 200
        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? or 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
  end
  Facter.add(:elasticsearch) do
    setcode do
      nodes
    end
    nodes unless nodes.empty?
  end
end

.ssl?(config) ⇒ Boolean

Returns:

  • (Boolean) 


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

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

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