|
|
@ -37,7 +37,7 @@ class Writer(multiprocessing.Process): |
|
|
|
if recs_queue: # MP |
|
|
|
super(Writer, self).__init__() |
|
|
|
self.recs_queue = recs_queue |
|
|
|
self.log = logging.getLogger(f"wr") |
|
|
|
self.log = logging.getLogger("wr") |
|
|
|
self.autosave_interval = autosave_interval |
|
|
|
self.ip4_cache = ip4_cache |
|
|
|
if not recs_queue: # No MP |
|
|
@ -45,7 +45,7 @@ class Writer(multiprocessing.Process): |
|
|
|
|
|
|
|
def open_db(self) -> None: |
|
|
|
self.db = database.Database() |
|
|
|
self.db.log = logging.getLogger(f"wr") |
|
|
|
self.db.log = logging.getLogger("wr") |
|
|
|
self.db.fill_ip4cache(max_size=self.ip4_cache) |
|
|
|
|
|
|
|
def exec_record(self, record: Record) -> None: |
|
|
@ -74,6 +74,7 @@ class Writer(multiprocessing.Process): |
|
|
|
block: typing.List[Record] |
|
|
|
for block in iter(self.recs_queue.get, None): |
|
|
|
|
|
|
|
assert block |
|
|
|
record: Record |
|
|
|
for record in block: |
|
|
|
self.exec_record(record) |
|
|
|