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
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
|
require 'json'
require 'puppet'
require File.expand_path(File.join(File.dirname(__FILE__), '..', 'rabbitmq_cli'))
Puppet::Type.type(:rabbitmq_queue).provide(:rabbitmqadmin, parent: Puppet::Provider::RabbitmqCli) do
confine feature: :posix
def should_vhost
if @should_vhost
@should_vhost
else
@should_vhost = resource[:name].rpartition('@').last
end
end
def self.all_vhosts
rabbitmqctl_list('vhosts').split(%r{\n})
end
def self.all_queues(vhost)
rabbitmqctl_list('queues', '-p', vhost, 'name', 'durable', 'auto_delete', 'arguments').split(%r{\n})
end
def self.instances
resources = []
all_vhosts.each do |vhost|
all_queues(vhost).map do |line|
next if line =~ %r{^federation:}
name, durable, auto_delete, arguments = line.split("\t")
# Convert output of arguments from the rabbitmqctl command to a json string.
if !arguments.nil?
arguments = arguments.gsub(%r{^\[(.*)\]$}, '').gsub(%r{\{("(?:.|\\")*?"),}, '{\1:').gsub(%r{\},\{}, ',')
arguments = '{}' if arguments == ''
else
arguments = '{}'
end
queue = {
durable: durable,
auto_delete: auto_delete,
arguments: JSON.parse(arguments),
ensure: :present,
name: format('%s@%s', name, vhost)
}
resources << new(queue) if queue[:name]
end
end
resources
end
def self.prefetch(resources)
packages = instances
resources.keys.each do |name|
if (provider = packages.find { |pkg| pkg.name == name })
resources[name].provider = provider
end
end
end
def exists?
@property_hash[:ensure] == :present
end
def create
vhost_opt = should_vhost ? "--vhost=#{should_vhost}" : ''
name = resource[:name].rpartition('@').first
arguments = resource[:arguments]
arguments = {} if arguments.nil?
rabbitmqadmin('declare',
'queue',
vhost_opt,
"--user=#{resource[:user]}",
"--password=#{resource[:password]}",
'-c',
'/etc/rabbitmq/rabbitmqadmin.conf',
"name=#{name}",
"durable=#{resource[:durable]}",
"auto_delete=#{resource[:auto_delete]}",
"arguments=#{arguments.to_json}")
@property_hash[:ensure] = :present
end
def destroy
vhost_opt = should_vhost ? "--vhost=#{should_vhost}" : ''
name = resource[:name].rpartition('@').first
rabbitmqadmin('delete', 'queue', vhost_opt, "--user=#{resource[:user]}", "--password=#{resource[:password]}", '-c', '/etc/rabbitmq/rabbitmqadmin.conf', "name=#{name}")
@property_hash[:ensure] = :absent
end
end
|
