| Index: appengine/swarming/server/bot_archive.py
|
| diff --git a/appengine/swarming/server/bot_archive.py b/appengine/swarming/server/bot_archive.py
|
| index c652fd7aa26e75a4b183767458ac1fc84221f8ac..d3626eb2fa2c32d8949fa29a3119cbae2df2bcdc 100644
|
| --- a/appengine/swarming/server/bot_archive.py
|
| +++ b/appengine/swarming/server/bot_archive.py
|
| @@ -377,27 +377,23 @@ def resolve_symlink(path):
|
|
|
|
|
| def yield_swarming_bot_files(
|
| - root_dir, host, host_version, additionals, enable_ts_monitoring):
|
| + root_dir, host, host_version, additionals, settings):
|
| """Yields all the files to map as tuple(filename, content).
|
|
|
| config.json is injected with json data about the server.
|
|
|
| This function guarantees that the output is sorted by filename.
|
| """
|
| - grpc_prefix = 'grpc://'
|
| - is_grpc = host.startswith(grpc_prefix)
|
| - if is_grpc:
|
| - host = host[len(grpc_prefix):]
|
| -
|
| items = {i: None for i in FILES}
|
| items.update(additionals)
|
| config = {
|
| - 'enable_ts_monitoring': enable_ts_monitoring,
|
| - 'is_grpc': is_grpc,
|
| + 'enable_ts_monitoring': settings.enable_ts_monitoring,
|
| 'server': host.rstrip('/'),
|
| 'server_version': host_version,
|
| + 'isolate_grpc_proxy': settings.bot_isolate_grpc_proxy,
|
| }
|
| items['config/config.json'] = json.dumps(config)
|
| + logging.debug('Bot config.json: %s', items['config/config.json'])
|
| for item, content in sorted(items.iteritems()):
|
| if content is not None:
|
| yield item, content
|
| @@ -407,7 +403,7 @@ def yield_swarming_bot_files(
|
|
|
|
|
| def get_swarming_bot_zip(
|
| - root_dir, host, host_version, additionals, enable_ts_monitoring):
|
| + root_dir, host, host_version, additionals, settings):
|
| """Returns a zipped file of all the files a bot needs to run.
|
|
|
| Arguments:
|
| @@ -425,7 +421,7 @@ def get_swarming_bot_zip(
|
| h = hashlib.sha256()
|
| with zipfile.ZipFile(zip_memory_file, 'w', zipfile.ZIP_DEFLATED) as zip_file:
|
| for name, content in yield_swarming_bot_files(
|
| - root_dir, host, host_version, additionals, enable_ts_monitoring):
|
| + root_dir, host, host_version, additionals, settings):
|
| zip_file.writestr(name, content)
|
| h.update(str(len(name)))
|
| h.update(name)
|
| @@ -441,7 +437,7 @@ def get_swarming_bot_zip(
|
|
|
|
|
| def get_swarming_bot_version(
|
| - root_dir, host, host_version, additionals, enable_ts_monitoring):
|
| + root_dir, host, host_version, additionals, settings):
|
| """Returns the SHA256 hash of the bot code, representing the version.
|
|
|
| Arguments:
|
| @@ -456,7 +452,7 @@ def get_swarming_bot_version(
|
| try:
|
| # TODO(maruel): Deduplicate from zip_package.genereate_version().
|
| for name, content in yield_swarming_bot_files(
|
| - root_dir, host, host_version, additionals, enable_ts_monitoring):
|
| + root_dir, host, host_version, additionals, settings):
|
| h.update(str(len(name)))
|
| h.update(name)
|
| h.update(str(len(content)))
|
|
|