wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-29 11:26:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

implementing reading and writing tasks (#11)

Browse Source
This commit is contained in:
Philipp Moritz
2016-09-18 18:06:42 -07:00
committed by Robert Nishihara
parent c238ae4aa0
commit 37f035dbd0
6 changed files with 86 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
task.c
+24 -2
View File
@@ -47,12 +47,16 @@ struct task_spec_impl {
task_arg args_and_returns[0];
};
/* The size of a task specification is given by the following expression. */
#define TASK_SPEC_SIZE(NUM_ARGS, NUM_RETURNS, ARGS_VALUE_SIZE) \
(sizeof(task_spec) + ((NUM_ARGS) + (NUM_RETURNS)) * sizeof(task_arg) + \
(ARGS_VALUE_SIZE))
task_spec *alloc_task_spec(function_id func_id,
int64_t num_args,
int64_t num_returns,
int64_t args_value_size) {
int64_t size = sizeof(task_spec) +
(num_args + num_returns) * sizeof(task_arg) + args_value_size;
int64_t size = TASK_SPEC_SIZE(num_args, num_returns, args_value_size);
task_spec *task = malloc(size);
memset(task, 0, size);
task->func_id = func_id;
@@ -63,6 +67,11 @@ task_spec *alloc_task_spec(function_id func_id,
return task;
}
int64_t task_size(task_spec *spec) {
return TASK_SPEC_SIZE(spec->num_args, spec->num_returns,
spec->args_value_size);
}
int64_t task_num_args(task_spec *spec) {
return spec->num_args;
}
@@ -131,3 +140,16 @@ void free_task_spec(task_spec *spec) {
CHECK(spec->arg_index == spec->num_args); /* Task was fully constructed */
free(spec);
}
void write_task(int fd, task_spec *spec) {
write_bytes(fd, (uint8_t *) spec, task_size(spec));
}
task_spec *read_task(int fd) {
uint8_t *bytes;
int64_t length;
read_bytes(fd, &bytes, &length);
task_spec *spec = (task_spec *) bytes;
CHECK(task_size(spec) == length);
return spec;
}