8b9e1fa08b
Single (instance level) connection to a replica (messages from all databases get multiplexed through it) ReplicationClient split in two: ReplicationClient and ReplicationStorageClient New ReplicationClient, moved under replication, handles the raw connection, owned by MainRoleData ReplicationStorageClient handles the storage <-> replica state machine and holds to a stream Removed epoch and storage from *Clients rpc::Stream proactively aborts on error and sets itself to a defunct state Removed HandleRpcFailure, instead we simply log the error and let the FrequentCheck handle re-connection replica_state is now a synced variable ReplicaStorageClient state machine bugfixes Single FrequentCheck that goes through DBMS Moved ReplicationState under DbmsHandler Moved some replication startup logic under the DbmsHandler's constructor Removed InMemoryReplicationClient CreateReplicationClient has been removed from Storage Simplified GetRecoverySteps and made safer --------- Co-authored-by: Gareth Lloyd <gareth.lloyd@memgraph.io>
82 lines
3.4 KiB
C++
82 lines
3.4 KiB
C++
// Copyright 2023 Memgraph Ltd.
|
|
//
|
|
// Use of this software is governed by the Business Source License
|
|
// included in the file licenses/BSL.txt; by using this file, you agree to be bound by the terms of the Business Source
|
|
// License, and you may not use this file except in compliance with the Business Source License.
|
|
//
|
|
// As of the Change Date specified in that file, in accordance with
|
|
// the Business Source License, use of this software will be governed
|
|
// by the Apache License, Version 2.0, included in the file
|
|
// licenses/APL.txt.
|
|
|
|
#include <gflags/gflags.h>
|
|
|
|
#include "dbms/dbms_handler.hpp"
|
|
#include "glue/auth_checker.hpp"
|
|
#include "glue/auth_handler.hpp"
|
|
#include "requests/requests.hpp"
|
|
#include "storage/v2/config.hpp"
|
|
#include "telemetry/telemetry.hpp"
|
|
#include "utils/system_info.hpp"
|
|
#include "utils/uuid.hpp"
|
|
|
|
DEFINE_string(endpoint, "http://127.0.0.1:9000/", "Endpoint that should be used for the test.");
|
|
DEFINE_int64(interval, 1, "Interval used for reporting telemetry in seconds.");
|
|
DEFINE_int64(duration, 10, "Duration of the test in seconds.");
|
|
DEFINE_string(storage_directory, "", "Path to the storage directory where to save telemetry data.");
|
|
DEFINE_string(root_directory, "", "Path to the database durability root dir.");
|
|
|
|
int main(int argc, char **argv) {
|
|
gflags::SetVersionString("telemetry");
|
|
gflags::ParseCommandLineFlags(&argc, &argv, true);
|
|
|
|
// Memgraph backend
|
|
std::filesystem::path data_directory{std::filesystem::temp_directory_path() / "MG_telemetry_integration_test"};
|
|
memgraph::utils::Synchronized<memgraph::auth::Auth, memgraph::utils::WritePrioritizedRWLock> auth_{data_directory /
|
|
"auth"};
|
|
memgraph::glue::AuthQueryHandler auth_handler(&auth_, "");
|
|
memgraph::glue::AuthChecker auth_checker(&auth_);
|
|
|
|
memgraph::storage::Config db_config;
|
|
memgraph::storage::UpdatePaths(db_config, data_directory);
|
|
memgraph::replication::ReplicationState repl_state(ReplicationStateRootPath(db_config));
|
|
|
|
memgraph::dbms::DbmsHandler dbms_handler(db_config
|
|
#ifdef MG_ENTERPRISE
|
|
,
|
|
&auth_, false, false
|
|
#endif
|
|
);
|
|
memgraph::query::InterpreterContext interpreter_context_({}, &dbms_handler, &repl_state, &auth_handler,
|
|
&auth_checker);
|
|
|
|
memgraph::requests::Init();
|
|
memgraph::telemetry::Telemetry telemetry(FLAGS_endpoint, FLAGS_storage_directory, memgraph::utils::GenerateUUID(),
|
|
memgraph::utils::GetMachineId(), false, FLAGS_root_directory,
|
|
std::chrono::seconds(FLAGS_interval), 1);
|
|
|
|
// User defined collector
|
|
uint64_t counter = 0;
|
|
telemetry.AddCollector("test", [&counter]() -> nlohmann::json {
|
|
++counter;
|
|
return {{"vertices", counter}, {"edges", counter}};
|
|
});
|
|
|
|
// Memgraph specific collectors
|
|
telemetry.AddStorageCollector(dbms_handler, auth_);
|
|
#ifdef MG_ENTERPRISE
|
|
telemetry.AddDatabaseCollector(dbms_handler);
|
|
#else
|
|
telemetry.AddDatabaseCollector();
|
|
#endif
|
|
telemetry.AddClientCollector();
|
|
telemetry.AddEventsCollector();
|
|
telemetry.AddQueryModuleCollector();
|
|
telemetry.AddExceptionCollector();
|
|
telemetry.AddReplicationCollector();
|
|
|
|
std::this_thread::sleep_for(std::chrono::seconds(FLAGS_duration));
|
|
|
|
return 0;
|
|
}
|