diff --git a/pgkit/application/pg.py b/pgkit/application/pg.py index 78d96c5..2a3129b 100644 --- a/pgkit/application/pg.py +++ b/pgkit/application/pg.py @@ -26,9 +26,10 @@ def shell(name, host, port, version, dbname, username, password, slot, replica_p master.shell() -def recovery(name, host, port, version, dbname, username, password, slot, replica_port, time_to_recover): +def recovery(name, host, port, version, dbname, username, password, slot, + replica_port, use_separate_receivewal_service, time_to_recover): master = Master(name, host, port, version, dbname, username, password, slot) - replica = Replica(master, replica_port, None) + replica = Replica(master, replica_port, None, use_separate_receivewal_service) replica.recovery(time_to_recover) diff --git a/pgkit/cli/commands/pitr.py b/pgkit/cli/commands/pitr.py index acb7b03..5962611 100644 --- a/pgkit/cli/commands/pitr.py +++ b/pgkit/cli/commands/pitr.py @@ -70,6 +70,7 @@ def recover(name, target_time): config['password'], config['slot'], config['replica_port'], + config.get('use_separate_receivewal_service', False), time_to_recover=target_time )