Skip to content

Commit

Permalink
Issue #126: drop proxy user id from batch job
Browse files Browse the repository at this point in the history
PR #128 follow up: no need to keep proxy user id in batch job itself

also be more simple/strict in `argv` handling batch_job.py
  • Loading branch information
soxofaan committed Mar 9, 2022
1 parent 7c31bd0 commit 011f20a
Show file tree
Hide file tree
Showing 4 changed files with 12 additions and 14 deletions.
20 changes: 9 additions & 11 deletions openeogeotrellis/deploy/batch_job.py
Original file line number Diff line number Diff line change
Expand Up @@ -225,23 +225,21 @@ def main(argv: List[str]) -> None:
logger.info("argv: {a!r}".format(a=argv))
logger.info("pid {p}; ppid {pp}; cwd {c}".format(p=os.getpid(), pp=os.getppid(), c=os.getcwd()))

if len(argv) < 6:
print("usage: %s "
"<job specification input file> <job directory> <results output file name> <user log file name> "
"<metadata file name> [api version] [dependencies]" % argv[0],
file=sys.stderr)
exit(1)
if len(argv) != 9:
raise Exception(
f"usage: {argv[0]} "
"<job specification input file> <job directory> <results output file name> <user log file name> "
"<metadata file name> <api version> <dependencies> <user id>"
)

job_specification_file = argv[1]
job_dir = Path(argv[2])
output_file = job_dir / argv[3]
log_file = job_dir / argv[4]
metadata_file = job_dir / argv[5]
api_version = argv[6] if len(argv) >= 7 else None
dependencies = _deserialize_dependencies(argv[7]) if len(argv) >= 8 else {}
# TODO: do we still need proxy user id inside the batch job itself?
proxy_user_id = argv[8] if len(argv) >= 9 else None
user_id = argv[9] if len(argv) >= 10 else None
api_version = argv[6]
dependencies = _deserialize_dependencies(argv[7])
user_id = argv[8]

_create_job_dir(job_dir)

Expand Down
2 changes: 1 addition & 1 deletion openeogeotrellis/deploy/submit_batch_job.sh
Original file line number Diff line number Diff line change
Expand Up @@ -134,4 +134,4 @@ spark-submit \
--conf spark.yarn.tags=openeo \
--jars "${extensions}","${backend_assembly}" \
--name "${jobName}" \
"${main_py_file}" "$(basename "${processGraphFile}")" "${outputDir}" "${outputFileName}" "${userLogFileName}" "${metadataFileName}" "${apiVersion}" "${dependencies}" "${proxyUser}" "${userId}"
"${main_py_file}" "$(basename "${processGraphFile}")" "${outputDir}" "${outputFileName}" "${userLogFileName}" "${metadataFileName}" "${apiVersion}" "${dependencies}" "${userId}"
2 changes: 1 addition & 1 deletion openeogeotrellis/deploy/submit_batch_job_spark24.sh
Original file line number Diff line number Diff line change
Expand Up @@ -144,4 +144,4 @@ spark-submit \
--conf spark.yarn.tags=openeo \
--jars "${extensions}","${backend_assembly}" \
--name "${jobName}" \
"${main_py_file}" "$(basename "${processGraphFile}")" "${outputDir}" "${outputFileName}" "${userLogFileName}" "${metadataFileName}" "${apiVersion}" "${dependencies}" "${proxyUser}" "${userId}"
"${main_py_file}" "$(basename "${processGraphFile}")" "${outputDir}" "${outputFileName}" "${userLogFileName}" "${metadataFileName}" "${apiVersion}" "${dependencies}" "${userId}"
2 changes: 1 addition & 1 deletion openeogeotrellis/deploy/submit_batch_job_spark3.sh
Original file line number Diff line number Diff line change
Expand Up @@ -165,4 +165,4 @@ spark-submit \
--conf spark.yarn.tags=openeo \
--jars "${extensions}","${backend_assembly}" \
--name "${jobName}" \
"${main_py_file}" "$(basename "${processGraphFile}")" "${outputDir}" "${outputFileName}" "${userLogFileName}" "${metadataFileName}" "${apiVersion}" "${dependencies}" "${proxyUser}" "${userId}"
"${main_py_file}" "$(basename "${processGraphFile}")" "${outputDir}" "${outputFileName}" "${userLogFileName}" "${metadataFileName}" "${apiVersion}" "${dependencies}" "${userId}"

0 comments on commit 011f20a

Please sign in to comment.