| Index: scripts/slave/goma_utils.py
|
| diff --git a/scripts/slave/goma_utils.py b/scripts/slave/goma_utils.py
|
| index 85c1f31ee6a0b0dd0adc0e858de04e6dbf955e46..1b1bed5336e498e9925b27e19ef4b4297477b21b 100644
|
| --- a/scripts/slave/goma_utils.py
|
| +++ b/scripts/slave/goma_utils.py
|
| @@ -78,7 +78,8 @@ def GetLatestGomaCompilerProxySubprocInfo():
|
| return GetLatestGlogInfoFile('compiler_proxy-subproc')
|
|
|
|
|
| -def UploadToGomaLogGS(file_path, gs_filename, text_to_append=None):
|
| +def UploadToGomaLogGS(
|
| + file_path, gs_filename, text_to_append=None, override_gsutil=None):
|
| """Upload a file to Google Cloud Storage (gs://chrome-goma-log).
|
|
|
| Note that the uploaded file would automatically be gzip compressed.
|
| @@ -104,14 +105,14 @@ def UploadToGomaLogGS(file_path, gs_filename, text_to_append=None):
|
| shutil.copyfileobj(f_in, gzipf_out)
|
| if text_to_append:
|
| gzipf_out.write(text_to_append)
|
| - slave_utils.GSUtilCopy(temp.name, gs_path)
|
| + slave_utils.GSUtilCopy(temp.name, gs_path, override_gsutil=override_gsutil)
|
| print "Copied log file to %s" % gs_path
|
| finally:
|
| os.remove(temp.name)
|
| return log_path
|
|
|
|
|
| -def UploadGomaCompilerProxyInfo():
|
| +def UploadGomaCompilerProxyInfo(override_gsutil=None):
|
| """Upload goma compiler_proxy.INFO to Google Storage."""
|
| latest_subproc_info = GetLatestGomaCompilerProxySubprocInfo()
|
| if latest_subproc_info:
|
| @@ -125,13 +126,16 @@ def UploadGomaCompilerProxyInfo():
|
| return
|
| # Since a filename of compiler_proxy.INFO is fairly unique,
|
| # we might be able to upload it as-is.
|
| - log_path = UploadToGomaLogGS(latest_info, os.path.basename(latest_info))
|
| + log_path = UploadToGomaLogGS(
|
| + latest_info, os.path.basename(latest_info),
|
| + override_gsutil=override_gsutil)
|
| viewer_url = ('http://chromium-build-stats.appspot.com/compiler_proxy_log/'
|
| + log_path)
|
| print 'Visualization at %s' % viewer_url
|
|
|
|
|
| -def UploadNinjaLog(outdir, compiler, command, exit_status):
|
| +def UploadNinjaLog(
|
| + outdir, compiler, command, exit_status, override_gsutil=None):
|
| """Upload .ninja_log to Google Cloud Storage (gs://chrome-goma-log),
|
| in the same folder with goma's compiler_proxy.INFO.
|
|
|
| @@ -173,7 +177,8 @@ def UploadNinjaLog(outdir, compiler, command, exit_status):
|
| hostname, username, mtime.strftime('%Y%m%d-%H%M%S'), pid)
|
| additional_text = '# end of ninja log\n' + json.dumps(info)
|
| log_path = UploadToGomaLogGS(
|
| - ninja_log_path, ninja_log_filename, additional_text)
|
| + ninja_log_path, ninja_log_filename, additional_text,
|
| + override_gsutil=override_gsutil)
|
| viewer_url = 'http://chromium-build-stats.appspot.com/ninja_log/' + log_path
|
| print 'Visualization at %s' % viewer_url
|
|
|
|
|