def _prepare_socket_file(self, socket_path, default_prefix): """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 socket_path is not None: if os.path.exists(socket_path): raise Exception("Socket file {} exists!".format(socket_path)) socket_dir = os.path.dirname(socket_path) try_to_create_directory(socket_dir) return socket_path return self._make_inc_temp( prefix=default_prefix, directory_name=self._sockets_dir)
def _prepare_socket_file(self, socket_path, default_prefix): """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 socket_path is not None: if os.path.exists(socket_path): raise Exception("Socket file {} exists!".format(socket_path)) socket_dir = os.path.dirname(socket_path) try_to_create_directory(socket_dir) return socket_path return self._make_inc_temp(prefix=default_prefix, directory_name=self._sockets_dir)
def _init_temp(self): # Create an dictionary to store temp file index. self._incremental_dict = collections.defaultdict(lambda: 0) self._temp_dir = self._ray_params.temp_dir if self._temp_dir is None: date_str = datetime.datetime.today().strftime("%Y-%m-%d_%H-%M-%S") self._temp_dir = self._make_inc_temp( prefix="session_{date_str}_{pid}".format(pid=os.getpid(), date_str=date_str), directory_name="/tmp/ray") try_to_create_directory(self._temp_dir) # Create a directory to be used for socket files. self._sockets_dir = os.path.join(self._temp_dir, "sockets") try_to_create_directory(self._sockets_dir) # Create a directory to be used for process log files. self._logs_dir = os.path.join(self._temp_dir, "logs") try_to_create_directory(self._logs_dir)
def _init_temp(self): # Create an dictionary to store temp file index. self._incremental_dict = collections.defaultdict(lambda: 0) self._temp_dir = self._ray_params.temp_dir if self._temp_dir is None: date_str = datetime.datetime.today().strftime("%Y-%m-%d_%H-%M-%S") self._temp_dir = self._make_inc_temp( prefix="session_{date_str}_{pid}".format( pid=os.getpid(), date_str=date_str), directory_name="/tmp/ray") try_to_create_directory(self._temp_dir) # Create a directory to be used for socket files. self._sockets_dir = os.path.join(self._temp_dir, "sockets") try_to_create_directory(self._sockets_dir) # Create a directory to be used for process log files. self._logs_dir = os.path.join(self._temp_dir, "logs") try_to_create_directory(self._logs_dir)
def _init_temp(self, redis_client): # Create a dictionary to store temp file index. self._incremental_dict = collections.defaultdict(lambda: 0) if self.head: self._temp_dir = self._ray_params.temp_dir else: temp_dir = _get_with_retry(redis_client, "temp_dir") self._temp_dir = ray.utils.decode(temp_dir) try_to_create_directory(self._temp_dir) if self.head: self._session_dir = os.path.join(self._temp_dir, self.session_name) else: session_dir = _get_with_retry(redis_client, "session_dir") self._session_dir = ray.utils.decode(session_dir) session_symlink = os.path.join(self._temp_dir, SESSION_LATEST) # Send a warning message if the session exists. try_to_create_directory(self._session_dir) try_to_symlink(session_symlink, self._session_dir) # Create a directory to be used for socket files. self._sockets_dir = os.path.join(self._session_dir, "sockets") try_to_create_directory(self._sockets_dir) # Create a directory to be used for process log files. self._logs_dir = os.path.join(self._session_dir, "logs") try_to_create_directory(self._logs_dir) old_logs_dir = os.path.join(self._logs_dir, "old") try_to_create_directory(old_logs_dir) # Create a directory to be used for runtime environment. self._runtime_env_dir = os.path.join(self._session_dir, "runtime_resources") try_to_create_directory(self._runtime_env_dir) import ray._private.runtime_env as runtime_env runtime_env.PKG_DIR = self._runtime_env_dir