"""Кворум и большинство — два разных числа, и путают их постоянно. Блок 13.

ЧТО ЗДЕСЬ ПРОВЕРЯЕТСЯ. В настройке `sentinel monitor <имя> <адрес> <порт>
<кворум>` последнее число задаёт, сколько наблюдателей должны согласиться, что
мастер упал. Его читают как «сколько sentinel достаточно, чтобы переключить», и
это неверно: согласие открывает процедуру, а выполнить её может только тот, кого
выбрало БОЛЬШИНСТВО ВСЕХ известных sentinel. Числа разные, и второе настройкой
не задаётся вовсе.

КАК ЭТО ПОКАЗАНО. Пять sentinel, кворум 2 — и гасятся ТРИ. Остаются двое: их
ровно столько, сколько требует кворум, и меньше, чем большинство от пяти.
Дальше роняется мастер, и печатается, что выжившие записали у себя в журнале.
Числа подобраны именно так, чтобы одно условие выполнялось, а второе нет: на
трёх sentinel с кворумом 2 оба ломаются одновременно, и разницу не увидеть.

ЗАПУСК: python3 bench/valkey-cluster/quorum.py
"""

from __future__ import annotations

import time

from common import (
    Cluster,
    cli,
    free_port,
    head,
    master_addr,
    note,
    preamble,
    row,
)

SENTINEL_NAME = "cache"
PAD = 34
INTERESTING = (
    "+sdown master", "+odown master", "+new-epoch", "+try-failover",
    "+vote-for-leader", "+elected-leader", "+switch-master",
    "-failover-abort-not-elected", "-odown master",
)


def journal(cluster: Cluster, port: int, since: int) -> int:
    """Напечатать журнал до КОНЦА ПЕРВОГО круга попытки переключения.

    Круг повторяется с новой эпохой снова и снова, пока мастер молчит, и
    печатать их все — значит утопить содержательное в повторе. Печатается один
    круг целиком и отдельной строкой то, сколько их было всего.
    """
    lines = cluster.sentinel_log(port)
    shown, rounds = 0, 0
    for line in lines[since:]:
        for mark in INTERESTING:
            if mark in line:
                text = line.split(" # ")[-1].split(" * ")[-1]
                if rounds == 0:
                    print("    " + text[:66])
                    shown += 1
                if "-failover-abort-not-elected" in text:
                    rounds += 1
                elif rounds > 0 and "+try-failover" in text:
                    rounds += 1
                break
    if rounds:
        print("    ... и так по кругу, с новой эпохой каждый раз, пока мастер молчит")
    return len(lines)


def main() -> None:
    preamble("пять sentinel, кворум 2: что открывает кворум и что решает большинство")

    cluster = Cluster()
    try:
        master = free_port()
        replicas = [free_port(), free_port()]
        sentinels = [free_port() for _ in range(5)]
        cluster.start_data(master)
        for r in replicas:
            cluster.start_data(r, replicaof=master)
        for s in sentinels:
            cluster.start_sentinel(s, master, quorum=2, name=SENTINEL_NAME)
        time.sleep(4)

        head(13, "КВОРУМ ОТКРЫВАЕТ ПЕРЕКЛЮЧЕНИЕ, А ВЫПОЛНЯЕТ ЕГО БОЛЬШИНСТВО")
        row("sentinel в наборе", str(len(sentinels)), pad=PAD)
        row("кворум в настройке", "2", pad=PAD)
        row("большинство от пяти", "3 (нигде не настраивается)", pad=PAD)
        row("мастер", f"127.0.0.1:{master}", pad=PAD)
        print()

        watcher = sentinels[0]
        mark = len(cluster.sentinel_log(watcher))

        # Гасятся ТРИ sentinel из пяти. Набор при этом не меняется: выжившие
        # по-прежнему считают, что их пятеро, — они просто перестанут получать
        # от погашенных голоса.
        for s in sentinels[2:]:
            cluster.kill(s)
        time.sleep(2)
        row("осталось живых sentinel", "2 из 5", pad=PAD)
        row("что оставшийся считает мастером",
            ":".join(map(str, master_addr(watcher, SENTINEL_NAME) or ("?", "?"))), pad=PAD)
        print()

        cluster.kill(master)
        print(f"  t=0.00  мастер {master} убит")
        t0 = time.time()
        switched = None
        while time.time() - t0 < 30:
            addr = master_addr(watcher, SENTINEL_NAME)
            if addr and addr[1] != master:
                switched = addr
                break
            time.sleep(0.2)

        print()
        if switched:
            row("переключение произошло", f"{switched[0]}:{switched[1]}", pad=PAD)
        else:
            row("за 30 с переключения НЕ произошло", "", pad=PAD)
            row("кого оставшийся всё ещё зовёт мастером",
                ":".join(map(str, master_addr(watcher, SENTINEL_NAME) or ("?", "?"))), pad=PAD)
        print()
        print("  журнал оставшегося sentinel:")
        journal(cluster, watcher, mark)

        # Проверка обратного: вернуть ОДИН погашенный — и большинство (3 из 5)
        # снова есть, при том же самом кворуме.
        print()
        cluster.start_sentinel(sentinels[2], master, quorum=2, name=SENTINEL_NAME)
        print("  поднят ещё один sentinel: живых стало 3 из 5, кворум тот же")
        t0 = time.time()
        after = None
        while time.time() - t0 < 40:
            addr = master_addr(watcher, SENTINEL_NAME)
            if addr and addr[1] != master:
                after = addr
                break
            time.sleep(0.2)
        print()
        if after:
            row("переключение произошло через", f"{time.time() - t0:.2f} с после этого", pad=PAD)
            row("новый мастер", f"{after[0]}:{after[1]}", pad=PAD)
            row("он принимает запись", cli(after[1], "set", "k", "v"), pad=PAD)
        else:
            row("за 40 с переключения так и НЕ произошло", "", pad=PAD)

        note(
            """
            Журнал читается подряд, и в нём видны обе стадии. Сначала
            +odown: кворум собран, мастер объявлен упавшим — это ровно то
            число, что стоит в настройке, и оно выполнилось. Потом
            +try-failover и +vote-for-leader: выживший sentinel пытается
            стать ведущим. И следом -failover-abort-not-elected: голосов двух
            из пяти не хватило, чтобы им стать, и переключение отменено.

            То есть кворум отвечает на вопрос «сломалось ли», а большинство —
            на вопрос «кому чинить». Второе число нигде не настраивается: это
            всегда большинство ОТ ЧИСЛА ИЗВЕСТНЫХ sentinel, и уменьшить его,
            понизив кворум, нельзя.

            Отсюда практическое правило, которое из одной настройки не
            выводится: потерю большинства sentinel система не переживёт, каким
            бы низким ни был кворум, — а значит, ставить их надо нечётным
            числом и в разные места отказа. Вторая половина блока это и
            показывает: кворум не менялся вовсе, вернулся ровно один
            наблюдатель, и переключение, стоявшее насмерть, состоялось.
            """
        )
    finally:
        cluster.stop_all()


if __name__ == "__main__":
    main()