wehub-resource-sync
44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# Standard
|
|
from typing import Dict, List
|
|
|
|
|
|
def get_all_server_infos(config, worker_count) -> List[Dict[str, str]]:
|
|
"""
|
|
Generate a list of server information (scheduler and workers) based on the config.
|
|
|
|
Args:
|
|
config: The configuration object containing server details.
|
|
worker_count: The number of worker servers.
|
|
|
|
Returns:
|
|
List[Dict[str, str]]: A JSON list with server information.
|
|
"""
|
|
servers = []
|
|
include_index_list = getattr(config, "internal_api_server_include_index_list", None)
|
|
socket_path_prefix = getattr(config, "internal_api_server_socket_path_prefix", None)
|
|
|
|
# Add scheduler info (index 0)
|
|
if include_index_list is None or 0 in include_index_list:
|
|
port = config.internal_api_server_port_start
|
|
server_info = {
|
|
"name": f"{config.lmcache_instance_id}_scheduler",
|
|
"host": config.internal_api_server_host,
|
|
"port": f"{socket_path_prefix}_{port}" if socket_path_prefix else port,
|
|
}
|
|
servers.append(server_info)
|
|
|
|
# Add workers info (index 1 to worker_count)
|
|
for worker_id in range(worker_count):
|
|
port_offset = 1 + worker_id
|
|
if include_index_list is None or port_offset in include_index_list:
|
|
port = config.internal_api_server_port_start + port_offset
|
|
server_info = {
|
|
"name": f"{config.lmcache_instance_id}_worker{worker_id}",
|
|
"host": config.internal_api_server_host,
|
|
"port": f"{socket_path_prefix}_{port}" if socket_path_prefix else port,
|
|
}
|
|
servers.append(server_info)
|
|
|
|
return servers
|