"""Чем отличается набор доступных команд — блоки 5-7.

ГЛАВНОЕ, ЧТО ЗДЕСЬ ПРОВЕРЯЕТСЯ. «Перейдём на кластер» звучит как решение про
эксплуатацию, а оказывается решением про КОД: часть команд, работавших годами,
в кластере перестаёт работать, и перестаёт по-разному. Что-то отказывает
всегда, что-то — только когда ключи легли в разные слоты, то есть в проде и
через полгода.

ПОЧЕМУ ЭТО НЕ «ОГРАНИЧЕНИЯ VALKEY», А СЛЕДСТВИЕ ПРЕДЫДУЩИХ БЛОКОВ. Все три
отказа ниже выводятся из одного: в кластере адресуется не узел, а слот, и
команда выполняется ЦЕЛИКОМ на одном узле. Значит, команда, которой нужны два
ключа из разных слотов, не может быть выполнена нигде. А в режиме Sentinel
узел один, и тех же ограничений нет вовсе.

ЧТО ЗДЕСЬ ПЕЧАТАЕТСЯ. Дословные ответы сервера на одни и те же команды,
отправленные в двух режимах. Ни одного измерения времени.

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

from __future__ import annotations

import time

from common import Cluster, cli, free_port, head, note, preamble, role_of, row

SENTINEL_NAME = "cache"
PAD = 42


def pair(label: str, cluster_answer: str, sentinel_answer: str) -> None:
    row(label, "", pad=PAD)
    row("  в кластере", cluster_answer, pad=PAD)
    row("  под sentinel", sentinel_answer, pad=PAD)


def block5(cport: int, sport: int) -> None:
    head(5, "ОДНА КОМАНДА НА ДВА КЛЮЧА: ГДЕ ПРОХОДИТ ГРАНИЦА")

    # Ключи подобраны так, чтобы разошлись по слотам: это видно из CLUSTER
    # KEYSLOT прямо здесь, а не утверждается на словах.
    row("CLUSTER KEYSLOT foo", cli(cport, "cluster", "keyslot", "foo"), pad=PAD)
    row("CLUSTER KEYSLOT bar", cli(cport, "cluster", "keyslot", "bar"), pad=PAD)
    row("CLUSTER KEYSLOT {u}:a", cli(cport, "cluster", "keyslot", "{u}:a"), pad=PAD)
    row("CLUSTER KEYSLOT {u}:b", cli(cport, "cluster", "keyslot", "{u}:b"), pad=PAD)
    print()
    pair("MSET foo 1 bar 2",
         cli(cport, "mset", "foo", "1", "bar", "2"),
         cli(sport, "mset", "foo", "1", "bar", "2"))
    print()
    pair("MSET {u}:a 1 {u}:b 2",
         cli(cport, "mset", "{u}:a", "1", "{u}:b", "2"),
         cli(sport, "mset", "{u}:a", "1", "{u}:b", "2"))

    note(
        """
        Отказ не про «две записи сразу», а про два РАЗНЫХ слота: те же две
        записи с общей фигурной скобкой проходят. Скобки — не украшение
        имени, а указание хешировать только то, что внутри; поэтому
        {u}:a и {u}:b попадают в один слот и оказываются на одном узле.

        Отсюда правило проектирования ключей, которое дороже всего обходится
        задним числом: в кластере набор ключей, которые понадобится трогать
        одной командой, надо решать ДО первой записи. Под sentinel такого
        решения принимать не надо вовсе — узел один, любые ключи рядом.

        Ответ MOVED во второй строке второй пары — не отказ: узел сообщает,
        что оба ключа лежат вместе, но не у него. Клиент кластера сходил бы
        по адресу и выполнил команду целиком.
        """
    )


def block6(cport: int, sport: int) -> None:
    head(6, "НУМЕРОВАННЫЕ БАЗЫ: В ОДНОМ РЕЖИМЕ ЕСТЬ, В ДРУГОМ НЕТ")

    pair("SELECT 1", cli(cport, "select", "1"), cli(sport, "select", "1"))
    print()
    pair("SWAPDB 0 1", cli(cport, "swapdb", "0", "1"), cli(sport, "swapdb", "0", "1"))
    print()
    pair("CONFIG GET databases",
         " ".join(cli(cport, "config", "get", "databases").split()),
         " ".join(cli(sport, "config", "get", "databases").split()))

    note(
        """
        Третья пара объясняет первые две: в кластере база ровно одна, и
        SELECT 1 — это выход за границу, а не запрещённая команда. Формулировка
        ответа тут важнее, чем кажется: сервер говорит про диапазон, то есть
        SELECT 0 в кластере работает и будет работать.

        Практическая цена этого пункта обычно недооценена. Разделение
        окружений по номеру базы — «тесты в 1, кэш сессий в 2» — в кластере
        не переносится вообще, и переписывать приходится не строчку конфига,
        а привычку.
        """
    )


def block7(cport: int, sport: int, sentinel_port: int) -> None:
    head(7, "КОМАНДЫ УПРАВЛЕНИЯ: ТРИ РАЗНЫХ НАБОРА НА ОДНОМ ДВОИЧНОМ ФАЙЛЕ")

    row("CLUSTER INFO", "", pad=PAD)
    row("  узел кластера", cli(cport, "cluster", "info").splitlines()[0], pad=PAD)
    row("  узел под sentinel", cli(sport, "cluster", "info").splitlines()[0], pad=PAD)
    row("  сам sentinel", cli(sentinel_port, "cluster", "info").splitlines()[0], pad=PAD)
    print()
    row("SENTINEL get-master-addr-by-name cache", "", pad=PAD)
    row("  узел кластера",
        cli(cport, "sentinel", "get-master-addr-by-name", SENTINEL_NAME).replace("\n", ":"), pad=PAD)
    row("  узел под sentinel",
        cli(sport, "sentinel", "get-master-addr-by-name", SENTINEL_NAME).replace("\n", ":"), pad=PAD)
    row("  сам sentinel",
        cli(sentinel_port, "sentinel", "get-master-addr-by-name", SENTINEL_NAME).replace("\n", ":"), pad=PAD)
    print()
    row("GET foo", "", pad=PAD)
    row("  узел под sentinel", cli(sport, "get", "foo") or "(пусто)", pad=PAD)
    row("  сам sentinel", cli(sentinel_port, "get", "foo"), pad=PAD)

    note(
        """
        Один и тот же valkey-server в трёх ролях отвечает на одни и те же
        команды тремя разными способами. Это и есть ответ на вопрос «а можно
        ли совместить»: нельзя не потому, что кто-то запретил, а потому что
        режимов три и набор команд у каждого свой.

        Последняя пара — самая практическая. Sentinel не хранит данных, и
        GET у него не просто пустой, а неизвестный. Клиент, по ошибке
        настроенный на порт sentinel как на порт данных, не получит
        «пусто» — он получит отказ на каждую команду.
        """
    )


def main() -> None:
    preamble("одни и те же команды в двух режимах, ответы сервера дословно")

    cluster = Cluster()
    try:
        cports = [free_port() for _ in range(3)]
        for p in cports:
            cluster.start_data(p, cluster=True)
        cluster.form_cluster(cports, replicas=0)

        m = free_port()
        rep = free_port()
        sen = free_port()
        cluster.start_data(m)
        cluster.start_data(rep, replicaof=m)
        cluster.start_sentinel(sen, m, quorum=1, name=SENTINEL_NAME)
        time.sleep(2)

        # Узел кластера для вопросов берётся первый; какие слоты на нём —
        # неважно, потому что все три блока про то, что команда вообще
        # ОТКЛОНЕНА, а не про то, где лежит ключ.
        assert role_of(m) == "master"
        block5(cports[0], m)
        block6(cports[0], m)
        block7(cports[0], m, sen)
    finally:
        cluster.stop_all()


if __name__ == "__main__":
    main()