diff --git a/src/io/simulator/simulator.hpp b/src/io/simulator/simulator.hpp index 8e3ad85c0..6bad9239e 100644 --- a/src/io/simulator/simulator.hpp +++ b/src/io/simulator/simulator.hpp @@ -23,6 +23,7 @@ namespace memgraph::io::simulator { class Simulator { std::mt19937 rng_; std::shared_ptr simulator_handle_; + uint16_t auto_port_ = 0; public: explicit Simulator(SimulatorConfig config) @@ -30,6 +31,11 @@ class Simulator { void ShutDown() { simulator_handle_->ShutDown(); } + Io RegisterNew() { + Address address = Address::TestAddress(auto_port_++); + return Register(address); + } + Io Register(Address address) { std::uniform_int_distribution seed_distrib; uint64_t seed = seed_distrib(rng_);