comparison contrib/perf.py @ 51560:8e8776a28683

perf: create the temporary target next to the source in stream-consume

See inline comment for rational.

51558:c4aab3661f25

    if not (os.path.isfile(filename) and os.access(filename, os.R_OK)):
        raise error.Abort("not a readable file: %s" % filename)

    run_variables = [None, None]

    @contextlib.contextmanager
    def context():
        with open(filename, mode='rb') as bundle:
            with tempfile.TemporaryDirectory() as tmp_dir:
                tmp_dir = fsencode(tmp_dir)
                run_variables[0] = bundle
                run_variables[1] = tmp_dir
                yield
                run_variables[0] = None

51560:8e8776a28683

    if not (os.path.isfile(filename) and os.access(filename, os.R_OK)):
        raise error.Abort("not a readable file: %s" % filename)

    run_variables = [None, None]

    # we create the new repository next to the other one for two reasons:
    # - this way we use the same file system, which are relevant for benchmark
    # - if /tmp/ is small, the operation could overfills it.
    source_repo_dir = os.path.dirname(repo.root)

    @contextlib.contextmanager
    def context():
        with open(filename, mode='rb') as bundle:
            with tempfile.TemporaryDirectory(
                prefix=b'hg-perf-stream-consume-',
                dir=source_repo_dir,
            ) as tmp_dir:
                tmp_dir = fsencode(tmp_dir)
                run_variables[0] = bundle
                run_variables[1] = tmp_dir
                yield
                run_variables[0] = None