mirror of
https://github.com/facebook/sapling.git
synced 2024-10-09 08:18:15 +03:00
98d9269874
Summary: Create a fork of the Mercurial code that we can use to build server rpms. The hg servers will continue to exist for a few more months while we move the darkstorm and ediscovery use cases off them. In the mean time, we want to start making breaking changes to the client, so let's create a stable copy of the hg code to produce rpms for the hg servers. The fork is based off c7770c78d, the latest hg release. This copies the files as is, then adds some minor tweaks to get it to build: - Disables some lint checks that appear to be bypassed by path - sed replace eden/scm with eden/hg-server - Removed a dependency on scm/telemetry from the edenfs-client tests since scm/telemetry pulls in the original eden/scm/lib/configparser which conflicts with the hg-server conflict parser. allow-large-files Reviewed By: quark-zju Differential Revision: D27632557 fbshipit-source-id: b2f442f4ec000ea08e4d62de068750832198e1f4
95 lines
2.4 KiB
Python
95 lines
2.4 KiB
Python
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2.
|
|
|
|
# perftrace.py - Module for tracing performance
|
|
|
|
import inspect
|
|
import threading
|
|
from contextlib import contextmanager
|
|
|
|
from bindings import tracing
|
|
|
|
from . import util
|
|
|
|
|
|
# Native tracing utilities
|
|
|
|
tracer = util.tracer
|
|
|
|
threadlocal = threading.local()
|
|
|
|
|
|
def editspan(
|
|
meta,
|
|
_edit=tracer.edit,
|
|
_currentframe=inspect.currentframe,
|
|
_wrapfunc=tracing.wrapfunc,
|
|
_spanid=tracing.wrapfunc.spanid,
|
|
):
|
|
"""Edit the current native span. meta: [(key, value)]"""
|
|
stack = threadlocal.__dict__.get("stack", [])
|
|
if not stack:
|
|
return
|
|
spanid = stack[-1]
|
|
_edit(spanid, meta)
|
|
|
|
|
|
# PerfTrace wrappers
|
|
|
|
|
|
@contextmanager
|
|
def trace(name):
|
|
spanid = tracer.span([("name", name), ("cat", "perftrace")])
|
|
threadlocal.__dict__.setdefault("stack", []).append(spanid)
|
|
tracer.enter(spanid)
|
|
try:
|
|
yield
|
|
finally:
|
|
tracer.exit(spanid)
|
|
threadlocal.stack.pop()
|
|
|
|
|
|
def traceflag(flagname):
|
|
"""Records the given flag name as being associated with the latest trace
|
|
point."""
|
|
# XXX: No multi-flag support for now.
|
|
editspan([("%s" % flagname, "true")])
|
|
|
|
|
|
def tracevalue(name, value):
|
|
"""Records the given name=value as being associated with the latest trace
|
|
point. This will overwrite any previous value with that name."""
|
|
editspan([(name, str(value))])
|
|
|
|
|
|
def tracebytes(name, value):
|
|
"""Records the given name=value as being associated with the latest trace
|
|
point. The value is treated as bytes and will be added to any previous value
|
|
set to the same name."""
|
|
# XXX: Rust tracing does not do an addition - But there do not seem to be
|
|
# any users relying on the behavior.
|
|
editspan([(name, str(value))])
|
|
|
|
|
|
def tracefunc(name):
|
|
"""Decorator that indicates this entire function should be wrapped in a
|
|
trace."""
|
|
|
|
def wrapper(func):
|
|
func.meta = [("name", name), ("cat", "perftrace")]
|
|
if util.istest():
|
|
func.meta.append(("line", "_"))
|
|
func.isperftrace = True
|
|
|
|
def pushspan(spanid):
|
|
threadlocal.__dict__.setdefault("stack", []).append(spanid)
|
|
|
|
def popspan():
|
|
threadlocal.__dict__.setdefault("stack", []).pop()
|
|
|
|
return tracing.wrapfunc(func, push_callback=pushspan, pop_callback=popspan)
|
|
|
|
return wrapper
|