mirror of
https://github.com/vale981/ray
synced 2025-03-14 07:06:38 -04:00
12c9618c0c
* Failing test case * Local scheduler exits cleanly after plasma store dies * Tolerate one plasma store failure * Tolerate plasma store failures on all nodes except head node * Plasma manager heartbeats * Component failure tests * Don't run the helper for Python testing * Fix C test * Fix hanging plasma transfer test * Fix python3 * Consolidate ClientConnection code * Fix valgrind test * fix c test * We can restart worker nodes! * Fix flatbuffers bug * Address comments * Only register actual workers with the local scheduler * Fix bug * Fix segfaults * Add test case that tests for driver liveness, fix local scheduler bug * Clean up after tests * Allocate retry info on the stack * Send SIGKILL before waiting * Relax unit test conditions * Driver liveness test case and documentation
40 lines
1.6 KiB
C++
40 lines
1.6 KiB
C++
#include "db_client_table.h"
|
|
#include "redis.h"
|
|
|
|
void db_client_table_remove(DBHandle *db_handle,
|
|
DBClientID db_client_id,
|
|
RetryInfo *retry,
|
|
db_client_table_done_callback done_callback,
|
|
void *user_context) {
|
|
init_table_callback(db_handle, db_client_id, __func__, NULL, retry,
|
|
(table_done_callback) done_callback,
|
|
redis_db_client_table_remove, user_context);
|
|
}
|
|
|
|
void db_client_table_subscribe(
|
|
DBHandle *db_handle,
|
|
db_client_table_subscribe_callback subscribe_callback,
|
|
void *subscribe_context,
|
|
RetryInfo *retry,
|
|
db_client_table_done_callback done_callback,
|
|
void *user_context) {
|
|
DBClientTableSubscribeData *sub_data =
|
|
(DBClientTableSubscribeData *) malloc(sizeof(DBClientTableSubscribeData));
|
|
sub_data->subscribe_callback = subscribe_callback;
|
|
sub_data->subscribe_context = subscribe_context;
|
|
|
|
init_table_callback(db_handle, NIL_ID, __func__, sub_data, retry,
|
|
(table_done_callback) done_callback,
|
|
redis_db_client_table_subscribe, user_context);
|
|
}
|
|
|
|
void plasma_manager_send_heartbeat(DBHandle *db_handle) {
|
|
RetryInfo heartbeat_retry;
|
|
heartbeat_retry.num_retries = 0;
|
|
heartbeat_retry.timeout = HEARTBEAT_TIMEOUT_MILLISECONDS;
|
|
heartbeat_retry.fail_callback = NULL;
|
|
|
|
init_table_callback(db_handle, NIL_ID, __func__, NULL,
|
|
(RetryInfo *) &heartbeat_retry, NULL,
|
|
redis_plasma_manager_send_heartbeat, NULL);
|
|
}
|