Allows bootstrapping peer on addition (qdrant#529)

* Allows bootstrapping peer on addition * Review: warn on address replacement * Review: more detailed comments
snayan06 · May 2, 2022 · 40c4c1e · 40c4c1e
1 parent 794384f
commit 40c4c1e
Show file tree

Hide file tree

Showing 12 changed files with 613 additions and 81 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -33,6 +33,7 @@ schemars = { version = "0.8.8", features = ["uuid"] }
 itertools = "0.10"
 anyhow = "1.0.57"
 futures = "0.3.21"
+clap = { version = "3.1.12", features = ["derive"] }
 
 config = "~0.13.1"
 

diff --git a/lib/api/src/grpc/proto/raft_service.proto b/lib/api/src/grpc/proto/raft_service.proto
@@ -5,12 +5,37 @@ package qdrant;
 import "google/protobuf/empty.proto";
 
 service Raft {
-  /*
-  Send Raft message to another peer
-   */
-  rpc Send (RaftMessage) returns (google.protobuf.Empty) {}
+  // Send Raft message to another peer
+  rpc Send (RaftMessage) returns (google.protobuf.Empty);
+  // Send to bootstrap peer
+  // Returns uri by id if bootstrap knows this peer
+  rpc WhoIs (PeerId) returns (Uri);
+  // Send to bootstrap peer
+  // Proposes to add this peer Uri and ID to a map of all peers
+  // Returns all peers
+  rpc AddPeerToKnown (Peer) returns (AllPeers);
+  // Send to bootstrap peer
+  // Proposes to add this peer as participant of consensus
+  rpc AddPeerAsParticipant (Peer) returns (google.protobuf.Empty);
 }
 
 message RaftMessage {
     bytes message = 1;
 }
+
+message AllPeers {
+    repeated Peer all_peers = 1;
+}
+
+message Peer {
+    string uri = 1;
+    uint64 id = 2;
+}
+
+message PeerId {
+  uint64 id = 1;
+}
+
+message Uri {
+  string uri = 1;
+}