module ForemanMaintain::Concerns::BaseDatabase
Public Instance Methods
backup_db_command(file_path, config = configuration)
click to toggle source
TODO: refactor to use #dump_db
# File lib/foreman_maintain/concerns/base_database.rb, line 93 def backup_db_command(file_path, config = configuration) pg_dump_cmd = "pg_dump -Fc #{config['database']}" "runuser - postgres -c '#{pg_dump_cmd}' | bzip2 -9 > #{file_path}" end
backup_dir()
click to toggle source
TODO: remove the backup file path tools from here. Lib Utils::Backup?
# File lib/foreman_maintain/concerns/base_database.rb, line 99 def backup_dir @backup_dir ||= File.expand_path(ForemanMaintain.config.db_backup_dir) end
backup_file_path(config = configuration)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 50 def backup_file_path(config = configuration) dump_file_name = "#{config['database']}_#{Time.now.strftime('%Y-%m-%d_%H-%M-%S')}.dump" "#{backup_dir}/#{dump_file_name}.bz2" end
backup_global_objects(file)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 103 def backup_global_objects(file) execute!("runuser - postgres -c 'pg_dumpall -g > #{file}'") end
backup_local(backup_file, extra_tar_options = {})
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 79 def backup_local(backup_file, extra_tar_options = {}) dir = extra_tar_options.fetch(:data_dir, data_dir) FileUtils.cd(dir) do tar_options = { :archive => backup_file, :command => 'create', :transform => 's,^,var/lib/pgsql/data/,S', :files => '*' }.merge(extra_tar_options) feature(:tar).run(tar_options) end end
config_files()
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 16 def config_files [ '/etc/systemd/system/postgresql.service' ] end
configuration()
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 12 def configuration raise NotImplementedError end
data_dir()
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 4 def data_dir if check_min_version('foreman', '2.0') '/var/opt/rh/rh-postgresql12/lib/pgsql/data/' else '/var/lib/pgsql/data/' end end
db_version(config = configuration)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 153 def db_version(config = configuration) if ping(config) # Note - t removes headers, -A removes alignment whitespace server_version_cmd = psql_command(config) + ' -c "SHOW server_version" -t -A' version_string = execute!(server_version_cmd, :hidden_patterns => [config['password']]) version(version_string) else raise_service_error end end
delete_records_by_ids(tbl_name, rec_ids)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 125 def delete_records_by_ids(tbl_name, rec_ids) quotize_rec_ids = rec_ids.map { |el| "'#{el}'" }.join(',') unless quotize_rec_ids.empty? psql(<<-SQL) BEGIN; DELETE FROM #{tbl_name} WHERE id IN (#{quotize_rec_ids}); COMMIT; SQL end end
dropdb(config = configuration)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 140 def dropdb(config = configuration) if local? execute!("runuser - postgres -c 'dropdb #{config['database']}'") else delete_statement = psql(<<-SQL) select string_agg('drop table if exists \"' || tablename || '\" cascade;', '') from pg_tables where schemaname = 'public'; SQL psql(delete_statement) end end
dump_db(file, config = configuration)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 55 def dump_db(file, config = configuration) execute!(dump_command(config) + " > #{file}", :hidden_patterns => [config['password']]) end
find_base_directory(directory)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 136 def find_base_directory(directory) find_dir_containing_file(directory, 'postgresql.conf') end
local?(config = configuration)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 22 def local?(config = configuration) ['localhost', '127.0.0.1', `hostname`.strip].include? config['host'] end
perform_backup(config = configuration)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 107 def perform_backup(config = configuration) file_path = backup_file_path(config) backup_cmd = backup_db_command(file_path, config) execute!(backup_cmd, :hidden_patterns => [config['password']]) puts "\n Note: Database backup file path - #{file_path}" puts "\n In case of any exception, use above dump file to restore DB." end
ping(config = configuration)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 44 def ping(config = configuration) execute?(psql_command(config), :stdin => 'SELECT 1 as ping', :hidden_patterns => [config['password']]) end
psql(query, config = configuration)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 34 def psql(query, config = configuration) if ping(config) execute(psql_command(config), :stdin => query, :hidden_patterns => [config['password']]) else raise_service_error end end
query(sql, config = configuration)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 26 def query(sql, config = configuration) parse_csv(query_csv(sql, config)) end
query_csv(sql, config = configuration)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 30 def query_csv(sql, config = configuration) psql(%{COPY (#{sql}) TO STDOUT WITH CSV HEADER}, config) end
restore_dump(file, localdb, config = configuration)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 59 def restore_dump(file, localdb, config = configuration) if localdb dump_cmd = "runuser - postgres -c 'pg_restore -C -d postgres #{file}'" execute!(dump_cmd) else # TODO: figure out how to completely ignore errors. Currently this # sometimes exits with 1 even though errors are ignored by pg_restore dump_cmd = base_command(config, 'pg_restore') + ' --no-privileges --clean --disable-triggers -n public ' \ "-d #{config['database']} #{file}" execute!(dump_cmd, :hidden_patterns => [config['password']], :valid_exit_statuses => [0, 1]) end end
restore_pg_globals(pg_globals, config = configuration)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 74 def restore_pg_globals(pg_globals, config = configuration) execute!(base_command(config, 'psql') + " -f #{pg_globals} postgres 2>/dev/null", :hidden_patterns => [config['password']]) end
table_exist?(table_name)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 115 def table_exist?(table_name) sql = <<-SQL SELECT EXISTS ( SELECT * FROM information_schema.tables WHERE table_name = '#{table_name}' ) SQL result = query(sql) return false if result.nil? || (result && result.empty?) result.first['exists'].eql?('t') end
Private Instance Methods
base_command(config, command = 'psql')
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 166 def base_command(config, command = 'psql') "PGPASSWORD='#{config[%(password)]}' "\ "#{command} -h #{config['host'] || 'localhost'} "\ " -p #{config['port'] || '5432'} -U #{config['username']}" end
dump_command(config)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 176 def dump_command(config) base_command(config, 'pg_dump') + " -Fc #{config['database']}" end
psql_command(config)
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 172 def psql_command(config) base_command(config, 'psql') + " -d #{config['database']}" end
raise_service_error()
click to toggle source
# File lib/foreman_maintain/concerns/base_database.rb, line 180 def raise_service_error raise Error::Fail, 'Please check whether database service is up & running state.' end