62 lines
1.6 KiB
Python
62 lines
1.6 KiB
Python
import random
|
|
from abc import abstractmethod
|
|
from typing import Set, Optional, List, Self
|
|
from carnot.carnot import Overlay, Id, Committee, View
|
|
|
|
|
|
class EntropyOverlay(Overlay):
|
|
@abstractmethod
|
|
def advance(self, entropy: bytes) -> Self:
|
|
pass
|
|
|
|
|
|
class FlatOverlay(EntropyOverlay):
|
|
|
|
def __init__(self, current_leader: Id, nodes: List[Id], entropy: bytes):
|
|
self.current_leader = current_leader
|
|
self.nodes = nodes
|
|
self.entropy = entropy
|
|
|
|
def next_leader(self) -> Id:
|
|
random.seed(a=self.entropy, version=2)
|
|
return random.choice(self.nodes)
|
|
|
|
def advance(self, entropy: bytes):
|
|
return FlatOverlay(self.next_leader(), self.nodes, entropy)
|
|
|
|
def is_leader(self, _id: Id):
|
|
return _id == self.leader()
|
|
|
|
def leader(self) -> Id:
|
|
return self.current_leader
|
|
|
|
def is_member_of_leaf_committee(self, _id: Id) -> bool:
|
|
return True
|
|
|
|
def is_member_of_root_committee(self, _id: Id) -> bool:
|
|
return True
|
|
|
|
def is_member_of_child_committee(self, parent: Id, child: Id) -> bool:
|
|
return False
|
|
|
|
def parent_committee(self, _id: Id) -> Optional[Committee]:
|
|
return None
|
|
|
|
def leaf_committees(self) -> Set[Committee]:
|
|
return {frozenset(self.nodes)}
|
|
|
|
def root_committee(self) -> Committee:
|
|
return set(self.nodes)
|
|
|
|
def is_child_of_root_committee(self, _id: Id) -> bool:
|
|
return True
|
|
|
|
def leader_super_majority_threshold(self, _id: Id) -> int:
|
|
return ((len(self.nodes) * 2) // 3) + 1
|
|
|
|
def super_majority_threshold(self, _id: Id) -> int:
|
|
return 0
|
|
|
|
|
|
|