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
|
# frozen_string_literal: true
Puppet::Type.type(:postgresql_replication_slot).provide(:ruby) do
desc 'For confinement'
commands psql: 'psql'
def self.instances
run_sql_command('SELECT * FROM pg_replication_slots;')[0].split("\n").select { |l| l.include?('|') }.map do |l|
name, *_others = l.strip.split(%r{\s+\|\s+})
new(name: name,
ensure: :present)
end
end
def self.prefetch(resources)
instances.each do |i|
slot = resources[i.name]
slot.provider = i if slot
end
end
def exists?
@property_hash[:ensure] == :present
end
def create
output = self.class.run_sql_command("SELECT * FROM pg_create_physical_replication_slot('#{resource[:name]}');")
raise Puppet::Error, "Failed to create replication slot #{resource[:name]}:\n#{output[0]}" unless output[1].success?
@property_hash[:ensure] = :present
end
def destroy
output = self.class.run_sql_command("SELECT pg_drop_replication_slot('#{resource[:name]}');")
raise Puppet::Error, "Failed to destroy replication slot #{resource[:name]}:\n#{output[0]}" unless output[1].success?
@property_hash[:ensure] = :absent
end
private
def self.run_sql_command(sql)
command = ['psql', '-t', '-c', sql]
run_command(command, 'postgres', 'postgres')
end
def self.run_command(command, user, group)
output = Puppet::Util::Execution.execute(command, uid: user,
gid: group,
failonfail: false,
combine: true,
override_locale: true,
custom_environment: {})
[output, $CHILD_STATUS.dup]
end
end
|
