kache 1.2.2

Compiler cache for Rust, C/C++ and CUDA, with S3, GCS, shared-filesystem and OCI remotes.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

"""Running the measuring instrument and finding the tools it measures."""

import os
import shutil
import signal
import subprocess
from pathlib import Path


# The longest one engine pass may take. A run's own deadline can shorten it.
MEASUREMENT_TIMEOUT = 1200


def run_measurement(command, timeout=MEASUREMENT_TIMEOUT, grace=0, **kwargs):
    """Run one measurement in its own process group.

    A failed or timed-out measurement can leave compiler children alive, so
    the whole group is stopped before its scratch directory is removed. With `grace`,
    the group gets SIGINT first and that many seconds to clean up: the
    contention driver starts its builds in groups of their own and only it
    can stop them.
    """
    with subprocess.Popen(command, start_new_session=True, **kwargs) as process:
        try:
            status = process.wait(timeout=timeout)
            if status:
                raise subprocess.CalledProcessError(status, command)
        except BaseException:
            try:
                if grace:
                    os.killpg(process.pid, signal.SIGINT)
                    try:
                        process.wait(timeout=grace)
                    except subprocess.TimeoutExpired:
                        pass
                # Also after a clean exit: children can outlive the leader.
                os.killpg(process.pid, signal.SIGKILL)
            except ProcessLookupError:
                pass
            process.wait()
            raise


def tool_path(binary):
    """Absolute path of a tool, real binary rather than a mise shim.

    A mise shim is a symlink to the mise binary that dispatches on argv[0].
    Following it gives `.../mise`, which invoked as `mise --start-server`
    fails; ask mise where the tool really is instead.
    """
    found = Path(shutil.which(binary) or binary).absolute()
    if found.is_symlink() and Path(os.path.realpath(found)).stem == "mise":
        real = subprocess.run(
            ["mise", "which", binary], capture_output=True, text=True, check=False
        )
        target = real.stdout.strip()
        if real.returncode == 0 and target:
            return str(Path(target).absolute())
        return str(found)
    return str(found.resolve())


def installed(binary):
    return Path(binary).is_file() or shutil.which(binary) is not None