Class: Puppet::Provider::ElasticPlugin

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

Instance Method Summary collapse

Instance Method Details

#batch_capable? ⇒ Boolean

Returns:

  • (Boolean) 


166
167
168
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 166

def batch_capable?
  Puppet::Util::Package.versioncmp(es_version, '2.2.0') >= 0
end

#create ⇒ Object 



116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 116

def create
  commands = []
  if is2x?
    commands << "-Des.path.conf=#{homedir}"
    if @resource[:proxy]
      commands += proxy_args(@resource[:proxy])
    end
  end
  commands << 'install'
  commands << '--batch' if batch_capable?
  commands += is1x? ? install1x : install2x
  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

  writepluginfile
end

#destroy ⇒ Object 



146
147
148
149
150
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 146

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

#es_version ⇒ Object 



152
153
154
155
156
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 152

def es_version
  Puppet_X::Elastic::EsVersioning.version(
    resource[:elasticsearch_package_name], resource.catalog
  )
end

#exists? ⇒ Boolean

Returns:

  • (Boolean) 


16
17
18
19
20
21
22
23
24
25
26
27
28
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 16

def exists?
  if !File.exists?(pluginfile)
    debug "Plugin file #{pluginfile} does not exist"
    return false
  elsif File.exists?(pluginfile) && readpluginfile != pluginfile_content
    debug "Got #{readpluginfile} Expected #{pluginfile_content}. Removing for reinstall"
    self.destroy
    return false
  else
    debug "Plugin exists"
    return true
  end
end

#homedir ⇒ Object 



7
8
9
10
11
12
13
14
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 7

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

#install1x ⇒ Object 



68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 68

def install1x
  if !@resource[:url].nil?
    [
      Puppet_X::Elastic::plugin_name(@resource[:name]),
      '--url',
      @resource[:url]
    ]
  elsif !@resource[:source].nil?
    [
      Puppet_X::Elastic::plugin_name(@resource[:name]),
      '--url',
      "file://#{@resource[:source]}"
    ]
  else
    [
      @resource[:name]
    ]
  end
end

#install2x ⇒ Object 



88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 88

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

#is1x? ⇒ Boolean

Returns:

  • (Boolean) 


158
159
160
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 158

def is1x?
  Puppet::Util::Package.versioncmp(es_version, '2.0.0') < 0
end

#is2x? ⇒ Boolean

Returns:

  • (Boolean) 


162
163
164
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 162

def is2x?
  (Puppet::Util::Package.versioncmp(es_version, '2.0.0') >= 0) && (Puppet::Util::Package.versioncmp(es_version, '3.0.0') < 0)
end

#plugin_version(plugin_name) ⇒ Object 



170
171
172
173
174
175
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 170

def plugin_version(plugin_name)
  _vendor, _plugin, version = plugin_name.split('/')
  return es_version if is2x? && version.nil?
  return version.scan(/\d+\.\d+\.\d+(?:\-\S+)?/).first unless version.nil?
  return false
end

#pluginfile ⇒ Object 



41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 41

def pluginfile
  if @resource[:plugin_path]
    File.join(
      @resource[:plugin_dir],
      @resource[:plugin_path],
      '.name'
    )
  else
    File.join(
      @resource[:plugin_dir],
      Puppet_X::Elastic::plugin_name(@resource[:name]),
      '.name'
    )
  end
end

#pluginfile_content ⇒ Object 



30
31
32
33
34
35
36
37
38
39
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 30

def pluginfile_content
  return @resource[:name] if is1x?

  if @resource[:name].split("/").count == 1 # Official plugin
    version = plugin_version(@resource[:name])
    return "#{@resource[:name]}/#{version}"
  else
    return @resource[:name]
  end
end

#proxy_args(url) ⇒ Object 



104
105
106
107
108
109
110
111
112
113
114
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 104

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

#readpluginfile ⇒ Object 



63
64
65
66
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 63

def readpluginfile
  f = File.open(pluginfile)
  f.readline
end

#with_environment(&block) ⇒ Object

Run a command wrapped in necessary env vars



178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 178

def with_environment(&block)
  env_vars = {
    'ES_JAVA_OPTS' => [],
  }
  saved_vars = {}

  if not is2x?
    env_vars['ES_JAVA_OPTS'] << "-Des.path.conf=#{homedir}"
    if @resource[:proxy]
      env_vars['ES_JAVA_OPTS'] += proxy_args(@resource[:proxy])
    end
  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.call

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

  return ret
end

#writepluginfile ⇒ Object 



57
58
59
60
61
 
# File 'lib/puppet/provider/elastic_plugin.rb', line 57

def writepluginfile
  File.open(pluginfile, 'w') do |file|
    file.write pluginfile_content
  end
end