Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(255)

Side by Side Diff: scripts/slave/recipe_modules/goma/resources/cloudtail_utils.py

Issue 2237403002: Add cloudtail start/stop in recipe_modules/goma (Closed) Base URL: https://chromium.googlesource.com/chromium/tools/build.git@master
Patch Set: remove import os Created 4 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 #!/usr/bin/env python
2 # Copyright (c) 2016 The Chromium Authors. All rights reserved.
3 # Use of this source code is governed by a BSD-style license that can be
4 # found in the LICENSE file.
5
6 import argparse
7 import os
8 import signal
9 import subprocess
10 import sys
11
12 from slave import goma_utils
13
14
15 def start_cloudtail(args, f):
16 """Write process id of started cloudtail to file object f"""
17
18 proc = subprocess.Popen([args.cloudtail_path,
19 'tail',
20 '--log-id', 'goma_compiler_proxy'
21 '--path',
22 goma_utils.GetLatestGomaCompilerProxyInfo()])
23
24 f.write(str(proc.pid))
25
26
27 def main():
28 parser = argparse.ArgumentParser(
29 description='cloudtail utility for goma recipe module.')
30
31 parser.add_argument('--cloudtail-path',
32 help='path of cloudtail binary')
33 parser.add_argument('--start-cloudtail', action='store_true',
34 help='start cloudtail if this option is enabled')
35 parser.add_argument('--killed-pid',
36 help=('pid that is killed.'
37 ' Do not set this option and'
38 ' --start-cloudtail at the same time'))
39
40 args = parser.parse_args()
41
42 if args.start_cloudtail:
43 # Do not set killed_pid and start_cloudtail at the same time
44 assert args.killed_pid is None
45 start_cloudtail(args, sys.stdout)
46
47 if args.killed_pid is not None:
48 killed_pid = int(args.killed_pid)
49 os.kill(killed_pid, signal.SIGTERM)
50 pid, retcode = os.waitpid(killed_pid, os.WNOHANG)
Yoshisato Yanagisawa 2016/08/16 03:45:23 Don't you need to retry for a while?
tikuta 2016/08/16 03:59:31 Done.
51 if pid == 0 and retcode == 0:
52 os.kill(killed_pid, os.SIGKILL)
53
54
55 if '__main__' == __name__:
56 sys.exit(main())
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698