ruby.rb
View Options

	Puppet::Type.type(:postgresql_psql).provide(:ruby) do

	def run_unless_sql_command(sql)
	# for the 'unless' queries, we wrap the user's query in a 'SELECT COUNT',
	# which makes it easier to parse and process the output.
	run_sql_command('SELECT COUNT(*) FROM (' << sql << ') count')
	end

	def run_sql_command(sql)
	if resource[:search_path]
	sql = "set search_path to #{Array(resource[:search_path]).join(',')}; #{sql}"
	end

	command = [resource[:psql_path]]
	command.push("-d", resource[:db]) if resource[:db]
	command.push("-p", resource[:port]) if resource[:port]
	command.push("-t", "-c", '"' + sql.gsub('"', '\"') + '"')

	if resource[:cwd]
	Dir.chdir resource[:cwd] do
	run_command(command, resource[:psql_user], resource[:psql_group])
	end
	else
	run_command(command, resource[:psql_user], resource[:psql_group])
	end
	end

	private

	def get_environment
	environment = {}
	if envlist = resource[:environment]
	envlist = [envlist] unless envlist.is_a? Array
	envlist.each do \|setting\|
	if setting =~ /^(\w+)=((.\|\n)+)$/
	env_name = $1
	value = $2
	if environment.include?(env_name) \|\| environment.include?(env_name.to_sym)
	warning "Overriding environment setting '#{env_name}' with '#{value}'"
	end
	environment[env_name] = value
	else
	warning "Cannot understand environment setting #{setting.inspect}"
	end
	end
	end
	return environment
	end

	def run_command(command, user, group)
	command = command.join ' '
	environment = get_environment
	if Puppet::PUPPETVERSION.to_f < 3.4
	require 'puppet/util/execution'
	Puppet::Util::Execution.withenv environment do
	Puppet::Util::SUIDManager.run_and_capture(command, user, group)
	end
	else
	output = Puppet::Util::Execution.execute(command, {
	:uid => user,
	:gid => group,
	:failonfail => false,
	:combine => true,
	:override_locale => true,
	:custom_environment => environment
	})
	[output, $CHILD_STATUS.dup]
	end
	end

	end

File Metadata

Mime Type: text/x-ruby
Expires: Jul 4 2025, 10:41 AM (4 w, 3 d ago)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 3452526

ruby.rb
No OneTemporary
Actions

ruby.rb
View Options

File Metadata

Event Timeline

ruby.rbNo OneTemporaryActions

ruby.rbView Options

File Metadata

Event Timeline

ruby.rb
No OneTemporary
Actions

ruby.rb
View Options