Merge pull request #2161 from lonvia/timeout-for-replication

Reintroduce timeout for replication file download
This commit is contained in:
Sarah Hoffmann 2021-02-04 16:52:24 +01:00 committed by GitHub
commit 82ef02cd1a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -4,6 +4,7 @@ database administration and querying.
"""
import datetime as dt
import os
import socket
import sys
import time
import argparse
@ -261,11 +262,16 @@ class UpdateReplication:
together with --once""")
group.add_argument('--osm2pgsql-cache', metavar='SIZE', type=int,
help='Size of cache to be used by osm2pgsql (in MB)')
group = parser.add_argument_group('Download parameters')
group.add_argument('--socket-timeout', dest='socket_timeout', type=int, default=60,
help='Set timeout for file downloads.')
@staticmethod
def _init_replication(args):
from .tools import replication, refresh
socket.setdefaulttimeout(args.socket_timeout)
LOG.warning("Initialising replication updates")
conn = connect(args.config.get_libpq_dsn())
replication.init_replication(conn, base_url=args.config.REPLICATION_URL)