mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-25 19:03:46 +03:00
137 lines
3.8 KiB
Python
137 lines
3.8 KiB
Python
import datetime
|
|
import io
|
|
import struct
|
|
import timeit
|
|
from typing import Any, Iterator
|
|
|
|
from .data_codegen import DATA, Obj
|
|
|
|
ITERATIONS = 50000
|
|
|
|
|
|
def serialize_builtin(value: Any) -> bytes:
|
|
if value is None:
|
|
return b""
|
|
elif isinstance(value, bytes):
|
|
return value
|
|
elif isinstance(value, str):
|
|
return value.encode("utf-8")
|
|
elif isinstance(value, int):
|
|
return struct.pack("<i" if value < 2**32 else "<q", value)
|
|
elif isinstance(value, datetime.datetime):
|
|
return struct.pack("<i", int(value.timestamp()))
|
|
else:
|
|
raise RuntimeError(f"not a builtin type: {type(value)}")
|
|
|
|
|
|
def overhead(obj: Obj) -> None:
|
|
for v in obj.__dict__.values():
|
|
for x in v if isinstance(v, list) else [v]:
|
|
if isinstance(x, Obj):
|
|
overhead(x)
|
|
else:
|
|
serialize_builtin(x)
|
|
|
|
|
|
def strategy_concat(obj: Obj) -> bytes:
|
|
res = b""
|
|
for v in obj.__dict__.values():
|
|
for x in v if isinstance(v, list) else [v]:
|
|
if isinstance(x, Obj):
|
|
res += strategy_concat(x)
|
|
else:
|
|
res += serialize_builtin(x)
|
|
return res
|
|
|
|
|
|
def strategy_append(obj: Obj) -> bytes:
|
|
res = bytearray()
|
|
for v in obj.__dict__.values():
|
|
for x in v if isinstance(v, list) else [v]:
|
|
if isinstance(x, Obj):
|
|
res += strategy_append(x)
|
|
else:
|
|
res += serialize_builtin(x)
|
|
return bytes(res)
|
|
|
|
|
|
def strategy_append_reuse(obj: Obj) -> bytes:
|
|
def do_append(o: Obj, res: bytearray) -> None:
|
|
for v in o.__dict__.values():
|
|
for x in v if isinstance(v, list) else [v]:
|
|
if isinstance(x, Obj):
|
|
do_append(x, res)
|
|
else:
|
|
res += serialize_builtin(x)
|
|
|
|
buffer = bytearray()
|
|
do_append(obj, buffer)
|
|
return bytes(buffer)
|
|
|
|
|
|
def strategy_join(obj: Obj) -> bytes:
|
|
return b"".join(
|
|
strategy_join(x) if isinstance(x, Obj) else serialize_builtin(x)
|
|
for v in obj.__dict__.values()
|
|
for x in (v if isinstance(v, list) else [v])
|
|
)
|
|
|
|
|
|
def strategy_join_flat(obj: Obj) -> bytes:
|
|
def flatten(o: Obj) -> Iterator[bytes]:
|
|
for v in o.__dict__.values():
|
|
for x in v if isinstance(v, list) else [v]:
|
|
if isinstance(x, Obj):
|
|
yield from flatten(x)
|
|
else:
|
|
yield serialize_builtin(x)
|
|
|
|
return b"".join(flatten(obj))
|
|
|
|
|
|
def strategy_write(obj: Obj) -> bytes:
|
|
def do_write(o: Obj, buffer: io.BytesIO) -> None:
|
|
for v in o.__dict__.values():
|
|
for x in v if isinstance(v, list) else [v]:
|
|
if isinstance(x, Obj):
|
|
do_write(x, buffer)
|
|
else:
|
|
buffer.write(serialize_builtin(x))
|
|
|
|
buffer = io.BytesIO()
|
|
do_write(obj, buffer)
|
|
return buffer.getvalue()
|
|
|
|
|
|
def main() -> None:
|
|
strategies = [
|
|
v
|
|
for _, v in sorted(
|
|
((k, v) for k, v in globals().items() if k.startswith("strategy_")),
|
|
key=lambda t: t[0],
|
|
)
|
|
]
|
|
for a, b in zip(strategies[:-1], strategies[1:]):
|
|
if a(DATA) != b(DATA):
|
|
raise ValueError("strategies produce different output")
|
|
|
|
print("measuring overhead...", end="", flush=True)
|
|
overhead_duration = timeit.timeit(
|
|
"strategy(DATA)",
|
|
number=ITERATIONS,
|
|
globals={"strategy": overhead, "DATA": DATA},
|
|
)
|
|
print(f" {overhead_duration:.04f}s")
|
|
|
|
for strategy in strategies:
|
|
duration = timeit.timeit(
|
|
"strategy(DATA)",
|
|
number=ITERATIONS,
|
|
globals={"strategy": strategy, "DATA": DATA},
|
|
)
|
|
print(f"{strategy.__name__:.>30} took {duration - overhead_duration:.04f}s")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|