From 8057db1737c4d075e8e41974b9ac1e4bed7bf78b Mon Sep 17 00:00:00 2001 From: Charles Beauville Date: Thu, 28 Mar 2024 14:47:21 -0400 Subject: [PATCH] Fix validation --- src/cc/flwr/src/communicator.cc | 2 +- src/cc/flwr/src/message_handler.cc | 14 ++++++++++---- 2 files changed, 11 insertions(+), 5 deletions(-) diff --git a/src/cc/flwr/src/communicator.cc b/src/cc/flwr/src/communicator.cc index 2ad05127956..98569b752fb 100644 --- a/src/cc/flwr/src/communicator.cc +++ b/src/cc/flwr/src/communicator.cc @@ -26,7 +26,7 @@ bool validate_task_ins(const flwr::proto::TaskIns &task_ins, bool validate_task_res(const flwr::proto::TaskRes &task_res) { // Retrieve initialized fields in TaskRes - return (!task_res.task_id().empty()); + return true; } flwr::proto::TaskRes diff --git a/src/cc/flwr/src/message_handler.cc b/src/cc/flwr/src/message_handler.cc index 22aa2567c1d..f7b9f791217 100644 --- a/src/cc/flwr/src/message_handler.cc +++ b/src/cc/flwr/src/message_handler.cc @@ -73,6 +73,10 @@ handle_task(flwr_local::Client *client, const flwr::proto::TaskIns &task_ins) { flwr::proto::TaskRes task_res; + task_res.set_task_id(""); + task_res.set_group_id(task_ins.group_id()); + task_res.set_run_id(task_ins.run_id()); + std::unique_ptr task = std::make_unique(); @@ -82,10 +86,12 @@ handle_task(flwr_local::Client *client, const flwr::proto::TaskIns &task_ins) { task->set_allocated_recordset(proto_recordset_ptr.release()); task->set_task_type(received_task.task_type()); - task->set_ttl(3600); - task->set_created_at(std::chrono::duration_cast( - std::chrono::system_clock::now().time_since_epoch()) - .count()); + task->set_ttl(""); + task->set_created_at(""); + task->set_allocated_consumer( + std::make_unique(received_task.producer()).release()); + task->set_allocated_producer( + std::make_unique(received_task.consumer()).release()); task_res.set_allocated_task(task.release());