Class: Puppet::Provider::ElasticPlugin

Inherits:
Puppet::Provider
  • Object
show all
Defined in:
lib/puppet/provider/elastic_plugin.rb

Overview

Generalized parent class for providers that behave like Elasticsearch’s plugin command line tool. rubocop:disable Metrics/ClassLength

Instance Method Summary collapse

Instance Method Details

#create ⇒ Object

Install this plugin on the host.



101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 101

def create
  commands = []
  commands += proxy_args(@resource[:proxy]) if @resource[:proxy]
  commands << 'install'
  commands << '--batch'
  commands += install_args
  debug("Commands: #{commands.inspect}")

  retry_count = 3
  retry_times = 0
  begin
    with_environment do
      plugin(commands)
    end
  rescue Puppet::ExecutionFailure => e
    retry_times += 1
    debug("Failed to install plugin. Retrying... #{retry_times} of #{retry_count}")
    sleep 2
    retry if retry_times < retry_count
    raise "Failed to install plugin. Received error: #{e.inspect}"
  end
end

#destroy ⇒ Object

Remove this plugin from the host.



125
126
127
128
129
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 125

def destroy
  with_environment do
    plugin(['remove', Puppet_X::Elastic.plugin_name(@resource[:name])])
  end
end

#exists? ⇒ Boolean

Returns:

  • (Boolean) 


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
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 23

def exists?
  # First, attempt to list whether the named plugin exists by finding a
  # plugin descriptor file, which each plugin should have. We must wildcard
  # the name to match meta plugins, see upstream issue for this change:
  # https://github.com/elastic/elasticsearch/pull/28022
  properties_files = Dir[File.join(@resource[:plugin_dir], plugin_path, '**', '*plugin-descriptor.properties')]
  return false if properties_files.empty?

  begin
    # Use the basic name format that the plugin tool supports in order to
    # determine the version from the resource name.
    plugin_version = Puppet_X::Elastic.plugin_version(@resource[:name])

    # Naively parse the Java .properties file to check version equality.
    # Because we don't have the luxury of installing arbitrary gems, perform
    # simple parse with a degree of safety checking in the call chain
    #
    # Note that x-pack installs "meta" plugins which bundle multiple plugins
    # in one. Therefore, we need to find the first "sub" plugin that
    # indicates which version of x-pack this is.
    properties = properties_files.sort.map do |prop_file|
      IO.readlines(prop_file).map(&:strip).reject do |line|
        line.start_with?('#') or line.empty?
      end.map do |property|
        property.split('=')
      end.reject do |pairs|
        pairs.length != 2
      end.to_h
    end.find { |prop| prop.key? 'version' }

    if properties and properties['version'] != plugin_version
      debug "Elasticsearch plugin #{@resource[:name]} not version #{plugin_version}, reinstalling"
      destroy
      return false
    end
  rescue ElasticPluginParseFailure
    debug "Failed to parse plugin version for #{@resource[:name]}"
  end

  # If there is no version string, we do not check version equality
  debug "No version found in #{@resource[:name]}, not enforcing any version"
  true
end

#homedir ⇒ Object

Elasticsearch’s home directory.

Returns:

  • String



14
15
16
17
18
19
20
21
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 14

def homedir
  case Facter.value('osfamily')
  when 'OpenBSD'
    '/usr/local/elasticsearch'
  else
    '/usr/share/elasticsearch'
  end
end

#install_args ⇒ Array<String>

Intelligently returns the correct installation arguments for Elasticsearch.

Returns:

  • (Array<String>) —

    arguments to pass to the plugin installation utility



75
76
77
78
79
80
81
82
83
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 75

def install_args
  if !@resource[:url].nil?
    [@resource[:url]]
  elsif !@resource[:source].nil?
    ["file://#{@resource[:source]}"]
  else
    [@resource[:name]]
  end
end

#plugin_path ⇒ Object 



67
68
69
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 67

def plugin_path
  @resource[:plugin_path] || Puppet_X::Elastic.plugin_name(@resource[:name])
end

#proxy_args(url) ⇒ Object

Format proxy arguments for consumption by the elasticsearch plugin management tool (i.e., Java properties).

Returns:

  • Array of flags for command-line tools



90
91
92
93
94
95
96
97
98
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 90

def proxy_args(url)
  parsed = URI(url)
  %w[http https].map do |schema|
    [:host, :port, :user, :password].map do |param|
      option = parsed.send(param)
      "-D#{schema}.proxy#{param.to_s.capitalize}=#{option}" unless option.nil?
    end
  end.flatten.compact
end

#with_environment(&block) ⇒ Object

Run a command wrapped in necessary env vars



132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 132

def with_environment(&block)
  env_vars = {
    'ES_JAVA_OPTS' => @resource[:java_opts],
    'ES_PATH_CONF' => @resource[:configdir]
  }
  saved_vars = {}

  # Use 'java_home' param if supplied, otherwise default to Elasticsearch shipped JDK
  env_vars['JAVA_HOME'] = if @resource[:java_home].nil? or @resource[:java_home] == ''
                            "#{homedir}/jdk"
                          else
                            @resource[:java_home]
                          end

  env_vars['ES_JAVA_OPTS'] = env_vars['ES_JAVA_OPTS'].join(' ')

  env_vars.each do |env_var, value|
    saved_vars[env_var] = ENV[env_var]
    ENV[env_var] = value
  end

  ret = block.yield

  saved_vars.each do |env_var, value|
    ENV[env_var] = value
  end

  ret
end