torchft.proto

syntax = "proto3";
package torchft;

message RaftMessageRequest {
    // Request message contains the serialized Raft proto message.
    bytes message = 1;
}

message RaftMessageResponse {
}

message NodeInfo {
    uint64 rank = 1;
    string address = 2;
}

message InfoRequest {
    NodeInfo requester = 1;
}

message InfoResponse {
    repeated NodeInfo peers = 1;
}

service CoordinatorService {
    rpc RaftMessage (RaftMessageRequest) returns (RaftMessageResponse);
    rpc Info (InfoRequest) returns (InfoResponse);
}

message QuorumMember {
    string replica_id = 1;
    string address = 2;
    string store_address = 3;
    int64 step = 4;
}

message Quorum {
    int64 quorum_id = 1;
    repeated QuorumMember participants = 2;
}

message LighthouseQuorumRequest {
    QuorumMember requester = 1;
}

message LighthouseQuorumResponse {
    Quorum quorum = 1;
}

service LighthouseService {
    rpc Quorum (LighthouseQuorumRequest) returns (LighthouseQuorumResponse);
}

message ManagerQuorumRequest {
    int64 rank = 1;
    int64 step = 2;
    string checkpoint_server_addr = 3;
}

message ManagerQuorumResponse {
    int64 quorum_id = 1;
    string address = 2;
    string store_address = 3;
    int64 max_step = 4;
    int64 num_max = 5;
    int64 replica_rank = 6;
    int64 replica_world = 7;
    bool heal = 8;
}

message CheckpointAddressRequest {
    int64 rank = 1;
}

message CheckpointAddressResponse {
    string checkpoint_server_address = 1;
}

message ShouldCommitRequest {
    bool should_commit = 1;
    int64 rank = 2;
    int64 step = 3;
}
message ShouldCommitResponse {
    bool should_commit = 1;
}

service ManagerService {
    rpc Quorum (ManagerQuorumRequest) returns (ManagerQuorumResponse);
    rpc CheckpointAddress(CheckpointAddressRequest) returns (CheckpointAddressResponse);
    rpc ShouldCommit(ShouldCommitRequest) returns (ShouldCommitResponse);
}