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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
# File 'manifests/server/database.pp', line 18
define postgresql::server::database (
Optional[String[1]] $comment = undef,
String[1] $dbname = $title,
Optional[String[1]] $owner = undef,
Optional[String[1]] $tablespace = undef,
String[1] $template = 'template0',
Optional[String[1]] $encoding = $postgresql::server::encoding,
Optional[String[1]] $locale = $postgresql::server::locale,
Boolean $istemplate = false,
String[1] $instance = 'main',
Hash $connect_settings = $postgresql::server::default_connect_settings,
String[1] $user = $postgresql::server::user,
String[1] $group = $postgresql::server::group,
Stdlib::Absolutepath $psql_path = $postgresql::server::psql_path,
String[1] $default_db = $postgresql::server::default_database,
Stdlib::Port $port = $postgresql::server::port
) {
$version = pick($connect_settings['DBVERSION'], $postgresql::server::_version)
$port_override = pick($connect_settings['PGPORT'], $port)
# Set the defaults for the postgresql_psql resource
Postgresql_psql {
db => $default_db,
psql_user => $user,
psql_group => $group,
psql_path => $psql_path,
port => $port_override,
connect_settings => $connect_settings,
instance => $instance,
}
# Optionally set the locale switch. Older versions of createdb may not accept
# --locale, so if the parameter is undefined its safer not to pass it.
$locale_option = $locale ? {
undef => '',
default => "LC_COLLATE = '${locale}' LC_CTYPE = '${locale}'",
}
$public_revoke_privilege = 'CONNECT'
$template_option = $template ? {
undef => '',
default => "TEMPLATE = \"${template}\"",
}
$encoding_option = $encoding ? {
undef => '',
default => "ENCODING = '${encoding}'",
}
$tablespace_option = $tablespace ? {
undef => '',
default => "TABLESPACE \"${tablespace}\"",
}
postgresql_psql { "CREATE DATABASE \"${dbname}\"":
command => "CREATE DATABASE \"${dbname}\" WITH ${template_option} ${encoding_option} ${locale_option} ${tablespace_option}",
unless => "SELECT 1 FROM pg_database WHERE datname = '${dbname}'",
require => Class['postgresql::server::service'],
}
# This will prevent users from connecting to the database unless they've been
# granted privileges.
~> postgresql_psql { "REVOKE ${public_revoke_privilege} ON DATABASE \"${dbname}\" FROM public":
refreshonly => true,
}
Postgresql_psql["CREATE DATABASE \"${dbname}\""]
-> postgresql_psql { "UPDATE pg_database SET datistemplate = ${istemplate} WHERE datname = '${dbname}'":
unless => "SELECT 1 FROM pg_database WHERE datname = '${dbname}' AND datistemplate = ${istemplate}",
}
if $comment {
Postgresql_psql["CREATE DATABASE \"${dbname}\""]
-> postgresql_psql { "COMMENT ON DATABASE \"${dbname}\" IS '${comment}'":
unless => "SELECT 1 FROM pg_catalog.pg_database d WHERE datname = '${dbname}' AND pg_catalog.shobj_description(d.oid, 'pg_database') = '${comment}'", # lint:ignore:140chars
db => $dbname,
}
}
if $owner {
postgresql_psql { "ALTER DATABASE \"${dbname}\" OWNER TO \"${owner}\"":
unless => "SELECT 1 FROM pg_database JOIN pg_roles rol ON datdba = rol.oid WHERE datname = '${dbname}' AND rolname = '${owner}'",
require => Postgresql_psql["CREATE DATABASE \"${dbname}\""],
}
if defined(Postgresql::Server::Role[$owner]) {
Postgresql::Server::Role[$owner] -> Postgresql_psql["ALTER DATABASE \"${dbname}\" OWNER TO \"${owner}\""]
}
}
if $tablespace {
postgresql_psql { "ALTER DATABASE \"${dbname}\" SET ${tablespace_option}":
unless => "SELECT 1 FROM pg_database JOIN pg_tablespace spc ON dattablespace = spc.oid WHERE datname = '${dbname}' AND spcname = '${tablespace}'", # lint:ignore:140chars
require => Postgresql_psql["CREATE DATABASE \"${dbname}\""],
}
if defined(Postgresql::Server::Tablespace[$tablespace]) {
# The tablespace must be there, before we create the database.
Postgresql::Server::Tablespace[$tablespace] -> Postgresql_psql["CREATE DATABASE \"${dbname}\""]
}
}
}
|