mirror of
https://github.com/wassname/ray.git
synced 2026-07-09 21:35:12 +08:00
Stephanie Wang
bef1d872b4
* Add raylet monitor script to timeout Raylet heartbeats * Unit test for removing a different client from the client table * Set node manager heartbeat according to global config * Doc and fixes * Add regression test for client table disconnect, refactor client table * Convert 'Terminate' methods to destructors * Destroy the Raylet on a SIGTERM * Clean up workers on a SIGTERM
88 lines
2.7 KiB
C++
88 lines
2.7 KiB
C++
#include <future>
|
|
#include <iostream>
|
|
|
|
#include <boost/asio.hpp>
|
|
#include <boost/bind.hpp>
|
|
#include <boost/function.hpp>
|
|
|
|
#include "common/common.h"
|
|
#include "common/common_protocol.h"
|
|
|
|
#include "ray/object_manager/object_store_notification_manager.h"
|
|
|
|
namespace ray {
|
|
|
|
ObjectStoreNotificationManager::ObjectStoreNotificationManager(
|
|
boost::asio::io_service &io_service, const std::string &store_socket_name)
|
|
: store_client_(), socket_(io_service) {
|
|
ARROW_CHECK_OK(
|
|
store_client_.Connect(store_socket_name.c_str(), "", PLASMA_DEFAULT_RELEASE_DELAY));
|
|
|
|
ARROW_CHECK_OK(store_client_.Subscribe(&c_socket_));
|
|
boost::system::error_code ec;
|
|
socket_.assign(boost::asio::local::stream_protocol(), c_socket_, ec);
|
|
assert(!ec.value());
|
|
NotificationWait();
|
|
}
|
|
|
|
ObjectStoreNotificationManager::~ObjectStoreNotificationManager() {
|
|
ARROW_CHECK_OK(store_client_.Disconnect());
|
|
}
|
|
|
|
void ObjectStoreNotificationManager::NotificationWait() {
|
|
boost::asio::async_read(socket_, boost::asio::buffer(&length_, sizeof(length_)),
|
|
boost::bind(&ObjectStoreNotificationManager::ProcessStoreLength,
|
|
this, boost::asio::placeholders::error));
|
|
}
|
|
|
|
void ObjectStoreNotificationManager::ProcessStoreLength(
|
|
const boost::system::error_code &error) {
|
|
notification_.resize(length_);
|
|
boost::asio::async_read(
|
|
socket_, boost::asio::buffer(notification_),
|
|
boost::bind(&ObjectStoreNotificationManager::ProcessStoreNotification, this,
|
|
boost::asio::placeholders::error));
|
|
}
|
|
|
|
void ObjectStoreNotificationManager::ProcessStoreNotification(
|
|
const boost::system::error_code &error) {
|
|
if (error) {
|
|
RAY_LOG(FATAL) << error.message();
|
|
}
|
|
|
|
const auto &object_info = flatbuffers::GetRoot<ObjectInfo>(notification_.data());
|
|
const auto &object_id = from_flatbuf(*object_info->object_id());
|
|
if (object_info->is_deletion()) {
|
|
ProcessStoreRemove(object_id);
|
|
} else {
|
|
ObjectInfoT result;
|
|
object_info->UnPackTo(&result);
|
|
ProcessStoreAdd(result);
|
|
}
|
|
NotificationWait();
|
|
}
|
|
|
|
void ObjectStoreNotificationManager::ProcessStoreAdd(const ObjectInfoT &object_info) {
|
|
for (auto handler : add_handlers_) {
|
|
handler(object_info);
|
|
}
|
|
}
|
|
|
|
void ObjectStoreNotificationManager::ProcessStoreRemove(const ObjectID &object_id) {
|
|
for (auto handler : rem_handlers_) {
|
|
handler(object_id);
|
|
}
|
|
}
|
|
|
|
void ObjectStoreNotificationManager::SubscribeObjAdded(
|
|
std::function<void(const ObjectInfoT &)> callback) {
|
|
add_handlers_.push_back(callback);
|
|
}
|
|
|
|
void ObjectStoreNotificationManager::SubscribeObjDeleted(
|
|
std::function<void(const ObjectID &)> callback) {
|
|
rem_handlers_.push_back(callback);
|
|
}
|
|
|
|
} // namespace ray
|