wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-29 05:17:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Prepare socket file when start ray (#3925)

Browse Source
This commit is contained in:
Yuhong Guo
2019-02-02 12:53:36 +08:00
committed by Hao Chen
parent 0f81bc9a33
commit 54cbb4396f
2 changed files with 29 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/node.py
+21 -1
View File
@@ -16,7 +16,8 @@ from ray.tempfile_services import (
get_logs_dir_path, get_object_store_socket_name, get_raylet_socket_name,
new_log_monitor_log_file, new_monitor_log_file,
new_raylet_monitor_log_file, new_plasma_store_log_file,
new_raylet_log_file, new_webui_log_file, set_temp_root)
new_raylet_log_file, new_webui_log_file, set_temp_root,
try_to_create_directory)
# Logger for this module. It should be configured at the entry point
# into the program using Ray. Ray configures it by default automatically
@@ -102,6 +103,23 @@ class Node(object):
"""Get the node's raylet socket name."""
return self._raylet_socket_name
def prepare_socket_file(self, socket_path):
"""Prepare the socket file for raylet and plasma.
This method helps to prepare a socket file.
1. Make the directory if the directory does not exist.
2. If the socket file exists, raise exception.
Args:
socket_path (string): the socket file to prepare.
"""
if not os.path.exists(socket_path):
path = os.path.dirname(socket_path)
if not os.path.isdir(path):
try_to_create_directory(path)
else:
raise Exception("Socket file {} exists!".format(socket_path))
def start_redis(self):
"""Start the Redis servers."""
assert self._redis_address is None
@@ -155,6 +173,7 @@ class Node(object):
self._plasma_store_socket_name = (
self._ray_params.plasma_store_socket_name
or get_object_store_socket_name())
self.prepare_socket_file(self._plasma_store_socket_name)
stdout_file, stderr_file = (new_plasma_store_log_file(
self._ray_params.redirect_output))
process_info = ray.services.start_plasma_store(
@@ -186,6 +205,7 @@ class Node(object):
# If the user specified a socket name, use it.
self._raylet_socket_name = (self._ray_params.raylet_socket_name
or get_raylet_socket_name())
self.prepare_socket_file(self._raylet_socket_name)
stdout_file, stderr_file = new_raylet_log_file(
redirect_output=self._ray_params.redirect_worker_output)
process_info = ray.services.start_raylet(