53c405c699
Summary: This diff changes the RPC layer to directly return `TResponse` to the user when issuing a `Call<...>` RPC call. The call throws an exception on failure (instead of the previous return `nullopt`). All servers (network, RPC and distributed) are set to have explicit `Shutdown` methods so that a controlled shutdown can always be performed. The object destructors now have `CHECK`s to enforce that the `AwaitShutdown` methods were called. The distributed memgraph is changed that none of the binaries (master/workers) crash when there is a communication failure. Instead, the whole cluster starts a graceful shutdown when a persistent communication error is detected. Transient errors are allowed during execution. The transaction that errored out will be aborted on the whole cluster. The cluster state is managed using a new Heartbeat RPC call. Reviewers: buda, teon.banek, msantl Reviewed By: teon.banek Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D1604
31 lines
849 B
C++
31 lines
849 B
C++
#include <gtest/gtest.h>
|
|
|
|
#include "communication/rpc/client_pool.hpp"
|
|
#include "communication/rpc/server.hpp"
|
|
#include "database/distributed_counters.hpp"
|
|
|
|
const std::string kLocal = "127.0.0.1";
|
|
|
|
TEST(CountersDistributed, All) {
|
|
communication::rpc::Server master_server({kLocal, 0});
|
|
database::MasterCounters master(&master_server);
|
|
communication::rpc::ClientPool master_client_pool(master_server.endpoint());
|
|
|
|
database::WorkerCounters w1(&master_client_pool);
|
|
database::WorkerCounters w2(&master_client_pool);
|
|
|
|
EXPECT_EQ(w1.Get("a"), 0);
|
|
EXPECT_EQ(w1.Get("a"), 1);
|
|
EXPECT_EQ(w2.Get("a"), 2);
|
|
EXPECT_EQ(w1.Get("a"), 3);
|
|
EXPECT_EQ(master.Get("a"), 4);
|
|
|
|
EXPECT_EQ(master.Get("b"), 0);
|
|
EXPECT_EQ(w2.Get("b"), 1);
|
|
w1.Set("b", 42);
|
|
EXPECT_EQ(w2.Get("b"), 42);
|
|
|
|
master_server.Shutdown();
|
|
master_server.AwaitShutdown();
|
|
}
|