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
|
# frozen_string_literal: true
$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))
require 'puppet/provider/elastic_rest'
Puppet::Type.type(:elasticsearch_snapshot_repository).provide(
:ruby,
parent: Puppet::Provider::ElasticREST,
api_uri: '_snapshot'
) do
desc 'A REST API based provider to manage Elasticsearch snapshot repositories.'
mk_resource_methods
def self.process_body(body)
Puppet.debug('Got to snapshot_repository.process_body')
JSON.parse(body).map do |object_name, api_object|
{
name: object_name,
ensure: :present,
type: api_object['type'],
compress: api_object['settings']['compress'],
location: api_object['settings']['location'],
chunk_size: api_object['settings']['chunk_size'],
max_restore_rate: api_object['settings']['max_restore_rate'],
max_snapshot_rate: api_object['settings']['max_snapshot_rate'],
provider: name
}.compact
end
end
def generate_body
Puppet.debug('Got to snapshot_repository.generate_body')
# Build core request body
body = {
'type' => resource[:type],
'settings' => {
'compress' => resource[:compress],
'location' => resource[:location]
}
}
# Add optional values
body['settings']['chunk_size'] = resource[:chunk_size] unless resource[:chunk_size].nil?
body['settings']['max_restore_rate'] = resource[:max_restore_rate] unless resource[:max_restore_rate].nil?
body['settings']['max_snapshot_rate'] = resource[:max_snapshot_rate] unless resource[:max_snapshot_rate].nil?
# Convert to JSON and return
JSON.generate(body)
end
end
|
