def build_task_descriptor(cls,
task_descriptor,
parent_task_record,
args,
n_outputs,
is_tail_spawn=False,
handler_name=None):
# This is needed to work around the fact that stdinout has its own implementation of build_task_descriptor, so
# we can't rely using cls.handler_name to find the actual executor.
if handler_name is None:
handler_name = cls.handler_name
if is_tail_spawn and len(
task_descriptor["expected_outputs"]) != n_outputs:
raise BlameUserException(
"SimpleExecutor being built with delegated outputs %s but n_outputs=%d"
% (task_descriptor["expected_outputs"], n_outputs))
# Throw early if the args are bad
cls.check_args_valid(args, n_outputs)
# Discover required ref IDs for this executor
reqd_refs = cls.get_required_refs(args)
task_descriptor["dependencies"].extend(reqd_refs)
sha = hashlib.sha1()
hash_update_with_structure(sha, [args, n_outputs])
name_prefix = "%s:%s:" % (handler_name, sha.hexdigest())
# Name our outputs
if not is_tail_spawn:
task_descriptor["expected_outputs"] = [
"%s%d" % (name_prefix, i) for i in range(n_outputs)
]
# Add the args dict
args_name = "%ssimple_exec_args" % name_prefix
args_ref = ref_from_object(args, "pickle", args_name)
parent_task_record.publish_ref(args_ref)
task_descriptor["dependencies"].append(args_ref)
task_descriptor["task_private"]["simple_exec_args"] = args_ref
BaseExecutor.build_task_descriptor(task_descriptor, parent_task_record)
if is_tail_spawn:
return None
else:
return [
SW2_FutureReference(x)
for x in task_descriptor["expected_outputs"]
]
def build_task_descriptor(cls, task_descriptor, parent_task_record, jar_lib=None, args=None, class_name=None, object_ref=None, n_outputs=1, is_tail_spawn=False, **kwargs):
# More good stuff goes here.
if jar_lib is None and kwargs.get("process_record_id", None) is None:
raise BlameUserException("All Java2 invocations must either specify jar libs or an existing process ID")
if class_name is None and object_ref is None and kwargs.get("process_record_id", None) is None:
raise BlameUserException("All Java2 invocations must specify either a class_name or an object_ref, or else give a process ID")
if jar_lib is not None:
task_descriptor["task_private"]["jar_lib"] = jar_lib
for jar_ref in jar_lib:
task_descriptor["dependencies"].append(jar_ref)
if not is_tail_spawn:
sha = hashlib.sha1()
hash_update_with_structure(sha, [args, n_outputs])
hash_update_with_structure(sha, class_name)
hash_update_with_structure(sha, object_ref)
hash_update_with_structure(sha, jar_lib)
name_prefix = "java2:%s:" % (sha.hexdigest())
task_descriptor["expected_outputs"] = ["%s%d" % (name_prefix, i) for i in range(n_outputs)]
if class_name is not None:
task_descriptor["task_private"]["class_name"] = class_name
if object_ref is not None:
task_descriptor["task_private"]["object_ref"] = object_ref
task_descriptor["dependencies"].append(object_ref)
if args is not None:
task_descriptor["task_private"]["args"] = args
add_package_dep(parent_task_record.package_ref, task_descriptor)
return ProcExecutor.build_task_descriptor(task_descriptor, parent_task_record, n_extra_outputs=0, is_tail_spawn=is_tail_spawn, accept_ref_list_for_single=True, **kwargs)
def build_task_descriptor(cls, task_descriptor, parent_task_record, jar_lib=None, args=None, class_name=None, object_ref=None, n_outputs=1, is_tail_spawn=False, **kwargs):
# More good stuff goes here.
if jar_lib is None and kwargs.get("process_record_id", None) is None:
raise BlameUserException("All Java2 invocations must either specify jar libs or an existing process ID")
if class_name is None and object_ref is None and kwargs.get("process_record_id", None) is None:
raise BlameUserException("All Java2 invocations must specify either a class_name or an object_ref, or else give a process ID")
if jar_lib is not None:
task_descriptor["task_private"]["jar_lib"] = jar_lib
for jar_ref in jar_lib:
task_descriptor["dependencies"].append(jar_ref)
if not is_tail_spawn:
sha = hashlib.sha1()
hash_update_with_structure(sha, [args, n_outputs])
hash_update_with_structure(sha, class_name)
hash_update_with_structure(sha, object_ref)
hash_update_with_structure(sha, jar_lib)
name_prefix = "java2:%s:" % (sha.hexdigest())
task_descriptor["expected_outputs"] = ["%s%d" % (name_prefix, i) for i in range(n_outputs)]
if class_name is not None:
task_descriptor["task_private"]["class_name"] = class_name
if object_ref is not None:
task_descriptor["task_private"]["object_ref"] = object_ref
task_descriptor["dependencies"].append(object_ref)
if args is not None:
task_descriptor["task_private"]["args"] = args
add_package_dep(parent_task_record.package_ref, task_descriptor)
return ProcExecutor.build_task_descriptor(task_descriptor, parent_task_record, n_extra_outputs=0, is_tail_spawn=is_tail_spawn, accept_ref_list_for_single=True, **kwargs)
def build_task_descriptor(cls, task_descriptor, parent_task_record, args, n_outputs, is_tail_spawn=False, handler_name=None):
# This is needed to work around the fact that stdinout has its own implementation of build_task_descriptor, so
# we can't rely using cls.handler_name to find the actual executor.
if handler_name is None:
handler_name = cls.handler_name
if is_tail_spawn and len(task_descriptor["expected_outputs"]) != n_outputs:
raise BlameUserException("SimpleExecutor being built with delegated outputs %s but n_outputs=%d" % (task_descriptor["expected_outputs"], n_outputs))
# Throw early if the args are bad
cls.check_args_valid(args, n_outputs)
# Discover required ref IDs for this executor
reqd_refs = cls.get_required_refs(args)
task_descriptor["dependencies"].extend(reqd_refs)
sha = hashlib.sha1()
hash_update_with_structure(sha, [args, n_outputs])
name_prefix = "%s:%s:" % (handler_name, sha.hexdigest())
# Name our outputs
if not is_tail_spawn:
task_descriptor["expected_outputs"] = ["%s%d" % (name_prefix, i) for i in range(n_outputs)]
# Add the args dict
args_name = "%ssimple_exec_args" % name_prefix
args_ref = ref_from_object(args, "pickle", args_name)
parent_task_record.publish_ref(args_ref)
task_descriptor["dependencies"].append(args_ref)
task_descriptor["task_private"]["simple_exec_args"] = args_ref
BaseExecutor.build_task_descriptor(task_descriptor, parent_task_record)
if is_tail_spawn:
return None
else:
return [SW2_FutureReference(x) for x in task_descriptor["expected_outputs"]]
def build_task_descriptor(cls,
task_descriptor,
parent_task_record,
binary,
fn_ref=None,
args=None,
n_outputs=1,
is_tail_spawn=False,
**kwargs):
if binary is None:
raise BlameUserException(
"All Haskell invocations must specify a binary")
if not isabs(binary):
binary = join(expanduser("~/.cabal/bin"), binary)
task_descriptor["task_private"]["binary"] = binary
if fn_ref is not None:
task_descriptor["task_private"]["fn_ref"] = fn_ref
task_descriptor["dependencies"].append(fn_ref)
if not is_tail_spawn:
sha = hashlib.sha1()
hash_update_with_structure(sha, [args, n_outputs])
hash_update_with_structure(sha, binary)
hash_update_with_structure(sha, fn_ref)
name_prefix = "hsk:%s:" % (sha.hexdigest())
task_descriptor["expected_outputs"] = [
"%s%d" % (name_prefix, i) for i in range(n_outputs)
]
if args is not None:
task_descriptor["task_private"]["args"] = args
return ProcExecutor.build_task_descriptor(
task_descriptor,
parent_task_record,
n_extra_outputs=0,
is_tail_spawn=is_tail_spawn,
accept_ref_list_for_single=True,
**kwargs)
def build_task_descriptor(cls, task_descriptor, parent_task_record, binary, fn_ref=None, args=None, n_outputs=1, is_tail_spawn=False, **kwargs):
if binary is None:
raise BlameUserException("All OCaml invocations must specify a binary")
task_descriptor["task_private"]["binary"] = binary
if fn_ref is not None:
task_descriptor["task_private"]["fn_ref"] = fn_ref
task_descriptor["dependencies"].append(fn_ref)
if not is_tail_spawn:
sha = hashlib.sha1()
hash_update_with_structure(sha, [args, n_outputs])
hash_update_with_structure(sha, binary)
hash_update_with_structure(sha, fn_ref)
name_prefix = "ocaml:%s:" % (sha.hexdigest())
task_descriptor["expected_outputs"] = ["%s%d" % (name_prefix, i) for i in range(n_outputs)]
if args is not None:
task_descriptor["task_private"]["args"] = args
return ProcExecutor.build_task_descriptor(task_descriptor, parent_task_record, n_extra_outputs=0, is_tail_spawn=is_tail_spawn, is_fixed=False, accept_ref_list_for_single=True, **kwargs)
