VVladimir Mandiccleanup performance tags
17c8fe91创建于 7月3日历史提交
import os
import time
import sys


try:
    default_min_time = float(os.environ.get('SD_MIN_TIMER', '0.1'))
except Exception:
    default_min_time = 0.1


class Timer:
    def __init__(self):
        self.start = time.time()
        self.records = {}
        self.total = 0
        self.profile = False

    def elapsed(self, reset=True):
        end = time.time()
        res = end - self.start
        if reset:
            self.start = end
        return res

    def add(self, name, t):
        if name not in self.records:
            self.records[name] = 0
        self.records[name] += t

    def ts(self, name, t):
        elapsed = time.time() - t
        self.add(name, elapsed)

    def record(self, category=None, extra_time=0, reset=True):
        e = self.elapsed(reset)
        if category is None:
            category = sys._getframe(1).f_code.co_name # pylint: disable=protected-access
        if category not in self.records:
            self.records[category] = 0
        self.records[category] += e + extra_time
        self.total += e + extra_time

    def summary(self, min_time=default_min_time, max_time=-1, total=True):
        if self.profile:
            min_time = -1
        self.total = sum(self.records.values())
        res = f"total={self.total:.2f} " if total else ''
        additions = [x for x in self.records.items() if x[1] >= min_time]
        if max_time > 0:
            additions = [x for x in additions if x[1] <= max_time]
        if not total:
            additions = [x for x in additions if x[0] != 'total']
        additions = sorted(additions, key=lambda x: x[1], reverse=True)
        if not additions:
            return res
        res += " ".join([f"{category}={time_taken:.2f}" for category, time_taken in additions])
        return res

    def get_total(self):
        return sum(self.records.values())

    def dct(self, min_time=default_min_time):
        self.total = sum(self.records.values())
        self.records['total'] = self.total
        if self.profile:
            res = {k: round(v, 4) for k, v in self.records.items()}
        else:
            res = {k: round(v, 2) for k, v in self.records.items() if v >= min_time}
        res = {k: v for k, v in sorted(res.items(), key=lambda x: x[1], reverse=True)} # noqa: C416 # pylint: disable=unnecessary-comprehension
        return res

    def reset(self):
        self.__init__()

startup = Timer()
process = Timer()
launch = Timer()
init = Timer()
load = Timer()