Browse Source

Profiling is now optional

master
Geoffrey Frogeye 2 years ago
parent
commit
bb9e6de62f
Signed by: geoffrey GPG Key ID: D8A7ECA00A8CD3DD
  1. 1
      .env.default
  2. 28
      database.py

1
.env.default

@ -1,3 +1,4 @@
RAPID7_API_KEY=
CACHE_SIZE=536870912
MASSDNS_HASHMAP_SIZE=1000
PROFILE=0

28
database.py

@ -11,6 +11,7 @@ import coloredlogs
import pickle
import numpy
import math
import os
TLD_LIST: typing.Set[str] = set()
@ -127,13 +128,23 @@ MatchCallable = typing.Callable[[Path,
class Profiler():
def __init__(self) -> None:
self.log = logging.getLogger('profiler')
self.time_last = time.perf_counter()
self.time_step = 'init'
self.time_dict: typing.Dict[str, float] = dict()
self.step_dict: typing.Dict[str, int] = dict()
do_profile = int(os.environ.get('PROFILE', '0'))
if do_profile:
self.log = logging.getLogger('profiler')
self.time_last = time.perf_counter()
self.time_step = 'init'
self.time_dict: typing.Dict[str, float] = dict()
self.step_dict: typing.Dict[str, int] = dict()
self.enter_step = self.enter_step_real
self.profile = self.profile_real
else:
self.enter_step = self.enter_step_dummy
self.profile = self.profile_dummy
def enter_step(self, name: str) -> None:
def enter_step_dummy(self, name: str) -> None:
return
def enter_step_real(self, name: str) -> None:
now = time.perf_counter()
try:
self.time_dict[self.time_step] += now - self.time_last
@ -144,7 +155,10 @@ class Profiler():
self.time_step = name
self.time_last = time.perf_counter()
def profile(self) -> None:
def profile_dummy(self) -> None:
return
def profile_real(self) -> None:
self.enter_step('profile')
total = sum(self.time_dict.values())
for key, secs in sorted(self.time_dict.items(), key=lambda t: t[1]):

Loading…
Cancel
Save