comparison mercurial/util.py @ 51549:a452807df09b

nocg: make the utility work are both a decorator and context manager

In some case, the context manager version will be simpler.

51548:1b17eeba9deb

import time
import traceback
import warnings

from typing import (
    Iterable,
    Iterator,
    List,
    Optional,
    Tuple,

def never(fn):
    return False


def nogc(func):
    """disable garbage collector

    Python's garbage collector triggers a GC each time a certain number of
    container objects (the number being defined by gc.get_threshold()) are
    allocated even when marked not to be tracked by the collector. Tracking has

    containers.

    This garbage collector issue have been fixed in 2.7. But it still affect
    CPython's performance.
    """

    def wrapper(*args, **kwargs):
        gcenabled = gc.isenabled()
        gc.disable()
        try:
            return func(*args, **kwargs)
        finally:
            if gcenabled:
                gc.enable()

    return wrapper


if pycompat.ispypy:

51549:a452807df09b

import time
import traceback
import warnings

from typing import (
    Any,
    Iterable,
    Iterator,
    List,
    Optional,
    Tuple,

def never(fn):
    return False


def nogc(func=None) -> Any:
    """disable garbage collector

    Python's garbage collector triggers a GC each time a certain number of
    container objects (the number being defined by gc.get_threshold()) are
    allocated even when marked not to be tracked by the collector. Tracking has

    containers.

    This garbage collector issue have been fixed in 2.7. But it still affect
    CPython's performance.
    """
    if func is None:
        return _nogc_context()
    else:
        return _nogc_decorator(func)


@contextlib.contextmanager
def _nogc_context():
    gcenabled = gc.isenabled()
    gc.disable()
    try:
        yield
    finally:
        if gcenabled:
            gc.enable()


def _nogc_decorator(func):
    def wrapper(*args, **kwargs):
        with _nogc_context():
            return func(*args, **kwargs)

    return wrapper


if pycompat.ispypy: