osm-search/Nominatim
1
1
Fork 0
mirror of https://github.com/osm-search/Nominatim.git synced 2024-11-23 05:35:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

improve progress logging during indexing

Browse Source 
Wait for 2 seconds before logging the first progress, so that we
have numbers that are a bit more reliable statistically speaking.

Also provides an actual implementation for the log_interval
parameter and fixes some small style issues.
This commit is contained in:
Sarah Hoffmann 2020-11-30 10:59:29 +01:00
parent 2e5c8b5cd3
commit 5016eace34
1 changed files with 29 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
nominatim/indexer/progress.py
View File

@ -2,13 +2,17 @@
#
# This file is part of Nominatim.
# Copyright (C) 2020 Sarah Hoffmann
"""
Helpers for progress logging.
"""
import logging
from datetime import datetime
log = logging.getLogger()
LOG = logging.getLogger()
class ProgressLogger(object):
INITIAL_PROGRESS = 10
class ProgressLogger:
""" Tracks and prints progress for the indexing process.
`name` is the name of the indexing step being tracked.
`total` sets up the total number of items that need processing.
@ -21,32 +25,40 @@ class ProgressLogger(object):
self.total_places = total
self.done_places = 0
self.rank_start_time = datetime.now()
self.next_info = 100 if log.isEnabledFor(logging.INFO) else total + 1
self.log_interval = log_interval
self.next_info = INITIAL_PROGRESS if LOG.isEnabledFor(logging.INFO) else total + 1
def add(self, num=1):
""" Mark `num` places as processed. Print a log message if the
logging is at least info and the log interval has past.
logging is at least info and the log interval has passed.
"""
self.done_places += num
if self.done_places >= self.next_info:
if self.done_places < self.next_info:
return
now = datetime.now()
done_time = (now - self.rank_start_time).total_seconds()
if done_time < 2:
self.next_info = self.done_places + INITIAL_PROGRESS
return
places_per_sec = self.done_places / done_time
eta = (self.total_places - self.done_places)/places_per_sec
eta = (self.total_places - self.done_places) / places_per_sec
log.info("Done {} in {} @ {:.3f} per second - {} ETA (seconds): {:.2f}"
.format(self.done_places, int(done_time),
places_per_sec, self.name, eta))
LOG.info("Done %d in %d @ %.3f per second - %s ETA (seconds): %.2f",
self.done_places, int(done_time),
places_per_sec, self.name, eta)
self.next_info += int(places_per_sec)
self.next_info += int(places_per_sec) * self.log_interval
def done(self):
""" Print final staticstics about the progress.
""" Print final statistics about the progress.
"""
rank_end_time = datetime.now()
diff_seconds = (rank_end_time-self.rank_start_time).total_seconds()
log.warning("Done {}/{} in {} @ {:.3f} per second - FINISHED {}\n".format(
LOG.warning("Done %d/%d in %d @ %.3f per second - FINISHED %s\n",
self.done_places, self.total_places, int(diff_seconds),
self.done_places/diff_seconds, self.name))
self.done_places/diff_seconds, self.name)