SERVER-82600 Add initial deflakinator script (#18994)

GitOrigin-RevId: 28acd1a9129aa6eb21f36736a3e2f84d49889927
This commit is contained in:
Zack Winter 2024-02-14 16:29:58 -08:00 committed by MongoDB Bot
parent fa000510e4
commit 8daab4ef5c
1 changed files with 62 additions and 0 deletions

62
buildscripts/deflakinator.py Executable file
View File

@ -0,0 +1,62 @@
#!/usr/bin/env python3
import argparse
import subprocess
import uuid
import json
_MAX_RUNS = 1000
def print_pb_version_ids(run_id: str):
result = subprocess.run(f"evergreen list-patches -j -n {_MAX_RUNS * 2}", shell=True, check=True,
stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
data = json.loads(result.stdout)
filtered_objects = [
obj for obj in data if obj.get("description") == f"flakinator run id: {run_id}"
]
versions = []
for obj in filtered_objects:
versions += [obj["version"]]
print("---")
print(
"For analysis, visit",
f"https://ui.honeycomb.io/mongodb-4b/environments/production?query=%7B%22time_range%22%3A86400%2C%22granularity%22%3A0%2C%22breakdowns%22%3A%5B%22evergreen.task.name%22%5D%2C%22calculations%22%3A%5B%7B%22op%22%3A%22COUNT%22%7D%5D%2C%22filters%22%3A%5B%7B%22column%22%3A%22evergreen.task.name%22%2C%22op%22%3A%22!%3D%22%2C%22value%22%3A%22lint_repo%22%7D%2C%7B%22column%22%3A%22evergreen.project.id%22%2C%22op%22%3A%22%3D%22%2C%22value%22%3A%22mongodb-mongo-master%22%7D%2C%7B%22column%22%3A%22evergreen.task.status%22%2C%22op%22%3A%22%3D%22%2C%22value%22%3A%22failed%22%7D%2C%7B%22column%22%3A%22evergreen.version.id%22%2C%22op%22%3A%22in%22%2C%22value%22%3A%5B%22\
{'%22%2C%22'.join(versions)}\
%22%5D%7D%5D%2C%22filter_combination%22%3A%22AND%22%2C%22orders%22%3A%5B%7B%22op%22%3A%22COUNT%22%2C%22order%22%3A%22descending%22%7D%5D%2C%22havings%22%3A%5B%5D%2C%22limit%22%3A1000%7D"
)
print("---")
def deflakinator(runs: int, evergreen_args: str):
"""Run the deflakinator."""
run_id = uuid.uuid4()
print("Kicking off evergreen patch builds:")
print("---")
command = f"evergreen patch {evergreen_args} --yes --finalize --description \"flakinator run id: {run_id}\""
for _ in range(runs):
print(command)
subprocess.run(command, shell=True, check=True, stdout=subprocess.PIPE,
stderr=subprocess.PIPE, text=True)
print_pb_version_ids(run_id)
def main():
"""Run the main function."""
parser = argparse.ArgumentParser()
parser.add_argument("--runs", type=int, choices=range(1, _MAX_RUNS), metavar=f"[1-{_MAX_RUNS}]",
help="Number of times to run each patch build")
parser.add_argument("--evergreen-args", type=str, help="Arguments to pass to evergreen patch")
args = parser.parse_args()
deflakinator(args.runs, args.evergreen_args)
if __name__ == "__main__":
main()