| 61 | |
| 62 | |
| 63 | class FedMLServerRunner: |
| 64 | FEDML_CLOUD_SERVER_PREFIX = "fedml-server-run-" |
| 65 | |
| 66 | def __init__(self, args, run_id=0, request_json=None, agent_config=None, edge_id=0): |
| 67 | self.origin_fedml_config_object = None |
| 68 | self.package_type = SchedulerConstants.JOB_PACKAGE_TYPE_DEFAULT |
| 69 | self.local_api_process = None |
| 70 | self.run_process_event = None |
| 71 | self.run_process_event_map = dict() |
| 72 | self.run_process_completed_event = None |
| 73 | self.run_process_completed_event_map = dict() |
| 74 | self.edge_status_queue = None |
| 75 | self.edge_status_queue_map = dict() |
| 76 | self.run_process = None |
| 77 | self.run_process_map = dict() |
| 78 | self.start_request_json = None |
| 79 | self.server_docker_image = None |
| 80 | self.cloud_server_name = None |
| 81 | self.run_as_cloud_agent = False |
| 82 | self.run_as_cloud_server = False |
| 83 | self.run_as_edge_server_and_agent = False |
| 84 | self.run_as_cloud_server_and_agent = False |
| 85 | self.fedml_packages_base_dir = None |
| 86 | self.fedml_packages_unzip_dir = None |
| 87 | self.mqtt_mgr = None |
| 88 | self.running_request_json = dict() |
| 89 | self.run_id = run_id |
| 90 | self.client_mqtt_mgr = None |
| 91 | self.client_mqtt_is_connected = False |
| 92 | self.client_mqtt_lock = None |
| 93 | self.unique_device_id = None |
| 94 | self.edge_id = edge_id |
| 95 | self.server_agent_id = 0 |
| 96 | if request_json is not None: |
| 97 | self.server_agent_id = request_json.get("server_id", 0) |
| 98 | self.process = None |
| 99 | self.args = args |
| 100 | self.request_json = copy.deepcopy(request_json) |
| 101 | self.version = args.version |
| 102 | self.device_id = args.device_id |
| 103 | self.cur_dir = os.path.split(os.path.realpath(__file__))[0] |
| 104 | if args.current_running_dir is not None: |
| 105 | self.cur_dir = args.current_running_dir |
| 106 | |
| 107 | image_version = self.version |
| 108 | if image_version == "local": |
| 109 | image_version = "dev" |
| 110 | self.server_docker_base_image = "/fedml-device-image:" + image_version |
| 111 | |
| 112 | self.agent_config = agent_config |
| 113 | self.fedml_data_base_package_dir = os.path.join("/", "fedml", "data") |
| 114 | self.fedml_data_local_package_dir = os.path.join("/", "fedml", "fedml-package", "fedml", "data") |
| 115 | self.fedml_data_dir = self.fedml_data_base_package_dir |
| 116 | self.fedml_config_dir = os.path.join("/", "fedml", "conf") |
| 117 | |
| 118 | self.FEDML_DYNAMIC_CONSTRAIN_VARIABLES = { |
| 119 | "${FEDSYS.RUN_ID}": "", |
| 120 | "${FEDSYS.PRIVATE_LOCAL_DATA}": "", |
no outgoing calls
no test coverage detected