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
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
# 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
# File lib/foreman_maintain/concerns/base_database.rb, line 103 def backup_global_objects(file) execute!("runuser - postgres -c 'pg_dumpall -g > #{file}'") end
# 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,^,#{data_dir[1..-1]},S", :files => '*' }.merge(extra_tar_options) feature(:tar).run(tar_options) end end
# File lib/foreman_maintain/concerns/base_database.rb, line 16 def config_files [ '/etc/systemd/system/postgresql.service' ] end
# File lib/foreman_maintain/concerns/base_database.rb, line 12 def configuration raise NotImplementedError end
# 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
# File lib/foreman_maintain/concerns/base_database.rb, line 154 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
# File lib/foreman_maintain/concerns/base_database.rb, line 126 def delete_records_by_ids(tbl_name, rec_ids) quotize_rec_ids = rec_ids.map { |el| "'#{el}'" }.join(',') unless quotize_rec_ids.empty? psql(" BEGIN; DELETE FROM #{tbl_name} WHERE id IN (#{quotize_rec_ids}); COMMIT; ") end end
# File lib/foreman_maintain/concerns/base_database.rb, line 141 def dropdb(config = configuration) if local? execute!("runuser - postgres -c 'dropdb #{config['database']}'") else delete_statement = psql(" select string_agg('drop table if exists \"' || tablename || '\" cascade;', '') from pg_tables where schemaname = 'public'; ") psql(delete_statement) end end
# 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
# File lib/foreman_maintain/concerns/base_database.rb, line 137 def find_base_directory(directory) find_dir_containing_file(directory, 'postgresql.conf') end
# File lib/foreman_maintain/concerns/base_database.rb, line 22 def local?(config = configuration) ['localhost', '127.0.0.1', %xhostname`.strip].include? config['host'] end
# 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
# 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
# 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
# File lib/foreman_maintain/concerns/base_database.rb, line 165 def psql_cmd_available? exit_status, _output = execute_with_status('which psql') exit_status == 0 end
# File lib/foreman_maintain/concerns/base_database.rb, line 26 def query(sql, config = configuration) parse_csv(query_csv(sql, config)) end
# File lib/foreman_maintain/concerns/base_database.rb, line 30 def query_csv(sql, config = configuration) psql(%Q{COPY (#{sql}) TO STDOUT WITH CSV HEADER}, config) end
# File lib/foreman_maintain/concerns/base_database.rb, line 170 def raise_psql_missing_error raise Error::Fail, 'The psql command not found.' ' Make sure system has psql utility installed.' end
# 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
# 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
# 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
# File lib/foreman_maintain/concerns/base_database.rb, line 177 def base_command(config, command = 'psql') "PGPASSWORD='#{config[%(password)]}' " "#{command} -h #{config['host'] || 'localhost'} " " -p #{config['port'] || '5432'} -U #{config['username']}" end
# File lib/foreman_maintain/concerns/base_database.rb, line 187 def dump_command(config) base_command(config, 'pg_dump') + " -Fc #{config['database']}" end
# File lib/foreman_maintain/concerns/base_database.rb, line 183 def psql_command(config) base_command(config, 'psql') + " -d #{config['database']}" end
# File lib/foreman_maintain/concerns/base_database.rb, line 191 def raise_service_error raise Error::Fail, 'Please check whether database service is up & running state.' end