mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 22:37:36 +08:00
Philipp Moritz
95 lines
3.1 KiB
Python
95 lines
3.1 KiB
Python
import logging
|
|
|
|
import ray
|
|
import ray.cloudpickle as pickle
|
|
from ray.experimental.internal_kv import _internal_kv_get, _internal_kv_put
|
|
from ray.gcs_utils import ActorTableData
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _calculate_key(name):
|
|
"""Generate a Redis key with the given name.
|
|
|
|
Args:
|
|
name: The name of the named actor.
|
|
|
|
Returns:
|
|
The key to use for storing a named actor in Redis.
|
|
"""
|
|
return b"Actor:" + name.encode("ascii")
|
|
|
|
|
|
def _get_actor(name):
|
|
if ray._raylet.gcs_actor_service_enabled():
|
|
worker = ray.worker.global_worker
|
|
handle = worker.core_worker.get_named_actor_handle(name)
|
|
else:
|
|
actor_name = _calculate_key(name)
|
|
pickled_state = _internal_kv_get(actor_name)
|
|
if pickled_state is None:
|
|
raise ValueError(
|
|
"The actor with name={} doesn't exist".format(name))
|
|
handle = pickle.loads(pickled_state)
|
|
# If the actor state is dead, that means that this name is reusable.
|
|
# We don't delete the name entry from key value store when
|
|
# the actor is killed because ray.kill is asynchronous,
|
|
# and it can cause worker leaks.
|
|
actor_info = ray.actors(actor_id=handle._actor_id.hex())
|
|
actor_state = actor_info.get("State", None)
|
|
if actor_state and actor_state == ActorTableData.DEAD:
|
|
raise ValueError("The actor with name={} is dead.".format(name))
|
|
return handle
|
|
|
|
|
|
def get_actor(name: str) -> ray.actor.ActorHandle:
|
|
"""Get a named actor which was previously created.
|
|
|
|
If the actor doesn't exist, an exception will be raised.
|
|
|
|
Args:
|
|
name: The name of the named actor.
|
|
|
|
Returns:
|
|
The ActorHandle object corresponding to the name.
|
|
"""
|
|
logger.warning("ray.util.get_actor has been moved to ray.get_actor and "
|
|
"will be removed in the future.")
|
|
return _get_actor(name)
|
|
|
|
|
|
def _register_actor(name, actor_handle):
|
|
if not isinstance(name, str):
|
|
raise TypeError("The name argument must be a string.")
|
|
if not isinstance(actor_handle, ray.actor.ActorHandle):
|
|
raise TypeError("The actor_handle argument must be an ActorHandle "
|
|
"object.")
|
|
actor_name = _calculate_key(name)
|
|
|
|
# First check if the actor already exists.
|
|
try:
|
|
_get_actor(name)
|
|
exists = True
|
|
except ValueError:
|
|
exists = False
|
|
|
|
if exists:
|
|
raise ValueError("An actor with name={} already exists or there "
|
|
"was timeout in getting this actor handle."
|
|
.format(name))
|
|
|
|
# Add the actor to Redis if it does not already exist.
|
|
_internal_kv_put(actor_name, pickle.dumps(actor_handle), overwrite=True)
|
|
|
|
|
|
def register_actor(name, actor_handle):
|
|
"""Register a named actor under a string key.
|
|
|
|
Args:
|
|
name: The name of the named actor.
|
|
actor_handle: The actor object to be associated with this name
|
|
"""
|
|
logger.warning("ray.util.register_actor is deprecated. To create a "
|
|
"named, detached actor, use Actor.options(name=\"name\").")
|
|
return _register_actor(name, actor_handle)
|