mirror of
https://github.com/mongodb/mongo.git
synced 2024-11-24 08:30:56 +01:00
132 lines
4.3 KiB
Python
132 lines
4.3 KiB
Python
# Copyright 2023 MongoDB Inc.
|
|
#
|
|
# Permission is hereby granted, free of charge, to any person obtaining
|
|
# a copy of this software and associated documentation files (the
|
|
# "Software"), to deal in the Software without restriction, including
|
|
# without limitation the rights to use, copy, modify, merge, publish,
|
|
# distribute, sublicense, and/or sell copies of the Software, and to
|
|
# permit persons to whom the Software is furnished to do so, subject to
|
|
# the following conditions:
|
|
#
|
|
# The above copyright notice and this permission notice shall be included
|
|
# in all copies or substantial portions of the Software.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
|
|
# KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
|
|
# WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
|
|
# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
|
|
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
|
# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
#
|
|
|
|
import functools
|
|
import os
|
|
import random
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
from typing import Callable, Dict, List
|
|
|
|
import SCons
|
|
|
|
|
|
def command_spawn_func(
|
|
sh: str,
|
|
escape: Callable[[str], str],
|
|
cmd: str,
|
|
args: List,
|
|
env: Dict,
|
|
target: List,
|
|
source: List,
|
|
):
|
|
retries = 0
|
|
success = False
|
|
|
|
build_env = target[0].get_build_env()
|
|
max_retries = build_env.get("BUILD_RETRY_ATTEMPTS", 10)
|
|
build_max_retry_delay = build_env.get("BUILD_RETRY_MAX_DELAY_SECONDS", 120)
|
|
|
|
while not success and retries <= max_retries:
|
|
try:
|
|
start_time = time.time()
|
|
if sys.platform[:3] == "win":
|
|
# have to use shell=True for windows because of https://github.com/python/cpython/issues/53908
|
|
proc = subprocess.run(
|
|
" ".join(args),
|
|
env=env,
|
|
close_fds=True,
|
|
shell=True,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.STDOUT,
|
|
text=True,
|
|
check=True,
|
|
)
|
|
else:
|
|
proc = subprocess.run(
|
|
[sh, "-c", " ".join(args)],
|
|
env=env,
|
|
close_fds=True,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.STDOUT,
|
|
text=True,
|
|
check=True,
|
|
)
|
|
except subprocess.CalledProcessError as exc:
|
|
print(f"{os.path.basename(__file__)} captured error:")
|
|
print(exc.stdout)
|
|
retries += 1
|
|
retry_delay = int((time.time() - start_time) + build_max_retry_delay * random.random())
|
|
print(
|
|
f"Failed while trying to build {target[0]}",
|
|
)
|
|
if retries <= max_retries:
|
|
print(f"trying again in {retry_delay} seconds with retry attempt {retries}")
|
|
time.sleep(retry_delay)
|
|
continue
|
|
|
|
# No more retries left
|
|
return exc.returncode
|
|
else:
|
|
if proc.stdout:
|
|
print(proc.stdout)
|
|
return proc.returncode
|
|
|
|
|
|
def generate(env):
|
|
original_command_execute = SCons.Action.CommandAction.execute
|
|
|
|
def build_retry_execute(command_action_instance, target, source, env, executor=None):
|
|
if (
|
|
"conftest" not in str(target[0])
|
|
and target[0].has_builder()
|
|
and target[0].get_builder().get_name(env)
|
|
in [
|
|
"Object",
|
|
"SharedObject",
|
|
"StaticObject",
|
|
"Program",
|
|
"StaticLibrary",
|
|
"SharedLibrary",
|
|
]
|
|
):
|
|
original_spawn = env["SPAWN"]
|
|
|
|
env["SPAWN"] = functools.partial(command_spawn_func, target=target, source=source)
|
|
result = original_command_execute(
|
|
command_action_instance, target, source, env, executor
|
|
)
|
|
env["SPAWN"] = original_spawn
|
|
|
|
else:
|
|
result = original_command_execute(
|
|
command_action_instance, target, source, env, executor
|
|
)
|
|
return result
|
|
|
|
SCons.Action.CommandAction.execute = build_retry_execute
|
|
|
|
|
|
def exists(env):
|
|
return True
|