def try_create_gcs_client( address: Optional[str], redis_password: Optional[str]) -> Optional[GcsClient]: """ Try to create a gcs client based on the the command line args or by autodetecting a running Ray cluster. """ address = canonicalize_bootstrap_address(address) return GcsClient(address=address)
def try_create_gcs_client( address: Optional[str], redis_password: Optional[str] ) -> Optional[GcsClient]: """ Try to create a gcs client based on the the command line args or by autodetecting a running Ray cluster. """ address = canonicalize_bootstrap_address(address) if use_gcs_for_bootstrap(): return GcsClient(address=address) else: if redis_password is None: redis_password = ray.ray_constants.REDIS_DEFAULT_PASSWORD return GcsClient.connect_to_gcs_by_redis_address(address, redis_password)
def memory_summary(address=None, redis_password=ray_constants.REDIS_DEFAULT_PASSWORD, group_by="NODE_ADDRESS", sort_by="OBJECT_SIZE", units="B", line_wrap=True, stats_only=False, num_entries=None): from ray.dashboard.memory_utils import memory_summary address = services.canonicalize_bootstrap_address(address) state = GlobalState() state._initialize_global_state( GcsClientOptions.from_redis_address(address, redis_password)) if stats_only: return get_store_stats(state) return (memory_summary(state, group_by, sort_by, line_wrap, units, num_entries) + get_store_stats(state))
def list_state_cli_group(ctx): address = services.canonicalize_bootstrap_address(None) gcs_client = GcsClient(address=address, nums_reconnect_retry=0) ray.experimental.internal_kv._initialize_internal_kv(gcs_client) api_server_url = ray._private.utils.internal_kv_get_with_retry( gcs_client, ray_constants.DASHBOARD_ADDRESS, namespace=ray_constants.KV_NAMESPACE_DASHBOARD, num_retries=20, ) if api_server_url is None: raise ValueError(( "Couldn't obtain the API server address from GCS. It is likely that " "the GCS server is down. Check gcs_server.[out | err] to see if it is " "still alive.")) assert use_gcs_for_bootstrap() ctx.ensure_object(dict) ctx.obj["api_server_url"] = f"http://{api_server_url.decode()}"