fix simulator missing client heartbeat call. (#1196)

Author: yhwen

nvflare/private/fed/app/deployer/simulator_deployer.py

@@ -22,7 +22,7 @@
 from nvflare.private.fed.client.admin_msg_sender import AdminMessageSender
 from nvflare.private.fed.client.client_req_processors import ClientRequestProcessors
 from nvflare.private.fed.client.fed_client import FederatedClient
-from nvflare.private.fed.simulator.simulator_client_engine import SimulatorClientEngine
+from nvflare.private.fed.simulator.simulator_client_engine import SimulatorClientEngine, SimulatorParentClientEngine
 from nvflare.private.fed.simulator.simulator_server import SimulatorServer
 
 from .base_client_deployer import BaseClientDeployer
@@ -69,6 +69,8 @@ def create_fl_client(self, client_name, args):
         deployer.build(build_ctx)
         federated_client = deployer.create_fed_client(args)
 
+        client_engine = SimulatorParentClientEngine()
+        federated_client.set_client_engine(client_engine)
         federated_client.register()
         federated_client.start_heartbeat()
         federated_client.run_manager = None

nvflare/private/fed/simulator/simulator_client_engine.py

@@ -14,6 +14,7 @@
 
 from nvflare.apis.shareable import ReservedHeaderKey, Shareable
 from nvflare.private.fed.client.client_engine import ClientEngine
+from nvflare.private.fed.simulator.simulator_const import SimulatorConstants
 
 
 class SimulatorClientEngine(ClientEngine):
@@ -26,3 +27,11 @@ def send_aux_command(self, shareable: Shareable, job_id):
             with run_manager.new_context() as fl_ctx:
                 topic = shareable.get_header(ReservedHeaderKey.TOPIC)
                 return run_manager.dispatch(topic=topic, request=shareable, fl_ctx=fl_ctx)
+
+
+class SimulatorParentClientEngine(ClientEngine):
+    def __init__(self):
+        pass
+
+    def get_all_job_ids(self):
+        return [SimulatorConstants.JOB_NAME]