Class: Puppet::Util::PostgresqlValidator

Inherits:
Object
  • Object
show all
Defined in:
lib/puppet/util/postgresql_validator.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(resource) ⇒ PostgresqlValidator

Returns a new instance of PostgresqlValidator.



6
7
8
 
# File 'lib/puppet/util/postgresql_validator.rb', line 6

def initialize(resource)
  @resource = resource
end

Instance Attribute Details

#resource ⇒ Object (readonly)

Returns the value of attribute resource.



4
5
6
 
# File 'lib/puppet/util/postgresql_validator.rb', line 4

def resource
  @resource
end

Instance Method Details

#attempt_connection(sleep_length, tries) ⇒ Object 



38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
 
# File 'lib/puppet/util/postgresql_validator.rb', line 38

def attempt_connection(sleep_length, tries)
  (0..tries-1).each do |try|
    Puppet.debug "PostgresqlValidator.attempt_connection: Attempting connection to #{@resource[:db_name]}"
    Puppet.debug "PostgresqlValidator.attempt_connection: #{build_validate_cmd}"
    result = execute_command
    if result && result.length > 0
      Puppet.debug "PostgresqlValidator.attempt_connection: Connection to #{@resource[:db_name] || parse_connect_settings.select { |elem| elem.match /PGDATABASE/ }} successful!"
      return true
    else
      Puppet.warning "PostgresqlValidator.attempt_connection: Sleeping for #{sleep_length} seconds"
      sleep sleep_length
    end
  end
  false
end

#build_psql_cmd ⇒ Object 



10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
 
# File 'lib/puppet/util/postgresql_validator.rb', line 10

def build_psql_cmd
  final_cmd = []

  cmd_init = "#{@resource[:psql_path]} --tuples-only --quiet --no-psqlrc"

  final_cmd.push cmd_init

  cmd_parts = {
    :host => "--host #{@resource[:host]}",
    :port => "--port #{@resource[:port]}",
    :db_username => "--username #{@resource[:db_username]}",
    :db_name => "--dbname #{@resource[:db_name]}",
    :command => "--command '#{@resource[:command]}'"
  }

  cmd_parts.each do |k,v|
    final_cmd.push v if @resource[k]
  end

  final_cmd.join ' '
end

#parse_connect_settings ⇒ Object 



32
33
34
35
36
 
# File 'lib/puppet/util/postgresql_validator.rb', line 32

def parse_connect_settings
  c_settings = @resource[:connect_settings] || {}
  c_settings.merge! ({ 'PGPASSWORD' => @resource[:db_password] }) if @resource[:db_password]
  return c_settings.map { |k,v| "#{k}=#{v}" }
end