Compare commits

..
26 Commits
Author SHA1 Message Date
gvsafronov d511e347df Update README.md 2026-10-06 21:58:56 +00:00
gvsafronov 0394ad893e Delete internal/storage/aof.go 2026-10-06 21:43:05 +00:00
gvsafronov 3f4ad87e59 Update internal/storage/transactions.go 2026-10-06 21:42:52 +00:00
gvsafronov bd6c8acda6 Update internal/storage/collection.go 2026-10-06 21:42:15 +00:00
gvsafronov dd5f11fd3d Update README.md 2026-10-06 21:41:33 +00:00
gvsafronov b75dffab84 Update README.md 2026-10-06 21:35:16 +00:00
gvsafronov 2541233e40 Update internal/cluster/raft_coordinator.go 2026-10-05 21:30:44 +00:00
gvsafronov f191e5b328 Update internal/storage/transactions.go 2026-10-05 21:29:54 +00:00
gvsafronov cc4ec0c029 Update internal/repl/pager.go 2026-10-05 21:29:14 +00:00
gvsafronov 06e9558839 Update internal/repl/repl.go 2026-10-05 21:28:43 +00:00
gvsafronov 3f1748bd6c Update internal/repl/repl.go 2026-10-05 21:27:58 +00:00
gvsafronov 79c1a0171e Update internal/config/dynamic_config.go 2026-10-05 21:27:17 +00:00
gvsafronov ba4f7bfaa7 Update internal/config/config.go 2026-10-05 21:26:41 +00:00
gvsafronov 1607f4108d Update cmd/futriis/main.go 2026-10-05 21:25:52 +00:00
gvsafronov 7804ce2338 Update internal/migration/schema_migrator.go 2026-10-04 21:04:40 +00:00
gvsafronov 904f8081bd Upload files to "internal/repl" 2026-10-04 21:03:46 +00:00
gvsafronov d3353b85ba Delete internal/repl/repl.go 2026-10-04 21:03:30 +00:00
gvsafronov 5762b7f6db Upload files to "internal/storage" 2026-10-04 21:03:01 +00:00
gvsafronov b4ccb910c4 Delete internal/storage/document.go 2026-10-04 21:02:48 +00:00
gvsafronov 43fda3f796 Upload files to "internal/storage" 2026-10-04 21:02:26 +00:00
gvsafronov cfe1538fcf Upload files to "internal/storage" 2026-10-04 21:02:12 +00:00
gvsafronov f0694ca563 Delete internal/storage/collection.go 2026-10-04 21:01:50 +00:00
gvsafronov 3c25122cd1 Delete internal/storage/persistence.go 2026-10-04 20:59:32 +00:00
gvsafronov 523f12fec0 Update internal/backup/backup_scheduler.go 2026-10-04 20:39:51 +00:00
gvsafronov 7fd163af92 Update internal/serializer/msgpack.go 2026-10-04 20:18:32 +00:00
gvsafronov afbc88c3e8 Update internal/compression/compression.go 2026-10-04 18:22:30 +00:00
15 changed files with 4617 additions and 3794 deletions

No files matched your search

+38 -46
View File
@@ -75,7 +75,8 @@ futriix — это лёгкая распределённая NoSQL‑СУБД н
Ключевые механизмы надёжности и масштабируемости:
* WAL — устойчивость к сбоям
* SAGA- для внешних batch
* Batch — атомарные группы операций
* SAGA — для внешних batch
* Горизонтальное масштабирование
@@ -227,7 +228,7 @@ futriix — это лёгкая распределённая NoSQL‑СУБД н
| **Документ (Document)** | `map[string]interface{}` | Динамическая схема полей |
| | Версия (`uint64`) | Для оптимистичных блокировок |
| | Временные метки | Создание, обновление, удаление (Unix‑миллисекунды) |
| **Транзакция (Transaction)** | WAL (`Write‑Ahead Log`) | Журнал упреждающей записи для durability |
| **Batch (Batch)** | WAL (`Write‑Ahead Log`) | Журнал упреждающей записи для durability |
| | Кэш операций | Локальное хранение изменений до коммита |
| | Список блокировок | Отслеживание затронутых документов |
| **Триггер (Trigger)** | `sync.Map` с составным ключом | `collection|event|name` для быстрого поиска |
@@ -246,7 +247,7 @@ futriix — это лёгкая распределённая NoSQL‑СУБД н
|:-:|----------|-------------------|-----------|-----------|
| 1 | **Wait-free операции доступа** | • Чтение: `sync.Map.Load()` — атомарная операция без блокировок<br>• Запись: `sync.Map.Store()` — атомарная замена значения<br>• Удаление: `sync.Map.Delete()` — атомарное удаление | **O(1) амортизированное** | *Амортизированная сложность — средняя стоимость операции в серии операций, даже если отдельные операции могут быть дорогими* |
| 2 | **Поиск по индексу** | **Уникальный индекс:**<br>1. `index.data.Load(value)` → получение ID документа<br>2. `docs.Load(ID)` → возврат документа<br><br>**Неуникальный индекс:**<br>1. `index.data.Range()` → сканирование всех записей<br>2. Сравнение значений с учётом типов (`compareValues`)<br>3. Сбор всех подходящих ID | **Уникальный: O(1)**<br><br>**Неуникальный: O(n)** | `n` — размер индекса |
| 3 | **Оптимистичная блокировка транзакций** | 1. `BeginTransaction()` → фиксация текущих версий документа<br>2. Выполнение операций в локальном кэше<br>3. `Prepare()` → проверка версий всех затронутых документов<br>4. `Commit()` → применение изменений, инкремент версий<br>5. При конфликте → `Abort()` и повтор | **E[T] = (1 / (1 - P_conflict)) × [O(M) + O(M × (V + K)) + O(WAL)]** | `𝔼[T]` — мат. ожидание времени выполнения<br>`1 - P_conflict` — вероятность успешного коммита<br>`M` — количество документов в транзакции<br>`V` — количество полей в документе<br>`K` — количество индексов<br>`WAL` — размер журнала |
| 3 | **Batch (атомарная группа операций)** | 1. `NewBatch()` → создание batch<br>2. Добавление операций (`AddInsert`, `AddUpdate`, `AddDelete`, `AddRestore`)<br>3. `Commit()` → запись всего batch в WAL с fsync<br>4. Применение операций к in-memory структурам<br>5. При ошибке — откат (best-effort) | **E[T] = O(M) + O(WAL)** | `M` — количество операций в batch<br>`WAL` — размер журнала |
| 4 | **Валидация ограничений (Constraints)** | Последовательная проверка документа:<br>1. Обязательные поля → проверка наличия<br>2. Минимальные значения (`MinValues`) → сравнение чисел<br>3. Максимальные значения (`MaxValues`) → сравнение чисел<br>4. Regex паттерны → сопоставление строк<br>5. Enum значения → поиск в разрешённом списке | **O(k + m)** | `k` — количество полей в документе<br>`m` — количество ограничений |
| 5 | **Выполнение триггеров** | Для каждого триггера события:<br>1. Проверка `Enabled`<br>2. Оценка условия (`TriggerCondition`)<br>3. Сопоставление оператора с полем документа<br>4. Выполнение действия (`abort`/`skip`/`modify`/`log`)<br>5. Логирование с временной меткой и длительностью | **O(T × (C + O))** | `T` — количество триггеров на событие<br>`C` — количество условий в триггере<br>`O` — количество операций в триггере |
| 6 | **Мягкое удаление** | **SoftDelete:**<br>1. `doc.DeletedAt = now`<br>2. `doc.Version++`<br>3. `docs.Store(id, doc)` — сохранение, не удаление<br>4. `removeFromIndexes(doc)` — исключение из поиска<br><br>**Restore:**<br>1. `doc.DeletedAt = 0`<br>2. `doc.Version++`<br>3. `docs.Store(id, doc)`<br>4. `addToIndexes(doc)` — возврат в индексы | **O(1) + O(K)** | `K` — количество индексов в коллекции |
@@ -271,6 +272,7 @@ futriix — это лёгкая распределённая NoSQL‑СУБД н
| Мягкое удаление | Флаг deleted_at + удаление из индексов | Возможность восстановления |
| Асинхронные триггеры | Буферизированный канал | Не блокируют основную операцию |
| Плагины в песочнице | Изолированное Lua‑состояние | Безопасное расширение |
| Batch | WAL + локальный кэш | Атомарность без ACID |
<br>
<br>
@@ -287,7 +289,7 @@ futriix — это лёгкая распределённая NoSQL‑СУБД н
| Поиск по неуникальному индексу | $O(n)$ | $n$ — размер индекса |
| Обновление документа | $O(1) + O(k)$ | + проверка версий |
| Удаление документа | $O(1) + O(k)$ | + возможная очистка индексов |
| Транзакция (commit) | $O(m) + \text{Raft}$ | $m$ — количество операций |
| Batch (commit) | $O(m) + \text{Raft}$ | $m$ — количество операций |
| Валидация ограничений | $O(f + c)$ | $f$ — поля, $c$ — ограничения |
| Экспорт базы данных | $O(N)$ | $N$ — общее количество документов |
@@ -314,12 +316,12 @@ futriix — это лёгкая распределённая NoSQL‑СУБД н
- **Процессор:** 64‑битный Intel или AMD (x86‑64)
- **Оперативная память:**
- Linux: минимум 6 ГБ
- Illumos: минимум 8 ГБ (из‑за особенностей управления памятью и ZFS по умолчанию)
- Linux: минимум 6 ГБ
- Illumos: минимум 8 ГБ (из‑за особенностей управления памятью и ZFS по умолчанию)
> [!IMPORTANT]
> **Важно!!!**
> Для нагрузочного тестирования, шардинга и одновременной работы множества узлов(от 5 и выше) рекомендуется не менее 16 ГБ оперативной памяти на узел
> Для нагрузочного тестирования, шардинга и одновременной работы множества узлов(от 5 и выше) рекомендуется не менее 16 ГБ оперативной памяти на узел
<br>
@@ -357,7 +359,7 @@ unzip -v
> * Дисковая подсистема: SSD/NVMe; HDD допустим только для тестов
> * Файловая система: предпочтительно ZFS (особенно на Illumos) или XFS/ext4 на Linux
> * Сеть: стабильное соединение между узлами, желательно 1 Gbps+; для Raft‑кластеров важна низкая задержка (latency)
> * Время: синхронизация времени между узлами (NTP/Chrony) обязательна для корректной работы MVCC, WAL и транзакций
> * Время: синхронизация времени между узлами (NTP/Chrony) обязательна для корректной работы WAL и batch-операций
<br>
@@ -392,7 +394,7 @@ go version
* **[wal]** — журнал предзаписи (сегменты, синхронизация)
* **[mvcc]** — многоверсионность (версии, TTL)
* **[batch]** — настройки batch-операций (размер, таймауты)
* **[saga]** — распределённые транзакции
@@ -544,8 +546,8 @@ curl -X POST http://localhost:8080/api/v1/config \
"replication.sync_replication": true,
"wal.segment_size_mb": 128,
"wal.sync_interval_sec": 3,
"mvcc.max_versions_per_doc": 20,
"mvcc.retention_days": 14,
"batch.max_size": 1000,
"batch.timeout_ms": 5000,
"saga.enabled": true,
"saga.coordinator_count": 5,
"backpressure.enabled": true,
@@ -629,8 +631,8 @@ $ ./futriix
### Тестирование
Для проверки корректности функционирования субд на уровне исходного кода, был разработа набор из пяти тестов: (регрессионный, smoke-тест, функциональный, интеграционный и нагрузочный).
Разработанный набор из пяти вышеупомянутых тестов на языке Lua обеспечивает комплексную проверку всех ключевых компонентов СУБД: CRUD-операций, индексов, транзакций, ограничений целостности, ACL, триггеров, MVCC-версионирования, а также взаимодействия API с хранилищем и кластерной координации.
Для проверки корректности функционирования субд на уровне исходного кода, был разработан набор из пяти тестов: (регрессионный, smoke-тест, функциональный, интеграционный и нагрузочный).
Разработанный набор из пяти вышеупомянутых тестов на языке Lua обеспечивает комплексную проверку всех ключевых компонентов СУБД: CRUD-операций, индексов, batch-операций, ограничений целостности, ACL, триггеров, версионирования через WAL, а также взаимодействия API с хранилищем и кластерной координации.
Регрессионный тест гарантирует, что изменения кода не нарушили существующую функциональность, smoke-тест выполняет быструю проверку доступности и базовой работоспособности системы.
Функциональный и интеграционный тесты проверяют корректность реализации бизнес-требований и взаимодействие между компонентами, а нагрузочный тест оценивает производительность (латентность, пропускную способность) под различными сценариями использования.
@@ -690,7 +692,7 @@ Futriix поддерживает привычный MongoDB‑синтаксис
> [!NOTE]
>**Выполнение операций**<br>
> Операции выполняются в рамках гарантий изоляции, обеспечиваемых MVCC и протоколом консенсуса Raft.
> Операции выполняются в рамках гарантий изоляции, обеспечиваемых WAL и протоколом консенсуса Raft.
> Для распределённых транзакций применяется протокол Saga.
@@ -944,19 +946,17 @@ futriix:~> db.startSession()
## Batch
В СУБД **futriix** для обеспечения надёжности (durability), атомарности и быстрого восстановления базы данных реализована двухуровневая система журналирования:
В СУБД **futriix** для обеспечения надёжности (durability), атомарности и быстрого восстановления базы данных реализована система журналирования на основе **WAL (Write-Ahead Log)**.
1. **WAL (Write-Ahead Log)** — сегментированный журнал предзаписи для **batch-операций** (атомарных групп изменений). По умолчанию хранится в каталоге `futriix/wal_segments/`.
2. **AOF (Append-Only File)** — журнал дозаписи для **одиночных мутаций**, выполненных вне batch (прямые вызовы `Insert`, `Update`, `Delete`, `Restore`, а также операции из REPL, HTTP-API и Lua-плагинов). По умолчанию хранится в каталоге `futriix/aof/`.
Все мутации — как групповые (batch), так и одиночные (прямые вызовы `Insert`, `Update`, `Delete`, `Restore` из REPL, HTTP-API и Lua-плагинов) — проходят через WAL. Это даёт:
Такое разделение даёт:
- **Атомарность** для сложных составных операций (batch).
- **Гарантированную сохранность** каждой отдельной мутации (AOF).
- **Быстрое восстановление** через воспроизведение обоих журналов при старте.
- **Разделение нагрузки**: batch-операции пишутся в WAL сегментами по 64 МБ, одиночные мутации — в AOF с ротацией по 64 МБ.
- **Гарантированную сохранность** каждой отдельной мутации.
- **Быстрое восстановление** через воспроизведение WAL при старте.
- **Единый источник истины** — один журнал, один fsync-путь, один формат хранения.
##№ Что такое Batch
### Что такое Batch
**Batch** — это группа операций, применяемых **атомарно**: либо все операции из группы применяются, либо ни одна. Batch заменяет полноценные ACID-транзакции в локальном узле и работает по принципу «всё или ничего» через WAL.
@@ -966,7 +966,6 @@ futriix:~> db.startSession()
- **Откат при ошибке применения**: если операция падает на середине применения, уже применённые операции откатываются (best-effort) — полная атомарность гарантируется на уровне WAL при восстановлении.
- **Короткое время жизни**: batch создаётся, наполняется операциями и коммитится в рамках одной логической операции. Не поддерживает savepoints, таймауты, распределённую координацию.
- **Реестр активных batch'ей**: `BatchManager.activeBatches` (см. `runtime_limits.go`) защищает документы, участвующие в незакоммиченном batch, от вытеснения (eviction) при нехватке памяти.
- **AOF-дублирование**: после коммита batch также пишется в AOF для ускорения восстановления.
**Операции batch:**
@@ -980,27 +979,19 @@ futriix:~> db.startSession()
| `batch.Commit()` | Применить batch атомарно через WAL |
### Что такое AOF
### Как WAL обеспечивает durability
**AOF (Append-Only File)** — журнал дозаписи, который логирует **каждую отдельную мутацию**, выполненную вне batch. Это нужно, потому что прямые вызовы `coll.Insert()`, `coll.Update()`, `coll.Delete()` идут мимо WAL, и без AOF такие изменения могли бы потеряться при сбое.
**WAL (Write-Ahead Log)** — сегментированный журнал предзаписи, который является **единственным источником истины** для durability. Каждая мутация сначала попадает в WAL, синхронизируется на диск, и только потом применяется к in-memory структурам.
**Ключевые свойства AOF:**
**Ключевые свойства WAL:**
- **Формат**: JSON-lines (по одной JSON-записи на строку) с CRC32 для верификации целостности.
- **Ротация**: файлы разбиваются по размеру 64 МБ, имена вида `aof_<N>.log`.
- **fsync**: каждая запись синхронизируется с диском через `file.Sync()` для гарантии сохранности.
- **Восстановление при старте**: `ReplayAOF(storage)` читает все файлы и применяет операции к хранилищу.
- **Идемпотентность**: повторное применение операции не ломает состояние (insert/update/delete идемпотентны по ключу).
- **Автоматическое логирование**: методы `Collection.Insert`, `Update`, `Delete`, `PermanentDelete`, `RestoreDeleted` вызывают `LogAOFInsert`, `LogAOFUpdate`, `LogAOFDelete`, `LogAOFRestore` соответственно.
**Типы записей AOF:**
| Тип | Описание |
|-----|----------|
| `insert` | Вставка документа |
| `update` | Обновление документа |
| `delete` | Удаление документа (soft или permanent) |
| `restore` | Восстановление мягко удалённого документа |
- **Сегментация**: файлы `wal_segment_<N>.log` разбиваются по 64 МБ с автоматической ротацией.
- **fsync на запись**: каждая запись (или группа записей в рамках `flushBatch`) синхронизируется с диском через `file.Sync()`.
- **CRC32 per record**: каждая запись защищена контрольной суммой, включающей LSN — обнаружение частичных/torn-записей при восстановлении.
- **WAL Index**: индекс `wal_index_<N>.json` для быстрого поиска записей по LSN.
- **Pipeline записи**: группировка нескольких команд в один fsync-цикл (`flushBatch`) для снижения накладных расходов.
- **Восстановление при старте**: `ReadAll()` читает все сегменты, применяет операции к хранилищу; это единственный путь восстановления.
- **Идемпотентность**: повторное применение операции не ломает состояние.
---
@@ -1041,6 +1032,7 @@ saga.AddStep("reserve_payment",
nil)
sagaManager.Execute(saga)
```
<p align="right">(<a href="#readme-top">К началу</a>)</p>
@@ -1846,7 +1838,7 @@ Idle → Preparing → Migrating → Delta Sync → Validating → Completed
В субд `futriix` также как и в традиционных субд, есть механизм **Ограничений (Constraints)**.
**Ограничения (Constraints)**-это декларативные гарантии целостности (правила), действующие локально на каждом узле и согласующиеся через оркестратор SAGA. Они обеспечивают, что каждый шаг распределённого сценария сохраняет инварианты данных; конфликты и нарушения фиксируются до фиксации шага, а при необходимости компенсируются в рамках eventual consistency.
Ограничения на уровне коллекции поддерживают: `обязательные поля (required)`, `уникальность (unique)`, `минимальные/максимальные значения (min/max)`, `regex-паттерны` и `enum-списки`(фиксированные наборы допустимых значений, которые жёстко задают «разрешённые состояния» поля. Они работают как "строгая шпаргалка" для данных: нельзя сохранить ничего лишнего, а движок сразу фиксирует несоответствия — ещё **до попадания в WAL** и **до участия в MVCC‑снапшотах**.) допустимых значений. Все ограничения проверяются автоматически при вставке и обновлении документов.
Ограничения на уровне коллекции поддерживают: `обязательные поля (required)`, `уникальность (unique)`, `минимальные/максимальные значения (min/max)`, `regex-паттерны` и `enum-списки`(фиксированные наборы допустимых значений, которые жёстко задают «разрешённые состояния» поля. Они работают как "строгая шпаргалка" для данных: нельзя сохранить ничего лишнего, а движок сразу фиксирует несоответствия — ещё **до попадания в WAL** и **до участия в batch-снапшотах**.) допустимых значений. Все ограничения проверяются автоматически при вставке и обновлении документов.
```sh
# Добавление обязательного поля
@@ -2757,8 +2749,8 @@ futriix:~> doc compression employees 550e8400-e29b-41d4-a716-446655440000
**Вопрос: Почему версия обозначена как «Futriix 2 i²», а не просто «2.0»?**<br>
<br>
<br>
**Ответ: «2» — номер версии: здесь собраны ключевые компоненты (шардинг, WAL, MVCC и др.).**
**i2i2 — математическая метафора: хотя ii — мнимая единица, её квадрат даёт абсолютно точный результат (i2=−1i2=−1). Так и Futriix: даже в сложной распределённой среде с параллельными операциями (wait‑free, lock‑free) система выдаёт строго предсказуемые результаты — за счёт Raft, ACID и MVCC. Это близко философии OpenIndiana/Illumos: детерминизм и контроль над ресурсами.**
**Ответ: «2» — номер версии: здесь собраны ключевые компоненты (шардинг, WAL, Batch и др.).**
**i2i2 — математическая метафора: хотя ii — мнимая единица, её квадрат даёт абсолютно точный результат (i2=−1i2=−1). Так и Futriix: даже в сложной распределённой среде с параллельными операциями (wait‑free, lock‑free) система выдаёт строго предсказуемые результаты — за счёт Raft, Batch и WAL. Это близко философии OpenIndiana/Illumos: детерминизм и контроль над ресурсами.**
<p align="right">(<a href="#readme-top">К началу</a>)</p>
@@ -2789,7 +2781,7 @@ futriix:~> doc compression employees 550e8400-e29b-41d4-a716-446655440000
- [x] Библиотека "raft-boltdb" заменить на встроенное файловое хранилище
- [x] Реализовать уникальные и составные индексы
- [x] Реализовать временные метки для основных объектов субд (таппл, коллекция, документ, поле, индекс, транзакция, ACL, узел кластера)
- [x] Реализовать MVCC версионирование - поддержка множественных версий документов
- [x] Реализовать Batch-версионирование - поддержка множественных версий документов через WAL
- [x] Реализовать временные метки для объектов субд (Ограничений, Импорта-Экспорта, Триггеров, Lua-плагинов)
- [x] Реализовать обработку состояний "split-brain"
- [x] Реализовать Pipeline репликации-группировку нескольких команд в один Raft лог для уменьшения сети
@@ -2808,7 +2800,7 @@ futriix:~> doc compression employees 550e8400-e29b-41d4-a716-446655440000
- [x] Реализовать Distributed Transactions - двухфазный коммит для распределённых транзакций
- [x] Реализовать Deadlock Detection - циклический детектор дедлоков с таймаутами (DFS по графу ожидания)
- [x] Реализовать Transaction Timeout - настраиваемый таймаут для долгих транзакций
- [x] Вынести настройи настройки MVCC,WAL и настройки плагинов из исходного кода проекта в файл конфигурации "config.toml"
- [x] Вынести настройи настройки Batch, WAL и настройки плагинов из исходного кода проекта в файл конфигурации "config.toml"
- [x] Реализовать выделение жирным шифтом все информационные сообщения в скриптах сборки
- [x] Реализовать ограничения на количество создаваемых Lua-состояний в плагинах
- [x] Реализовать WAL recovery сделать асинхронным для ускорения скорости работы субд
+20 -101
View File
@@ -10,58 +10,13 @@
* Файл: cmd/futriis/main.go
* Назначение: Точка входа в приложение СУБД futriis.
*
* ИСПРАВЛЕНО:
* - Валидация пути к конфигурации и проверка прав доступа к файлу
* - Валидация конфигурации после загрузки (ValidateConfig)
* - WAL и data-dir привязаны к директории futriis, а не CWD
* - Graceful shutdown: убраны os.Exit(1) до завершения defer'ов,
* используется переменная exitCode и os.Exit в самом конце
* - Ожидание запуска HTTP-сервера с таймаутом
* - SAGA оркестратор останавливается всегда (даже если disabled)
* - Проверка allow-list для плагинов перед StartPlugin
* - Заменён SIGHUP на SIGUSR1 (reload) + SIGTERM/SIGINT (shutdown)
* - Проверка nil для raftCoordinator перед использованием
* - repl.NewRepl теперь возвращает (*Repl, error)
* - Интеграция с Prometheus: /metrics endpoint
* - ДОБАВЛЕНО: поддержка TLS на HTTP API (вариант A — TLS внутри futriiX).
* Теперь используется api.NewHTTPServerWithTLS(..., &cfg.Security).
* - ИСПРАВЛЕНО (UI): убраны лишние пустые строки перед баннером, чтобы
* строка "futriis 3i²(by 29.09.2026)" шла сразу после сообщения ACL
* о созданном admin-пользователе.
* - ИСПРАВЛЕНО (UI): текст баннера окрашивается в точный #00bfff
* (Deep Sky Blue) через utils.PrintlnDeepSkyBlueReset, чтобы совпадать
* с цветом надписи "futriis 3i²(by 29.09.2026)".
* - ИСПРАВЛЕНО: acl.NewACLManager() теперь возвращает (*ACLManager, error).
* Обрабатываем ошибку и завершаемся через os.Exit(1) с логом,
* без panic и stack trace.
*
* ИСПРАВЛЕНО (2026-10): после замены ACID-транзакций на batch-операции
* функции storage.InitTransactionManager, storage.SetTransactionLogger,
* storage.StopTransactionManager были удалены. Заменены на:
* - storage.InitBatchManager(walPath) — инициализирует batch-менеджер,
* WAL и AOF (append-only file для мутаций вне batch);
* - storage.SetBatchLogger(logger) — устанавливает логгер;
* - storage.StopBatchManager() — останавливает batch-менеджер,
* WAL и AOF.
* Дополнительно: при shutdown логируется состояние batch-статистики
* (количество выполненных batch'ей, операций, ошибок) для диагностики.
*
* ДОБАВЛЕНО (2026-10, выборы лидера SAGA через Raft):
* - SAGA-оркестратор теперь создаётся через
* storage.NewSagaOrchestratorWithRaft и получает bridge из
* raftCoordinator (raftCoordinator.GetSagaLeaderBridge()).
* Это позволяет в кластерном режиме выбирать SAGA-лидера через Raft
* и выполнять компенсации только на одном узле.
* - В single-node режиме bridge также возвращает корректные значения
* (IsLeader() == true), так что поведение не меняется.
* - Раньше оркестратор создавался через NewSagaOrchestratorWithConfig
* с raftAccessor == nil, что в кластерном режиме приводило к тому,
* что каждый узел считал себя SAGA-лидером. Идемпотентность шагов
* через saga_executed_keys.json защищала от дублей операций, но не
* от одновременной компенсации одной SAGA с двух узлов.
* - Порядок shutdown безопасен: sagaOrchestrator.Stop() вызывается
* раньше raftCoordinator.Stop(), поэтому bridge ещё жив, когда
* оркестратор отписывается от него.
* ДОБАВЛЕНО (2026-10, динамическая конфигурация):
* После создания REPL передаём ему ссылку на DynamicConfigManager
* (через SetDynamicConfigManager), чтобы команды `config set`,
* `compression config set`, `cluster pipeline on/off` могли
* применять изменения через Raft и сохранять их на всех узлах.
* В fallback-режиме (Raft недоступен или узел не лидер) REPL сам
* запишет изменения в config.toml через config.SaveConfig.
*/
package main
@@ -146,13 +101,6 @@ func main() {
store := storage.NewStorage(cfg.Storage.PageSizeMB, logger)
storage.SetGlobalStorage(store)
// ========== BATCH MANAGER (замена TransactionManager) ==========
//
// ИСПРАВЛЕНО (2026-10): InitTransactionManager заменён на InitBatchManager.
// InitBatchManager инициализирует:
// - WAL (сегментированный журнал для batch-операций);
// - AOF (append-only file для мутаций вне batch).
// Директория AOF создаётся автоматически как filepath.Join(filepath.Dir(walPath), "aof").
walPath := filepath.Join(dataDir, "futriis.wal")
if err := storage.InitBatchManager(walPath); err != nil {
logger.Warn("Failed to initialize batch manager: " + err.Error())
@@ -166,8 +114,6 @@ func main() {
storage.InitTriggerManager(logger)
logger.Info("Trigger manager initialized")
// ИСПРАВЛЕНО: NewACLManager теперь возвращает (*ACLManager, error).
// Обрабатываем ошибку управляемо (без panic) и завершаемся с кодом 1.
aclManager, err := acl.NewACLManager()
if err != nil {
logger.Error("Failed to initialize ACL manager: " + err.Error())
@@ -184,27 +130,11 @@ func main() {
}
// ========== SAGA ORCHESTRATOR ==========
//
// ДОБАВЛЕНО (2026-10): используем NewSagaOrchestratorWithRaft и
// передаём bridge из raftCoordinator. Это позволяет в кластерном
// режиме выбирать SAGA-лидера через Raft, чтобы компенсации
// выполнялись только на одном узле.
//
// В single-node режиме bridge возвращает IsLeader() == true, так что
// поведение не меняется. В кластерном режиме только лидер Raft
// будет IsSagaLeader() == true, и только он будет выполнять SAGA.
//
// Если raftCoordinator == nil (не должно случаться — выше os.Exit(1)
// при ошибке), создаём оркестратор без Raft (single-node fallback).
var sagaOrchestrator *storage.SagaOrchestrator
if cfg.Saga.Enabled {
sagaConfig := &cfg.Saga
nodeID := cfg.Cluster.NodeIP + ":" + fmt.Sprint(cfg.Cluster.NodePort)
// Передаём bridge из Raft в качестве storage.SagaRaftAccessor.
// Тип bridge реализует интерфейс storage.SagaRaftAccessor
// (IsSagaLeader, GetSagaLeaderID, GetSagaCurrentTerm,
// RegisterSagaLeaderObserver).
var raftAccessor storage.SagaRaftAccessor
if raftCoordinator != nil {
raftAccessor = raftCoordinator.GetSagaLeaderBridge()
@@ -311,8 +241,6 @@ func main() {
}
// ========== HTTP API SERVER ==========
// ДОБАВЛЕНО: используем NewHTTPServerWithTLS и передаём cfg.Security.
// Если в config.toml [security].enable_tls = true, сервер поднимется по HTTPS.
httpPort := cfg.API.Port
httpServer := api.NewHTTPServerWithTLS(httpPort, store, raftCoordinator, aclManager, logger, &cfg.Security)
if metricsCollector != nil {
@@ -362,10 +290,6 @@ func main() {
logger.Info(" - Automatic data persistence - ENABLED")
}
// ИСПРАВЛЕНО (UI): убраны лишние пустые строки перед баннером.
// Внутри displayBanner уже есть utils.Println("") первой строкой,
// чтобы отделить баннер от ACL-сообщения. Строка "futriis 3i²(by
// 29.09.2026)" идёт сразу после сообщения ACL.
displayBanner(cfg.Cluster.Name, httpPort, raftCoordinator, cfg.Saga.Enabled, cfg.Metrics.Enabled)
replInstance, err := repl.NewRepl(store, raftCoordinator, logger, cfg, aclManager, pluginManager)
@@ -376,6 +300,19 @@ func main() {
}
defer replInstance.Close()
// ДОБАВЛЕНО (2026-10): передаём REPL ссылку на DynamicConfigManager,
// чтобы команды `config set`, `compression config set`,
// `cluster pipeline on/off` могли применять изменения через Raft.
// В fallback-режиме REPL пишет config.toml через config.SaveConfig.
if raftCoordinator != nil {
replInstance.SetDynamicConfigManager(
raftCoordinator.GetDynamicConfigManager(),
configPath,
)
} else {
replInstance.SetDynamicConfigManager(nil, configPath)
}
// ========== GRACEFUL SHUTDOWN ==========
sigChan := make(chan os.Signal, 1)
signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM, syscall.SIGUSR1)
@@ -431,10 +368,6 @@ func main() {
}
}
// ВАЖНО: sagaOrchestrator останавливается ДО raftCoordinator.
// Это гарантирует, что оркестратор отпишется от sagaLeaderBridge
// (и остановит leaderElector) до того, как bridge будет остановлен
// вместе с координатором. Двойной остановки bridge не будет.
if sagaOrchestrator != nil {
logger.Info("Stopping SAGA orchestrator...")
sagaStopDone := make(chan struct{})
@@ -497,8 +430,6 @@ func main() {
logger.Warn("Node shutdown timeout")
}
// ИСПРАВЛЕНО (2026-10): StopTransactionManager заменён на StopBatchManager.
// StopBatchManager синхронизирует WAL, закрывает его и закрывает AOF.
logger.Info("Stopping batch manager (WAL + AOF)...")
if stats := storage.GetBatchStats(); stats != nil {
logger.Info(fmt.Sprintf(" Batch stats: total_batches=%v, total_operations=%v, total_failed=%v, active_batches=%v",
@@ -515,7 +446,6 @@ func main() {
}()
select {
case <-batchStopDone:
// уже залогировано внутри горутины
case <-time.After(10 * time.Second):
logger.Warn("Batch manager shutdown timeout")
}
@@ -679,15 +609,6 @@ func isPortListening(port int) bool {
return true
}
// displayBanner выводит стартовый баннер.
//
// ИСПРАВЛЕНО (UI): текст баннера окрашивается точным цветом
// #00bfff (Deep Sky Blue) через utils.PrintlnDeepSkyBlueReset, чтобы
// совпадать с цветом надписи "futriis 3i²(by 29.09.2026)".
//
// Внутри уже есть utils.Println("") первой строкой, чтобы отделить
// баннер от ACL-сообщения. Дополнительные пустые строки в main() не
// добавляются (см. комментарий перед вызовом displayBanner).
func displayBanner(clusterName string, httpPort int, coordinator *cluster.RaftCoordinator, sagaEnabled, metricsEnabled bool) {
utils.Println("")
bannerLines := []string{
@@ -738,8 +659,6 @@ func displayBanner(clusterName string, httpPort int, coordinator *cluster.RaftCo
}...)
for _, line := range bannerLines {
// ИСПРАВЛЕНО: используем точный цвет #00bfff (Deep Sky Blue),
// совпадающий с цветом заголовка баннера.
utils.PrintlnDeepSkyBlueReset(line)
}
}
+25 -3
View File
@@ -58,6 +58,11 @@
// Все публичные методы *cluster.SagaManager (GetOrchestrator,
// GetMetrics, GetActiveSagas, Stop) сохранены — они используются
// в repl.go и main.go.
//
// ДОБАВЛЕНО (2026-10, кластерный SagaManager):
// - Метод GetAllSagas() у *SagaManager — возвращает все SAGA
// транзакции (активные и завершённые). Используется REPL-командой
// "saga list --all". Возвращает локальный тип cluster.SagaTransaction.
package cluster
@@ -828,7 +833,6 @@ func (rsm *RangeShardManager) Rebalance() error {
return nil
}
// =============================================================================
// INMEM STORE ДЛЯ RAFT (реализует raft.LogStore и raft.StableStore)
// =============================================================================
@@ -1551,6 +1555,26 @@ func (sm *SagaManager) GetActiveSagas() []*SagaTransaction {
return result
}
// GetAllSagas возвращает все SAGA транзакции — и активные, и завершённые.
//
// ДОБАВЛЕНО (2026-10): используется REPL-командой "saga list --all".
// В кластерной версии SAGA-менеджер хранит все транзакции в sync.Map
// (sm.sagas). Реплики и завершённые SAGA не удаляются — sync.Map
// накапливает их в течение жизни процесса. Это позволяет отдавать
// полный список без обращения к диску.
func (sm *SagaManager) GetAllSagas() []*SagaTransaction {
result := make([]*SagaTransaction, 0)
sm.sagas.Range(func(key, value interface{}) bool {
saga, ok := value.(*SagaTransaction)
if !ok || saga == nil {
return true
}
result = append(result, saga)
return true
})
return result
}
// Stop останавливает Saga-менеджер.
// Реализован, чтобы main.go мог вызвать sagaManager.Stop() единообразно
// как для storage.SagaOrchestrator, так и для cluster.SagaManager.
@@ -1963,7 +1987,6 @@ func (sbd *SplitBrainDetector) IsQuarantined(nodeID string) bool {
}
return false
}
// =============================================================================
// RECOVERY MANAGER - ПЕРЕИМЕНОВАН В ClusterNodeState ДЛЯ ИЗБЕЖАНИЯ КОНФЛИКТА
// =============================================================================
@@ -3134,7 +3157,6 @@ func (lfm *LeaderFallbackManager) Stop() {
close(lfm.stopChan)
lfm.wg.Wait()
}
// =============================================================================
// RAFT CLUSTER STATE
// =============================================================================
+305 -113
View File
@@ -10,10 +10,44 @@
// Файл: internal/compression/compression.go
// Назначение: Реализация сжатия данных с использованием различных алгоритмов.
// Поддерживаемый алгоритм: Brotli.
// Поддерживаемые алгоритмы: Snappy, Brotli, Zstd.
// Обеспечивает прозрачное сжатие/распаковку для документов.
// LZ4 был удалён в пользу Brotli для лучшего версионирования.
// Защита от collision magic bytes через усиленную проверку заголовка
// Защита от collision magic bytes через усиленную проверку заголовка.
//
// ИСПРАВЛЕНО:
// 1. Compress больше не паникует при config == nil — возвращает data
// как есть (это логично: без конфигурации сжатие выключено).
// 2. Compress учитывает оверхед заголовка при решении «сжимать или нет».
// Раньше проверялось `len(compressed) >= len(data)`, но не учитывалось,
// что заголовок (13 или 14 байт) может сделать итог больше исходных
// данных. Теперь сравнение: `len(compressed) + headerSize >= len(data)`.
// 3. Level для brotli теперь применяется (brotli.NewWriterLevel), а не
// игнорируется. Диапазон уровней brotli — 0..11, поэтому входной
// Level (1..9) нормализуется.
// 4. Введены константы MagicV1HeaderSize и MagicV2HeaderSize. Раньше
// в decompressV1 стояло магическое число 13, а MinCompressedSize = 14
// относилось к V2 — легко было перепутать.
// 5. zstd-энкодер и декодер кэшируются (sync.Map). zstd.Encoder.EncodeAll
// и zstd.Decoder.DecodeAll потокобезопасны, поэтому кэш безопасен.
// Раньше энкодер/декодер создавались на каждый вызов Compress /
// doDecompress — это дорого на горячем пути.
// 6. Логика парсинга заголовка (V1/V2) вынесена в parseHeader —
// убрано дублирование между Decompress, DecompressAuto, IsCompressed,
// GetCompressionType.
// 7. В doDecompress для CompressionNone добавлена проверка размера
// (защита в глубину от malformed-входа).
// 8. bytes.Equal(data[0:N], magic) заменено на bytes.HasPrefix —
// идиоматичнее и читается лучше.
// 9. Лимит originalSize вынесен в константу MaxOriginalSize.
// 10. Обновлён комментарий про LZ4: LZ4 удалён, остались snappy/brotli/zstd.
//
// ИСПРАВЛЕНО (компиляция):
// В Compress, в ветке case "zstd", переменная err была объявлена,
// но не использовалась: zstd.Encoder.EncodeAll не возвращает ошибку.
// Убрано объявление `encoder, err := ...` — теперь `encoder, e := ...`
// с локальной переменной e, которая проверяется тут же. Внешняя err
// по-прежнему используется в других ветках (snappy/brotli) и в
// финальной проверке оверхеда заголовка.
package compression
@@ -21,13 +55,18 @@ import (
"bytes"
"encoding/binary"
"fmt"
"sync"
"github.com/andybalholm/brotli"
"github.com/golang/snappy"
"github.com/klauspost/compress/zstd"
"github.com/andybalholm/brotli"
)
// Config представляет конфигурацию сжатия
// =============================================================================
// КОНФИГУРАЦИЯ
// =============================================================================
// Config представляет конфигурацию сжатия.
type Config struct {
Enabled bool // Включено ли сжатие
Algorithm string // Алгоритм сжатия: snappy, brotli, zstd
@@ -35,38 +74,183 @@ type Config struct {
MinSize int // Минимальный размер для сжатия (байт)
}
// MagicNumber используется для идентификации сжатых данных
// =============================================================================
// MAGIC NUMBERS И КОНСТАНТЫ ФОРМАТА
// =============================================================================
// MagicNumber используется для идентификации сжатых данных (V1).
var MagicNumber = []byte{0x46, 0x54, 0x52, 0x53} // "FTRS" - Futriis
// Расширенный magic с версией для предотвращения collision
// MagicNumberV2 — расширенный magic с версией для предотвращения collision.
var MagicNumberV2 = []byte{0x46, 0x54, 0x52, 0x53, 0x02} // "FTRS\x02"
// MinCompressedSize — минимальный размер валидного сжатого блока
// Для защиты от collision с пользовательскими данными
const MinCompressedSize = 14 // 5 (magic v2) + 1 (type) + 8 (size)
const (
// MagicV1HeaderSize — размер заголовка формата V1:
// 4 (magic) + 1 (type) + 8 (size) = 13 байт.
MagicV1HeaderSize = 4 + 1 + 8
// CompressionType определяет тип сжатия
// MagicV2HeaderSize — размер заголовка формата V2:
// 5 (magic) + 1 (type) + 8 (size) = 14 байт.
MagicV2HeaderSize = 5 + 1 + 8
// MinCompressedSize — минимальный размер валидного сжатого блока.
// Равен размеру заголовка V2 (самый свежий формат).
// Оставлен экспортируемым для обратной совместимости с внешним кодом,
// который мог его использовать.
MinCompressedSize = MagicV2HeaderSize
// MaxOriginalSize — верхняя граница originalSize, которую мы считаем
// разумной. 100 GiB. Защита от мусорных/враждебных заголовков.
MaxOriginalSize = uint64(100) * 1024 * 1024 * 1024
)
// CompressionType определяет тип сжатия.
type CompressionType byte
const (
CompressionNone CompressionType = 0x00
CompressionSnappy CompressionType = 0x01
CompressionBrotli CompressionType = 0x02
CompressionZstd CompressionType = 0x03
CompressionNone CompressionType = 0x00
CompressionSnappy CompressionType = 0x01
CompressionBrotli CompressionType = 0x02
CompressionZstd CompressionType = 0x03
)
// Compress сжимает данные с использованием указанного алгоритма
// =============================================================================
// КЭШ ZSTD ENCODER / DECODER
// =============================================================================
//
// zstd.Encoder.EncodeAll и zstd.Decoder.DecodeAll потокобезопасны
// (см. документацию klauspost/compress/zstd). Поэтому один общий
// encoder/decoder можно безопасно переиспользовать из многих горутин.
//
// Кэшируем по уровню для encoder и один общий для decoder.
var (
zstdEncoders sync.Map // map[zstd.EncoderLevel]*zstd.Encoder
zstdDecoder *zstd.Decoder
zstdDecOnce sync.Once
zstdDecErr error
)
// getZstdEncoder возвращает кэшированный zstd-энкодер для уровня.
func getZstdEncoder(level zstd.EncoderLevel) (*zstd.Encoder, error) {
if enc, ok := zstdEncoders.Load(level); ok {
return enc.(*zstd.Encoder), nil
}
enc, err := zstd.NewWriter(nil, zstd.WithEncoderLevel(level))
if err != nil {
return nil, fmt.Errorf("failed to create zstd encoder: %v", err)
}
actual, _ := zstdEncoders.LoadOrStore(level, enc)
// Если другая горутина успела первой — закрываем свой encoder.
if actual.(*zstd.Encoder) != enc {
enc.Close()
}
return actual.(*zstd.Encoder), nil
}
// getZstdDecoder возвращает общий zstd-декодер (создаётся один раз).
func getZstdDecoder() (*zstd.Decoder, error) {
zstdDecOnce.Do(func() {
zstdDecoder, zstdDecErr = zstd.NewReader(nil)
})
return zstdDecoder, zstdDecErr
}
// =============================================================================
// ПАРСИНГ ЗАГОЛОВКА
// =============================================================================
// headerInfo описывает разобранный заголовок сжатого блока.
type headerInfo struct {
version int
compType CompressionType
originalSize uint64
headerSize int
}
// parseHeader пытается разобрать заголовок сжатого блока (V2 или V1).
// Возвращает nil, если данные не являются сжатым блоком.
//
// ВНИМАНИЕ: для V1 проверяется валидность compType — это защита от
// ложных срабатываний, когда несжатые данные случайно начинаются
// с байт "FTRS".
func parseHeader(data []byte) *headerInfo {
// V2: 5-байтовый magic + 1 байт type + 8 байт size.
if len(data) >= MagicV2HeaderSize && bytes.HasPrefix(data, MagicNumberV2) {
compType := CompressionType(data[5])
if isValidCompressionType(compType) {
return &headerInfo{
version: 2,
compType: compType,
originalSize: binary.LittleEndian.Uint64(data[6:14]),
headerSize: MagicV2HeaderSize,
}
}
// Magic совпал, но тип невалиден — не считаем это сжатым блоком.
return nil
}
// V1: 4-байтовый magic + 1 байт type + 8 байт size.
if len(data) >= MagicV1HeaderSize && bytes.HasPrefix(data, MagicNumber) {
compType := CompressionType(data[4])
if isValidCompressionType(compType) {
return &headerInfo{
version: 1,
compType: compType,
originalSize: binary.LittleEndian.Uint64(data[5:13]),
headerSize: MagicV1HeaderSize,
}
}
return nil
}
return nil
}
// isValidCompressionType проверяет, что байт соответствует известному
// типу сжатия. Используется как защита от ложных срабатываний magic.
func isValidCompressionType(t CompressionType) bool {
switch t {
case CompressionNone, CompressionSnappy, CompressionBrotli, CompressionZstd:
return true
default:
return false
}
}
// =============================================================================
// СЖАТИЕ
// =============================================================================
// Compress сжимает данные с использованием указанного алгоритма.
//
// ИСПРАВЛЕНО:
// - config == nil больше не приводит к панике — возвращаем data как есть.
// - Учитывается оверхед заголовка при решении «сжимать или нет».
// - Level применяется не только к zstd, но и к brotli.
//
// ИСПРАВЛЕНО (компиляция):
// В ветке case "zstd" убрано объявление err, которое не использовалось:
// zstd.Encoder.EncodeAll не возвращает ошибку. Ошибка возможна только
// при создании encoder через getZstdEncoder — она проверяется тут же.
func Compress(data []byte, config *Config) ([]byte, error) {
if !config.Enabled {
// config == nil трактуем как «сжатие выключено».
if config == nil || !config.Enabled {
return data, nil
}
if len(data) < config.MinSize {
// MinSize <= 0 означает «сжимать всё, включая пустые данные».
// Если MinSize > 0 и данные меньше — не сжимаем.
if config.MinSize > 0 && len(data) < config.MinSize {
return data, nil
}
// Пустые данные сжимать бессмысленно.
if len(data) == 0 {
return data, nil
}
var compressed []byte
var err error
var compType CompressionType
switch config.Algorithm {
@@ -75,10 +259,19 @@ func Compress(data []byte, config *Config) ([]byte, error) {
compType = CompressionSnappy
case "brotli":
// brotli использует уровни 0..11. Входной Level — 1..9
// (см. Config.Level). Нормализуем: <=1 -> 1, >=9 -> 11 (best).
// Это даёт предсказуемое поведение: пользователь ставит 9 —
// получает максимальное сжатие.
level := normalizeBrotliLevel(config.Level)
buf := bytes.NewBuffer(nil)
writer := brotli.NewWriter(buf)
writer := brotli.NewWriterLevel(buf, level)
if _, err := writer.Write(data); err != nil {
// writer.Close() стоит вызвать даже при ошибке Write,
// чтобы освободить ресурсы, но ошибку Close игнорируем —
// основная уже есть.
_ = writer.Close()
return nil, fmt.Errorf("brotli write failed: %v", err)
}
if err := writer.Close(); err != nil {
@@ -88,9 +281,7 @@ func Compress(data []byte, config *Config) ([]byte, error) {
compType = CompressionBrotli
case "zstd":
var encoder *zstd.Encoder
var encoderLevel zstd.EncoderLevel
switch {
case config.Level <= 1:
encoderLevel = zstd.SpeedFastest
@@ -102,12 +293,15 @@ func Compress(data []byte, config *Config) ([]byte, error) {
encoderLevel = zstd.SpeedBestCompression
}
encoder, err = zstd.NewWriter(nil, zstd.WithEncoderLevel(encoderLevel))
if err != nil {
return nil, fmt.Errorf("failed to create zstd encoder: %v", err)
// ИСПРАВЛЕНО: вместо `encoder, err := ...` используем локальную e,
// чтобы не объявлять неиспользуемую err. getZstdEncoder возвращает
// (*zstd.Encoder, error), и эта ошибка проверяется тут же.
encoder, e := getZstdEncoder(encoderLevel)
if e != nil {
return nil, e
}
defer encoder.Close()
// EncodeAll потокобезопасен, используем общий encoder.
compressed = encoder.EncodeAll(data, nil)
compType = CompressionZstd
@@ -115,16 +309,18 @@ func Compress(data []byte, config *Config) ([]byte, error) {
return nil, fmt.Errorf("unsupported compression algorithm: %s", config.Algorithm)
}
// Проверяем, что сжатие действительно уменьшило размер
if len(compressed) >= len(data) {
// ИСПРАВЛЕНО: учитываем оверхед заголовка V2.
// Если сжатые данные + заголовок не меньше исходных —
// сжатие не имеет смысла, возвращаем исходные данные.
if len(compressed)+MagicV2HeaderSize >= len(data) {
return data, nil
}
// Используем MagicNumberV2 (5 байт) + type (1) + size (8) = 14 байт
header := make([]byte, 5+1+8)
// Формируем заголовок V2: 5 (magic) + 1 (type) + 8 (size).
header := make([]byte, MagicV2HeaderSize)
copy(header[0:5], MagicNumberV2)
header[5] = byte(compType)
binary.LittleEndian.PutUint64(header[6:], uint64(len(data)))
binary.LittleEndian.PutUint64(header[6:14], uint64(len(data)))
result := make([]byte, 0, len(header)+len(compressed))
result = append(result, header...)
@@ -133,63 +329,61 @@ func Compress(data []byte, config *Config) ([]byte, error) {
return result, nil
}
// Decompress распаковывает данные
// Поддержка обоих форматов (V1 и V2), усиленная валидация
// normalizeBrotliLevel приводит входной уровень (1..9) к диапазону
// brotli (0..11). Значения <=0 и >9 трактуются разумно:
// - level <= 0 -> brotli.DefaultCompression
// - level 1..9 -> линейное отображение: 1->1, 9->11
// - level > 9 -> 11 (максимум)
func normalizeBrotliLevel(level int) int {
if level <= 0 {
return brotli.DefaultCompression
}
if level >= 9 {
return 11
}
// Линейное отображение 1..9 -> 1..11.
// 1->1, 2->2, ..., 9->11. Формула: 1 + (level-1)*10/8.
return 1 + (level-1)*10/8
}
// =============================================================================
// РАСПАКОВКА
// =============================================================================
// Decompress распаковывает данные.
// Поддерживает оба формата (V1 и V2), усиленная валидация.
func Decompress(data []byte) ([]byte, error) {
// Проверяем V2 формат (5-байтовый magic)
if len(data) >= MinCompressedSize && bytes.Equal(data[0:5], MagicNumberV2) {
return decompressV2(data)
info := parseHeader(data)
if info == nil {
return nil, fmt.Errorf("invalid magic number")
}
// Проверяем V1 формат (4-байтовый magic) для обратной совместимости
if len(data) >= 4+1+8 && bytes.Equal(data[0:4], MagicNumber) {
return decompressV1(data)
}
return nil, fmt.Errorf("invalid magic number")
return decompressWithInfo(data, info)
}
// decompressV1 распаковывает данные в старом формате V1
func decompressV1(data []byte) ([]byte, error) {
if len(data) < 4+1+8 {
return nil, fmt.Errorf("data too short for compressed format V1")
// decompressWithInfo выполняет распаковку, используя разобранный заголовок.
func decompressWithInfo(data []byte, info *headerInfo) ([]byte, error) {
if len(data) < info.headerSize {
return nil, fmt.Errorf("data too short for compressed format V%d", info.version)
}
compType := CompressionType(data[4])
originalSize := binary.LittleEndian.Uint64(data[5:13])
compressedData := data[13:]
if originalSize == 0 {
return nil, fmt.Errorf("invalid original size")
}
return doDecompress(compType, compressedData, originalSize)
}
// decompressV2 распаковывает данные в новом формате V2
// Добавлена проверка согласованности размера
func decompressV2(data []byte) ([]byte, error) {
if len(data) < MinCompressedSize {
return nil, fmt.Errorf("data too short for compressed format V2")
}
compType := CompressionType(data[5])
originalSize := binary.LittleEndian.Uint64(data[6:14])
compressedData := data[14:]
// Проверка разумности размера
if originalSize == 0 || originalSize > 100*1024*1024*1024 { // Max 100GB
return nil, fmt.Errorf("invalid original size: %d", originalSize)
// Проверка разумности размера.
if info.originalSize == 0 || info.originalSize > MaxOriginalSize {
return nil, fmt.Errorf("invalid original size: %d", info.originalSize)
}
compressedData := data[info.headerSize:]
if len(compressedData) == 0 {
return nil, fmt.Errorf("no compressed data")
}
return doDecompress(compType, compressedData, originalSize)
return doDecompress(info.compType, compressedData, info.originalSize)
}
// doDecompress выполняет распаковку по типу
// doDecompress выполняет распаковку по типу.
//
// ИСПРАВЛЕНО:
// - для CompressionNone добавлена проверка размера (защита в глубину);
// - zstd-декодер берётся из кэша, а не создаётся на каждый вызов.
func doDecompress(compType CompressionType, compressedData []byte, originalSize uint64) ([]byte, error) {
var decompressed []byte
var err error
@@ -211,25 +405,32 @@ func doDecompress(compType CompressionType, compressedData []byte, originalSize
decompressed = buf.Bytes()
case CompressionZstd:
decoder, err := zstd.NewReader(nil)
decoder, err := getZstdDecoder()
if err != nil {
return nil, fmt.Errorf("failed to create zstd decoder: %v", err)
return nil, err
}
defer decoder.Close()
// DecodeAll потокобезопасен, используем общий decoder.
decompressed, err = decoder.DecodeAll(compressedData, nil)
if err != nil {
return nil, fmt.Errorf("zstd decode failed: %v", err)
}
case CompressionNone:
// ИСПРАВЛЕНО: проверяем согласованность размера даже для None.
// Это защита в глубину от malformed-входа. Сейчас Compress не
// пишет CompressionNone в заголовок, но на входе может оказаться
// подделанный блок.
if uint64(len(compressedData)) != originalSize {
return nil, fmt.Errorf("size mismatch for uncompressed: expected %d, got %d",
originalSize, len(compressedData))
}
return compressedData, nil
default:
return nil, fmt.Errorf("unsupported compression type: %d", compType)
}
// Проверяем, что распакованный размер совпадает с заявленным
// Проверяем, что распакованный размер совпадает с заявленным.
if uint64(len(decompressed)) != originalSize {
return nil, fmt.Errorf("size mismatch: expected %d, got %d", originalSize, len(decompressed))
}
@@ -237,52 +438,43 @@ func doDecompress(compType CompressionType, compressedData []byte, originalSize
return decompressed, nil
}
// DecompressAuto автоматически определяет, сжаты ли данные, и распаковывает при необходимости
// Усиленная проверка для предотвращения collision
// DecompressAuto автоматически определяет, сжаты ли данные, и распаковывает при необходимости.
// Усиленная проверка для предотвращения collision.
func DecompressAuto(data []byte) ([]byte, error) {
// Проверяем V2 magic (более надёжный)
if len(data) >= MinCompressedSize && bytes.Equal(data[0:5], MagicNumberV2) {
return Decompress(data)
info := parseHeader(data)
if info == nil {
// Данные не сжаты.
return data, nil
}
// Проверяем V1 magic (для обратной совместимости)
if len(data) >= 4+1+8 && bytes.Equal(data[0:4], MagicNumber) {
// Дополнительная проверка: тип сжатия должен быть валидным
compType := CompressionType(data[4])
if compType <= CompressionZstd {
return Decompress(data)
}
}
// Данные не сжаты
return data, nil
return decompressWithInfo(data, info)
}
// IsCompressed проверяет, сжаты ли данные
// Проверка обоих форматов
// IsCompressed проверяет, сжаты ли данные (V1 или V2).
func IsCompressed(data []byte) bool {
if len(data) >= MinCompressedSize && bytes.Equal(data[0:5], MagicNumberV2) {
return true
}
if len(data) >= 4+1+8 && bytes.Equal(data[0:4], MagicNumber) {
compType := CompressionType(data[4])
return compType <= CompressionZstd
}
return false
return parseHeader(data) != nil
}
// GetCompressionType возвращает тип сжатия данных
// GetCompressionType возвращает тип сжатия данных.
// Возвращает CompressionNone, если данные не сжаты.
func GetCompressionType(data []byte) CompressionType {
if len(data) >= MinCompressedSize && bytes.Equal(data[0:5], MagicNumberV2) {
return CompressionType(data[5])
}
if !IsCompressed(data) || len(data) < 5 {
info := parseHeader(data)
if info == nil {
return CompressionNone
}
return CompressionType(data[4])
return info.compType
}
// GetCompressionRatio возвращает коэффициент сжатия
// =============================================================================
// СТАТИСТИКА
// =============================================================================
// GetCompressionRatio возвращает коэффициент сжатия как долю:
// - 1.0 — данные не уменьшились (или выросли);
// - 0.5 — сжатые данные в два раза меньше исходных;
// - 0.0 — теоретический предел (пустые данные после сжатия).
//
// ВНИМАНИЕ: это именно доля compressed/original, а не «во сколько раз
// сжали». Если нужно «во сколько раз», используйте 1.0 / ratio.
func GetCompressionRatio(original, compressed []byte) float64 {
if len(original) == 0 {
return 1.0
+265 -429
View File
File diff suppressed because it is too large. Load diff
+85 -62
View File
@@ -9,7 +9,7 @@
*/
// Файл: internal/config/dynamic_config.go
// Назначение: Централизованное управление конфигурацией через Raft
// Назначение: Централизованное управление конфигурацией через Raft.
package config
@@ -32,8 +32,8 @@ import (
type atomicBool struct{ v int32 }
func (b *atomicBool) Load() bool { return atomic.LoadInt32(&b.v) != 0 }
func (b *atomicBool) Store(v bool) { atomic.StoreInt32(&b.v, boolToInt32(v)) }
func (b *atomicBool) Load() bool { return atomic.LoadInt32(&b.v) != 0 }
func (b *atomicBool) Store(v bool) { atomic.StoreInt32(&b.v, boolToInt32(v)) }
func (b *atomicBool) CAS(old, new bool) bool {
return atomic.CompareAndSwapInt32(&b.v, boolToInt32(old), boolToInt32(new))
}
@@ -55,7 +55,7 @@ func (a *atomicUint64) Add(d uint64) uint64 { return atomic.AddUint64(&a.v, d) }
// ТИПЫ ДЛЯ ДИНАМИЧЕСКОЙ КОНФИГУРАЦИИ
// =============================================================================
// DynamicConfigChange представляет изменение конфигурации
// DynamicConfigChange представляет изменение конфигурации.
type DynamicConfigChange struct {
ID string `json:"id"`
Version uint64 `json:"version"`
@@ -65,22 +65,22 @@ type DynamicConfigChange struct {
Changes map[string]interface{} `json:"changes"`
}
// DynamicConfigSnapshot представляет снэпшот конфигурации
// DynamicConfigSnapshot представляет снэпшот конфигурации.
type DynamicConfigSnapshot struct {
Version uint64 `json:"version"`
UpdatedAt int64 `json:"updated_at"`
Config *Config `json:"config"`
ChangeHistory []*DynamicConfigChange `json:"change_history"`
Version uint64 `json:"version"`
UpdatedAt int64 `json:"updated_at"`
Config *Config `json:"config"`
ChangeHistory []*DynamicConfigChange `json:"change_history"`
}
// ConfigChangeListener представляет слушатель изменений конфигурации
// ConfigChangeListener представляет слушатель изменений конфигурации.
type ConfigChangeListener func(change *DynamicConfigChange)
// =============================================================================
// DYNAMIC CONFIG MANAGER
// =============================================================================
// DynamicConfigManager управляет динамической конфигурацией через Raft
// DynamicConfigManager управляет динамической конфигурацией через Raft.
type DynamicConfigManager struct {
config *Config
logger *log.Logger
@@ -97,29 +97,28 @@ type DynamicConfigManager struct {
isLeader atomicBool
}
// ConfigFSM реализует конечный автомат для конфигурации
// ConfigFSM реализует конечный автомат для конфигурации.
type ConfigFSM struct {
config *DynamicConfigSnapshot
mu sync.RWMutex
logger *log.Logger
// onChange вызывается после успешного применения изменения.
// Может быть установлен менеджером для уведомления слушателей.
onChange func(change *DynamicConfigChange)
onChange func(change *DynamicConfigChange)
onChangeMu sync.RWMutex
}
// ConfigSnapshot реализует raft.FSMSnapshot
// ConfigSnapshot реализует raft.FSMSnapshot.
type ConfigSnapshot struct {
config *DynamicConfigSnapshot
}
// NewDynamicConfigManager создаёт новый менеджер динамической конфигурации
// NewDynamicConfigManager создаёт новый менеджер динамической конфигурации.
func NewDynamicConfigManager(cfg *Config, logger *log.Logger) *DynamicConfigManager {
if cfg == nil {
cfg = &Config{}
}
// Создаём начальный снэпшот
initialSnapshot := &DynamicConfigSnapshot{
Version: 1,
UpdatedAt: time.Now().UnixMilli(),
@@ -142,13 +141,11 @@ func NewDynamicConfigManager(cfg *Config, logger *log.Logger) *DynamicConfigMana
}
dm.version.Store(1)
// Подключаем FSM к менеджеру для уведомления слушателей
fsm.onChangeMu.Lock()
fsm.onChange = func(change *DynamicConfigChange) {
dm.version.Store(change.Version)
dm.mu.Lock()
dm.changeHistory = append(dm.changeHistory, change)
// Ограничиваем историю
if len(dm.changeHistory) > 1000 {
dm.changeHistory = dm.changeHistory[len(dm.changeHistory)-1000:]
}
@@ -160,19 +157,19 @@ func NewDynamicConfigManager(cfg *Config, logger *log.Logger) *DynamicConfigMana
return dm
}
// SetRaft устанавливает Raft для управления конфигурацией
// SetRaft устанавливает Raft для управления конфигурацией.
func (dm *DynamicConfigManager) SetRaft(r *raft.Raft) {
dm.mu.Lock()
defer dm.mu.Unlock()
dm.raft = r
}
// GetFSM возвращает FSM для конфигурации
// GetFSM возвращает FSM для конфигурации.
func (dm *DynamicConfigManager) GetFSM() *ConfigFSM {
return dm.fsm
}
// Start запускает менеджер динамической конфигурации
// Start запускает менеджер динамической конфигурации.
func (dm *DynamicConfigManager) Start() {
if dm.logger != nil {
dm.logger.Info("Dynamic config manager started")
@@ -203,9 +200,8 @@ func (dm *DynamicConfigManager) GetConfig() *Config {
return deepCopyConfig(dm.fsm.config.Config)
}
// GetVersion возвращает текущую версию конфигурации
// GetVersion возвращает текущую версию конфигурации.
func (dm *DynamicConfigManager) GetVersion() uint64 {
// Синхронизируем с FSM, чтобы значение было актуальным
dm.fsm.mu.RLock()
defer dm.fsm.mu.RUnlock()
if dm.fsm.config != nil {
@@ -214,7 +210,7 @@ func (dm *DynamicConfigManager) GetVersion() uint64 {
return dm.version.Load()
}
// GetChangeHistory возвращает историю изменений
// GetChangeHistory возвращает историю изменений.
func (dm *DynamicConfigManager) GetChangeHistory() []*DynamicConfigChange {
dm.mu.RLock()
defer dm.mu.RUnlock()
@@ -224,7 +220,7 @@ func (dm *DynamicConfigManager) GetChangeHistory() []*DynamicConfigChange {
return history
}
// ApplyChange применяет изменение конфигурации через Raft
// ApplyChange применяет изменение конфигурации через Raft.
func (dm *DynamicConfigManager) ApplyChange(changes map[string]interface{}, description string, changedBy string) error {
dm.mu.RLock()
r := dm.raft
@@ -234,14 +230,12 @@ func (dm *DynamicConfigManager) ApplyChange(changes map[string]interface{}, desc
return fmt.Errorf("raft not initialized")
}
// Проверяем, что мы лидер — через сам Raft, а не через внешний флаг
if r.State() != raft.Leader {
return fmt.Errorf("not the leader (current state: %s)", r.State())
}
currentVersion := dm.GetVersion()
// Создаём команду изменения
change := &DynamicConfigChange{
ID: fmt.Sprintf("config_%d_%d", currentVersion+1, time.Now().UnixNano()),
Version: currentVersion + 1,
@@ -251,7 +245,6 @@ func (dm *DynamicConfigManager) ApplyChange(changes map[string]interface{}, desc
Changes: changes,
}
// Сериализуем и применяем через Raft
data, err := json.Marshal(change)
if err != nil {
return fmt.Errorf("failed to marshal config change: %v", err)
@@ -262,7 +255,6 @@ func (dm *DynamicConfigManager) ApplyChange(changes map[string]interface{}, desc
return fmt.Errorf("raft apply failed: %v", err)
}
// Проверяем ошибку из FSM.Apply
if resp := future.Response(); resp != nil {
if err, ok := resp.(error); ok {
return fmt.Errorf("fsm apply failed: %v", err)
@@ -272,16 +264,15 @@ func (dm *DynamicConfigManager) ApplyChange(changes map[string]interface{}, desc
if dm.logger != nil {
dm.logger.Info(fmt.Sprintf("Applied config change: %s (version: %d)", description, change.Version))
}
return nil
}
// UpdateConfig обновляет конфигурацию
// UpdateConfig обновляет конфигурацию.
func (dm *DynamicConfigManager) UpdateConfig(updates map[string]interface{}, description string, changedBy string) error {
return dm.ApplyChange(updates, description, changedBy)
}
// ApplyConfigChange применяет изменение к FSM (вызывается из Raft)
// ApplyConfigChange применяет изменение к FSM (вызывается из Raft).
func (fsm *ConfigFSM) ApplyConfigChange(change *DynamicConfigChange) error {
if change == nil {
return fmt.Errorf("nil config change")
@@ -303,7 +294,6 @@ func (fsm *ConfigFSM) ApplyConfigChange(change *DynamicConfigChange) error {
config := fsm.config.Config
// Применяем изменения к конфигурации
for path, value := range change.Changes {
if err := setConfigField(config, path, value); err != nil {
fsm.mu.Unlock()
@@ -311,11 +301,9 @@ func (fsm *ConfigFSM) ApplyConfigChange(change *DynamicConfigChange) error {
}
}
// Обновляем метаданные
fsm.config.Version = change.Version
fsm.config.UpdatedAt = time.Now().UnixMilli()
fsm.config.ChangeHistory = append(fsm.config.ChangeHistory, change)
// Ограничиваем историю
if len(fsm.config.ChangeHistory) > 1000 {
fsm.config.ChangeHistory = fsm.config.ChangeHistory[len(fsm.config.ChangeHistory)-1000:]
}
@@ -326,32 +314,28 @@ func (fsm *ConfigFSM) ApplyConfigChange(change *DynamicConfigChange) error {
fsm.logger.Debug(fmt.Sprintf("Applied config change to FSM: %s", change.Description))
}
// Уведомляем слушателей (через callback, установленный менеджером)
fsm.onChangeMu.RLock()
cb := fsm.onChange
fsm.onChangeMu.RUnlock()
if cb != nil {
cb(change)
}
return nil
}
// Apply применяет команду к FSM (реализация raft.FSM)
// Apply применяет команду к FSM (реализация raft.FSM).
func (fsm *ConfigFSM) Apply(log *raft.Log) interface{} {
var change DynamicConfigChange
if err := json.Unmarshal(log.Data, &change); err != nil {
return fmt.Errorf("failed to unmarshal config change: %v", err)
}
if err := fsm.ApplyConfigChange(&change); err != nil {
return err
}
return nil
}
// Snapshot создаёт снэпшот состояния FSM
// Snapshot создаёт снэпшот состояния FSM.
func (fsm *ConfigFSM) Snapshot() (raft.FSMSnapshot, error) {
fsm.mu.RLock()
defer fsm.mu.RUnlock()
@@ -365,7 +349,6 @@ func (fsm *ConfigFSM) Snapshot() (raft.FSMSnapshot, error) {
}}, nil
}
// Глубокая копия через JSON, чтобы снэпшот не зависел от мутаций FSM
data, err := json.Marshal(fsm.config)
if err != nil {
return nil, fmt.Errorf("failed to marshal snapshot: %v", err)
@@ -384,7 +367,7 @@ func (fsm *ConfigFSM) Snapshot() (raft.FSMSnapshot, error) {
return &ConfigSnapshot{config: &snapshotData}, nil
}
// Restore восстанавливает состояние FSM из снэпшота
// Restore восстанавливает состояние FSM из снэпшота.
func (fsm *ConfigFSM) Restore(snapshot io.ReadCloser) error {
defer snapshot.Close()
@@ -407,37 +390,34 @@ func (fsm *ConfigFSM) Restore(snapshot io.ReadCloser) error {
fsm.mu.Lock()
fsm.config = &data
fsm.mu.Unlock()
return nil
}
// Persist сохраняет снэпшот
// Persist сохраняет снэпшот.
func (s *ConfigSnapshot) Persist(sink raft.SnapshotSink) error {
data, err := json.Marshal(s.config)
if err != nil {
sink.Cancel()
return err
}
if _, err := sink.Write(data); err != nil {
sink.Cancel()
return err
}
return sink.Close()
}
// Release освобождает ресурсы снэпшота
// Release освобождает ресурсы снэпшота.
func (s *ConfigSnapshot) Release() {}
// RegisterListener регистрирует слушатель изменений конфигурации
// RegisterListener регистрирует слушатель изменений конфигурации.
func (dm *DynamicConfigManager) RegisterListener(name string, listener ConfigChangeListener) {
dm.listenerMu.Lock()
defer dm.listenerMu.Unlock()
dm.listeners[name] = listener
}
// UnregisterListener удаляет слушатель
// UnregisterListener удаляет слушатель.
func (dm *DynamicConfigManager) UnregisterListener(name string) {
dm.listenerMu.Lock()
defer dm.listenerMu.Unlock()
@@ -445,8 +425,6 @@ func (dm *DynamicConfigManager) UnregisterListener(name string) {
}
// notifyListeners уведомляет слушателей об изменении.
// Каждый слушатель вызывается в отдельной горутине; паника в слушателе
// не должна ронять процесс.
func (dm *DynamicConfigManager) notifyListeners(change *DynamicConfigChange) {
dm.listenerMu.RLock()
defer dm.listenerMu.RUnlock()
@@ -468,28 +446,22 @@ func (dm *DynamicConfigManager) notifyListeners(change *DynamicConfigChange) {
// =============================================================================
// deepCopyConfig создаёт глубокую копию конфигурации через JSON.
// Если src == nil, возвращает &Config{} (не nil), чтобы избежать паник.
func deepCopyConfig(src *Config) *Config {
if src == nil {
return &Config{}
}
data, err := json.Marshal(src)
if err != nil {
// Возвращаем пустую копию, а не nil, чтобы не сломать вызывающий код
return &Config{}
}
var dst Config
if err := json.Unmarshal(data, &dst); err != nil {
return &Config{}
}
return &dst
}
// coerceInt пытается привести значение к int.
// Поддерживает int, int32, int64, uint, uint32, uint64, float32, float64, json.Number.
func coerceInt(value interface{}) (int, bool) {
switch v := value.(type) {
case int:
@@ -627,8 +599,7 @@ func coerceString(value interface{}) (string, bool) {
}
// setConfigField устанавливает значение поля конфигурации по пути.
// Поддерживает основные секции: cluster, storage, replication, wal, mvcc,
// saga, backpressure, api, log, plugins, autoscaling, monitoring, transactions.
// Используется FSM при применении изменений из Raft.
func setConfigField(config *Config, path string, value interface{}) error {
if path == "" {
return fmt.Errorf("empty config path")
@@ -1087,12 +1058,64 @@ func setConfigField(config *Config, path string, value interface{}) error {
}
return fmt.Errorf("transactions.max_savepoints_per_tx: expected int, got %T", value)
// ===== compression =====
case "compression.enabled":
if v, ok := coerceBool(value); ok {
config.Compression.Enabled = v
return nil
}
return fmt.Errorf("compression.enabled: expected bool, got %T", value)
case "compression.algorithm":
if v, ok := coerceString(value); ok {
config.Compression.Algorithm = v
return nil
}
return fmt.Errorf("compression.algorithm: expected string, got %T", value)
case "compression.level":
if v, ok := coerceInt(value); ok {
config.Compression.Level = v
return nil
}
return fmt.Errorf("compression.level: expected int, got %T", value)
case "compression.min_size":
if v, ok := coerceInt(value); ok {
config.Compression.MinSize = v
return nil
}
return fmt.Errorf("compression.min_size: expected int, got %T", value)
// ===== performance =====
case "performance.enable_pipeline":
if v, ok := coerceBool(value); ok {
config.Performance.EnablePipeline = v
return nil
}
return fmt.Errorf("performance.enable_pipeline: expected bool, got %T", value)
case "performance.batch_size":
if v, ok := coerceInt(value); ok {
config.Performance.BatchSize = v
return nil
}
return fmt.Errorf("performance.batch_size: expected int, got %T", value)
case "performance.max_connections":
if v, ok := coerceInt(value); ok {
config.Performance.MaxConnections = v
return nil
}
return fmt.Errorf("performance.max_connections: expected int, got %T", value)
case "performance.read_from_follower":
if v, ok := coerceBool(value); ok {
config.Performance.ReadFromFollower = v
return nil
}
return fmt.Errorf("performance.read_from_follower: expected bool, got %T", value)
default:
return fmt.Errorf("unknown config path: %s", path)
}
}
// GetLeaderStatus возвращает статус лидера
// GetLeaderStatus возвращает статус лидера.
func (dm *DynamicConfigManager) GetLeaderStatus() bool {
dm.mu.RLock()
r := dm.raft
@@ -1110,7 +1133,7 @@ func (dm *DynamicConfigManager) SetLeaderStatus(isLeader bool) {
dm.isLeader.Store(isLeader)
}
// GetListeners возвращает список зарегистрированных слушателей
// GetListeners возвращает список зарегистрированных слушателей.
func (dm *DynamicConfigManager) GetListeners() []string {
dm.listenerMu.RLock()
defer dm.listenerMu.RUnlock()
File diff suppressed because it is too large. Load diff
+72 -128
View File
@@ -9,27 +9,37 @@
*/
// Файл: internal/repl/pager.go
// Назначение: Пейджер для постраничного вывода длинных результатов.
// Использует внешний pager (less/more) если доступен, иначе встроенный.
// Совместимо с Linux и OpenIndiana.
// Назначение: Простой пейджер для длинных выводов REPL.
// Если количество строк превышает threshold, вывод отправляется в
// внешний пейджер (по умолчанию `less -R`), иначе печатается напрямую.
//
// ДОБАВЛЕНО (2026-10):
// - Threshold() / SetThreshold() — геттер и сеттер порога включения
// пейджера. Используются REPL-командой `pager size <N>` /
// `pager threshold <N>`. Значения <= 0 игнорируются, чтобы случайно
// не сломать вывод.
package repl
import (
"bufio"
"bytes"
"fmt"
"io"
"os"
"os/exec"
"runtime"
"strings"
)
// Pager предоставляет постраничный вывод.
// Pager — простой пейджер.
//
// Все поля читаются/пишутся только из REPL-цикла (последовательный
// доступ), поэтому мьютекс здесь не нужен. Если пейджер начнёт
// использоваться из фоновых горутин — добавьте sync.RWMutex.
type Pager struct {
enabled bool
threshold int
external string
lines int
enabled bool
threshold int
external string
lines int
}
// NewPager создаёт пейджер.
@@ -37,49 +47,40 @@ func NewPager(enabled bool, threshold int) *Pager {
p := &Pager{
enabled: enabled,
threshold: threshold,
lines: 24,
external: defaultPagerCommand(),
lines: 0,
}
p.external = detectPager()
return p
}
// detectPager ищет доступный внешний pager.
// Порядок: $PAGER -> less -> more -> "" (встроенный).
func detectPager() string {
if env := os.Getenv("PAGER"); env != "" {
return env
}
if path, err := exec.LookPath("less"); err == nil {
return path
}
if path, err := exec.LookPath("more"); err == nil {
return path
}
return ""
}
// Enabled возвращает, включён ли пейджер.
// Enabled возвращает статус пейджера.
func (p *Pager) Enabled() bool {
return p != nil && p.enabled
return p.enabled
}
// SetEnabled включает/выключает пейджер.
func (p *Pager) SetEnabled(v bool) {
if p != nil {
p.enabled = v
}
// SetEnabled включает или отключает пейджер.
func (p *Pager) SetEnabled(enabled bool) {
p.enabled = enabled
}
// SetThreshold устанавливает порог строк.
// Threshold возвращает порог включения пейджера (в строках).
// Если вывод короче порога — пейджер не используется.
func (p *Pager) Threshold() int {
return p.threshold
}
// SetThreshold устанавливает порог включения пейджера.
// Значения <= 0 игнорируются, чтобы случайно не сломать вывод.
func (p *Pager) SetThreshold(n int) {
if p != nil && n > 0 {
if n > 0 {
p.threshold = n
}
}
// Page выводит текст через пейджер, если он длиннее порога.
// Page отображает текст через пейджер, если он включён и текст достаточно
// длинный. Иначе печатает напрямую в stdout.
func (p *Pager) Page(text string) error {
if p == nil || !p.enabled {
if !p.enabled {
fmt.Print(text)
return nil
}
@@ -90,106 +91,49 @@ func (p *Pager) Page(text string) error {
return nil
}
if p.external != "" {
if err := p.pageExternal(text); err == nil {
return nil
}
// fallback на встроенный
if p.external == "" {
// Нет внешнего пейджера — печатаем напрямую.
fmt.Print(text)
return nil
}
return p.pageBuiltin(text)
return p.pipeToExternal(text)
}
// pageExternal запускает внешний pager.
func (p *Pager) pageExternal(text string) error {
args := []string{}
base := strings.ToLower(p.external)
if strings.Contains(base, "less") {
// -R: сохранять ANSI-цвета, -F: выйти если всё влезает, -X: не чистить экран
args = append(args, "-RFX")
// pipeToExternal отправляет текст во внешний пейджер.
func (p *Pager) pipeToExternal(text string) error {
parts := strings.Fields(p.external)
if len(parts) == 0 {
fmt.Print(text)
return nil
}
cmd := exec.Command(p.external, args...)
cmd := exec.Command(parts[0], parts[1:]...)
cmd.Stdin = bytes.NewBufferString(text)
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
stdin, err := cmd.StdinPipe()
if err != nil {
if err := cmd.Run(); err != nil {
// Если внешний пейджер упал — печатаем напрямую.
fmt.Print(text)
return err
}
if err := cmd.Start(); err != nil {
return err
}
if _, err := io.WriteString(stdin, text); err != nil {
stdin.Close()
_ = cmd.Wait()
return err
}
stdin.Close()
return cmd.Wait()
}
// pageBuiltin — встроенный пейджер (fallback).
// Управление: Enter — следующая строка, Space — страница, q — выход.
func (p *Pager) pageBuiltin(text string) error {
scanner := bufio.NewScanner(strings.NewReader(text))
buf := make([]byte, 0, 64*1024)
scanner.Buffer(buf, 1024*1024)
lines := []string{}
for scanner.Scan() {
lines = append(lines, scanner.Text())
}
if err := scanner.Err(); err != nil {
return err
}
pageSize := p.lines - 1
if pageSize < 5 {
pageSize = 5
}
reader := bufio.NewReader(os.Stdin)
pos := 0
for pos < len(lines) {
end := pos + pageSize
if end > len(lines) {
end = len(lines)
}
for i := pos; i < end; i++ {
fmt.Println(lines[i])
}
pos = end
if pos >= len(lines) {
break
}
fmt.Printf("\033[7m--More--\033[0m (Enter/Space/q) ")
ch, err := reader.ReadByte()
fmt.Print("\r\033[K")
if err != nil {
break
}
switch ch {
case 'q', 'Q':
return nil
case '\n', '\r':
// одна строка
if pos < len(lines) {
fmt.Println(lines[pos])
pos++
}
case ' ':
// страница
default:
// игнорируем
}
}
return nil
}
// defaultPagerCommand выбирает команду пейджера в зависимости от ОС.
func defaultPagerCommand() string {
switch runtime.GOOS {
case "windows":
return "more"
default:
// less -R: -R сохраняет ANSI-цвета, используется на Linux/BSD/illumos.
if _, err := exec.LookPath("less"); err == nil {
return "less -R"
}
if _, err := exec.LookPath("more"); err == nil {
return "more"
}
return ""
}
}
+1393 -371
View File
File diff suppressed because it is too large. Load diff
+51 -1
View File
@@ -10,18 +10,68 @@
// Файл: internal/serializer/msgpack.go
// Назначение: Сериализация и десериализация документов в формате MessagePack.
// Используется библиотека vmihailenco/msgpack для высокой производительности.
// Используется библиотека vmihailenco/msgpack/v5 для высокой производительности.
//
// MessagePack — это эффективный бинарный формат сериализации, который:
// - компактнее JSON (меньше размер);
// - быстрее при сериализации/десериализации;
// - поддерживает все основные типы данных;
// - позволяет сериализовать структуры Go с учётом тегов `msgpack`/`json`
// (поведение зависит от настроек библиотеки).
//
// Данный пакет предоставляет простую обёртку для унификации
// использования MessagePack во всём проекте.
// Package serializer предоставляет обёртки над сериализацией в формат
// MessagePack. Используется для эффективного хранения и передачи
// документов внутри СУБД futriis.
package serializer
import (
"github.com/vmihailenco/msgpack/v5"
)
// Marshal сериализует значение в формат MessagePack.
//
// Параметры:
// - v: любое значение Go (структура, карта, срез, примитив и т.д.).
//
// Возвращает:
// - []byte: сериализованные данные;
// - error: ошибка при сериализации (например, при наличии
// несериализуемых типов, таких как каналы или функции).
//
// Пример использования:
//
// data, err := serializer.Marshal(myStruct)
// if err != nil {
// return err
// }
// // data содержит бинарное представление myStruct
func Marshal(v interface{}) ([]byte, error) {
return msgpack.Marshal(v)
}
// Unmarshal десериализует данные из формата MessagePack в значение.
//
// Параметры:
// - data: бинарные данные в формате MessagePack;
// - v: указатель на значение, в которое нужно десериализовать данные.
//
// Возвращает:
// - error: ошибка при десериализации (например, если данные повреждены
// или не соответствуют типу v).
//
// Важно: v должен быть указателем (!), иначе функция вернёт ошибку.
//
// Пример использования:
//
// var myStruct MyStruct
// err := serializer.Unmarshal(data, &myStruct)
// if err != nil {
// return err
// }
// // myStruct содержит восстановленные данные
func Unmarshal(data []byte, v interface{}) error {
return msgpack.Unmarshal(data, v)
}
-481
View File
@@ -1,481 +0,0 @@
/*
* Copyright 2026 Safronov Grigorii
*
* Licensed under the CDDL, Version 1.0 (the "License");
* you may not use this file except in compliance with the License.
*
* You may obtain a copy of the License at
* https://opensource.org/licenses/CDDL-1.0
*/
// Файл: internal/storage/aof.go
// Назначение: Append-Only File (AOF) для мутаций, выполненных вне batch.
//
// ЗАЧЕМ НУЖЕН AOF:
// WAL покрывает только batch-операции (групповые мутации).
// Прямые вызовы coll.Insert / coll.Update / coll.Delete идут мимо WAL.
// AOF логирует каждую отдельную мутацию и позволяет:
// 1. Восстановить состояние после краха без потери данных.
// 2. Вести полный audit trail.
// 3. Использовать для репликации.
//
// ФОРМАТ:
// Каждая запись — JSON-строка, заканчивающаяся '\n'.
// Поля: type, database, collection, document_id, data, timestamp, crc.
// Файлы разбиваются по размеру (по умолчанию 64 МБ).
// При старте AOFManager сканирует директорию и загружает все записи.
package storage
import (
"bufio"
"encoding/json"
"fmt"
"os"
"path/filepath"
"sort"
"strings"
"sync"
"time"
)
// =============================================================================
// КОНСТАНТЫ
// =============================================================================
const (
// Максимальный размер одного AOF-файла.
AOFMaxFileSize = 64 * 1024 * 1024
// Префикс и суффикс AOF-файлов.
AOFFilePrefix = "aof_"
AOFFileSuffix = ".log"
// Размер буфера для bufio.Writer.
AOFBufferSize = 64 * 1024
)
// =============================================================================
// ТИПЫ
// =============================================================================
// AOFOperation — одна операция в AOF.
type AOFOperation struct {
Type string `json:"type"` // insert, update, delete, restore
Database string `json:"database"`
Collection string `json:"collection"`
DocumentID string `json:"document_id"`
Data map[string]interface{} `json:"data,omitempty"`
Timestamp int64 `json:"timestamp"`
CRC uint32 `json:"crc,omitempty"`
}
// AOFManager управляет append-only file.
type AOFManager struct {
mu sync.Mutex
dir string
currentFile *os.File
writer *bufio.Writer
currentSize int64
fileIndex uint32
fsyncEnabled bool
logger LoggerInterface
closed bool
}
// NewAOFManager создаёт AOF-менеджер.
func NewAOFManager(dir string, fsyncEnabled bool, logger LoggerInterface) (*AOFManager, error) {
if err := os.MkdirAll(dir, 0755); err != nil {
return nil, fmt.Errorf("failed to create AOF directory: %v", err)
}
aof := &AOFManager{
dir: dir,
fsyncEnabled: fsyncEnabled,
logger: logger,
}
// Находим максимальный индекс файла.
files, err := aof.listFiles()
if err != nil {
return nil, err
}
if len(files) > 0 {
lastFile := files[len(files)-1]
var idx uint32
if _, err := fmt.Sscanf(filepath.Base(lastFile), AOFFilePrefix+"%d"+AOFFileSuffix, &idx); err == nil {
aof.fileIndex = idx
}
}
// Открываем текущий файл на дозапись.
if err := aof.openCurrentFile(); err != nil {
return nil, err
}
return aof, nil
}
// SetLogger устанавливает логгер.
func (aof *AOFManager) SetLogger(logger LoggerInterface) {
aof.mu.Lock()
defer aof.mu.Unlock()
aof.logger = logger
}
// listFiles возвращает отсортированный список AOF-файлов.
func (aof *AOFManager) listFiles() ([]string, error) {
pattern := filepath.Join(aof.dir, AOFFilePrefix+"*"+AOFFileSuffix)
files, err := filepath.Glob(pattern)
if err != nil {
return nil, err
}
sort.Strings(files)
return files, nil
}
// openCurrentFile открывает текущий файл для дозаписи.
func (aof *AOFManager) openCurrentFile() error {
path := filepath.Join(aof.dir, fmt.Sprintf("%s%d%s", AOFFilePrefix, aof.fileIndex, AOFFileSuffix))
f, err := os.OpenFile(path, os.O_CREATE|os.O_APPEND|os.O_RDWR, 0644)
if err != nil {
return fmt.Errorf("failed to open AOF file: %v", err)
}
stat, err := f.Stat()
if err != nil {
f.Close()
return err
}
aof.currentFile = f
aof.writer = bufio.NewWriterSize(f, AOFBufferSize)
aof.currentSize = stat.Size()
return nil
}
// rotateIfNeeded проверяет размер и ротирует файл.
func (aof *AOFManager) rotateIfNeeded() error {
if aof.currentSize < AOFMaxFileSize {
return nil
}
if err := aof.writer.Flush(); err != nil {
return err
}
if aof.fsyncEnabled {
RealFsync(aof.currentFile)
}
if err := aof.currentFile.Close(); err != nil {
return err
}
aof.fileIndex++
return aof.openCurrentFile()
}
// Append записывает одну операцию в AOF.
func (aof *AOFManager) Append(op *AOFOperation) error {
aof.mu.Lock()
defer aof.mu.Unlock()
if aof.closed {
return fmt.Errorf("AOF is closed")
}
op.Timestamp = time.Now().UnixMilli()
// Вычисляем CRC.
data, err := json.Marshal(op)
if err != nil {
return err
}
op.CRC = crc32(data)
data, err = json.Marshal(op)
if err != nil {
return err
}
data = append(data, '\n')
if _, err := aof.writer.Write(data); err != nil {
return err
}
aof.currentSize += int64(len(data))
if err := aof.writer.Flush(); err != nil {
return err
}
if aof.fsyncEnabled {
RealFsync(aof.currentFile)
}
return aof.rotateIfNeeded()
}
// AppendBatch записывает все операции batch в AOF.
func (aof *AOFManager) AppendBatch(batch *Batch) error {
for _, op := range batch.Operations {
aofOp := &AOFOperation{
Type: op.Type,
Database: op.Database,
Collection: op.Collection,
DocumentID: op.DocumentID,
Data: op.Data,
}
if err := aof.Append(aofOp); err != nil {
return err
}
}
return nil
}
// ReadAll читает все операции из всех AOF-файлов.
func (aof *AOFManager) ReadAll() ([]*AOFOperation, error) {
aof.mu.Lock()
defer aof.mu.Unlock()
files, err := aof.listFiles()
if err != nil {
return nil, err
}
ops := make([]*AOFOperation, 0)
for _, file := range files {
fileOps, err := aof.readFile(file)
if err != nil {
if aof.logger != nil {
aof.logger.Warn(fmt.Sprintf("Failed to read AOF file %s: %v", file, err))
}
continue
}
ops = append(ops, fileOps...)
}
return ops, nil
}
// readFile читает операции из одного файла.
func (aof *AOFManager) readFile(path string) ([]*AOFOperation, error) {
f, err := os.Open(path)
if err != nil {
return nil, err
}
defer f.Close()
ops := make([]*AOFOperation, 0)
scanner := bufio.NewScanner(f)
scanner.Buffer(make([]byte, 0, 64*1024), 16*1024*1024)
for scanner.Scan() {
line := scanner.Bytes()
if len(line) == 0 {
continue
}
var op AOFOperation
if err := json.Unmarshal(line, &op); err != nil {
continue
}
// Проверяем CRC.
expectedCRC := op.CRC
op.CRC = 0
data, _ := json.Marshal(&op)
calculatedCRC := crc32(data)
op.CRC = expectedCRC
if calculatedCRC != expectedCRC {
continue
}
ops = append(ops, &op)
}
return ops, scanner.Err()
}
// Replay воспроизводит все операции AOF, применяя их к хранилищу.
//
// Используется при старте для восстановления состояния после краха.
func (aof *AOFManager) Replay(storage *Storage) error {
ops, err := aof.ReadAll()
if err != nil {
return err
}
if len(ops) == 0 {
return nil
}
if aof.logger != nil {
aof.logger.Info(fmt.Sprintf("Replaying %d AOF operations", len(ops)))
}
for _, op := range ops {
db, err := storage.GetDatabase(op.Database)
if err != nil {
// База данных могла быть создана позже — пропускаем.
continue
}
coll, err := db.GetCollection(op.Collection)
if err != nil {
continue
}
switch op.Type {
case "insert":
doc := NewDocumentWithID(op.DocumentID)
for k, v := range op.Data {
doc.SetField(k, v)
}
doc.CreatedAt = op.Timestamp
doc.UpdatedAt = op.Timestamp
coll.Insert(doc)
case "update":
coll.Update(op.DocumentID, op.Data)
case "delete":
coll.Delete(op.DocumentID)
case "restore":
coll.RestoreDeleted(op.DocumentID)
}
}
return nil
}
// Close закрывает AOF-менеджер.
func (aof *AOFManager) Close() error {
aof.mu.Lock()
defer aof.mu.Unlock()
if aof.closed {
return nil
}
aof.closed = true
if aof.writer != nil {
if err := aof.writer.Flush(); err != nil {
return err
}
}
if aof.currentFile != nil {
if aof.fsyncEnabled {
RealFsync(aof.currentFile)
}
return aof.currentFile.Close()
}
return nil
}
// GetStats возвращает статистику AOF.
func (aof *AOFManager) GetStats() map[string]interface{} {
aof.mu.Lock()
defer aof.mu.Unlock()
files, _ := aof.listFiles()
var totalSize int64
for _, f := range files {
if stat, err := os.Stat(f); err == nil {
totalSize += stat.Size()
}
}
return map[string]interface{}{
"file_count": len(files),
"total_size": totalSize,
"current_file": fmt.Sprintf("%s%d%s", AOFFilePrefix, aof.fileIndex, AOFFileSuffix),
"current_size": aof.currentSize,
"fsync_enabled": aof.fsyncEnabled,
"dir": aof.dir,
}
}
// =============================================================================
// ИНТЕГРАЦИЯ С COLLECTION
// =============================================================================
// LogAOFInsert логирует вставку документа.
func LogAOFInsert(dbName, collName string, doc *Document) {
if globalBatchManager == nil || globalBatchManager.aof == nil {
return
}
op := &AOFOperation{
Type: "insert",
Database: dbName,
Collection: collName,
DocumentID: doc.ID,
Data: doc.GetFields(),
}
if err := globalBatchManager.aof.Append(op); err != nil {
if globalBatchManager.logger != nil {
globalBatchManager.logger.Warn(fmt.Sprintf("AOF insert failed: %v", err))
}
}
}
// LogAOFUpdate логирует обновление документа.
func LogAOFUpdate(dbName, collName, docID string, updates map[string]interface{}) {
if globalBatchManager == nil || globalBatchManager.aof == nil {
return
}
op := &AOFOperation{
Type: "update",
Database: dbName,
Collection: collName,
DocumentID: docID,
Data: updates,
}
if err := globalBatchManager.aof.Append(op); err != nil {
if globalBatchManager.logger != nil {
globalBatchManager.logger.Warn(fmt.Sprintf("AOF update failed: %v", err))
}
}
}
// LogAOFDelete логирует удаление документа.
func LogAOFDelete(dbName, collName, docID string) {
if globalBatchManager == nil || globalBatchManager.aof == nil {
return
}
op := &AOFOperation{
Type: "delete",
Database: dbName,
Collection: collName,
DocumentID: docID,
}
if err := globalBatchManager.aof.Append(op); err != nil {
if globalBatchManager.logger != nil {
globalBatchManager.logger.Warn(fmt.Sprintf("AOF delete failed: %v", err))
}
}
}
// LogAOFRestore логирует восстановление документа.
func LogAOFRestore(dbName, collName, docID string) {
if globalBatchManager == nil || globalBatchManager.aof == nil {
return
}
op := &AOFOperation{
Type: "restore",
Database: dbName,
Collection: collName,
DocumentID: docID,
}
if err := globalBatchManager.aof.Append(op); err != nil {
if globalBatchManager.logger != nil {
globalBatchManager.logger.Warn(fmt.Sprintf("AOF restore failed: %v", err))
}
}
}
// GetAOFStats возвращает статистику AOF.
func GetAOFStats() map[string]interface{} {
if globalBatchManager == nil || globalBatchManager.aof == nil {
return map[string]interface{}{"error": "AOF not initialized"}
}
return globalBatchManager.aof.GetStats()
}
// ReplayAOF воспроизводит AOF при старте.
func ReplayAOF(storage *Storage) error {
if globalBatchManager == nil || globalBatchManager.aof == nil {
return nil
}
return globalBatchManager.aof.Replay(storage)
}
// =============================================================================
// ВСПОМОГАТЕЛЬНЫЕ
// =============================================================================
// trimAOFExtension убирает суффикс из имени файла.
func trimAOFExtension(name string) string {
return strings.TrimSuffix(name, AOFFileSuffix)
}
File diff suppressed because it is too large. Load diff
File diff suppressed because it is too large. Load diff
+125 -37
View File
@@ -9,12 +9,30 @@
*/
// Файл: internal/storage/persistence.go
// Назначение: Персистентное хранение данных на диске с поддержкой checkpoint и recovery
// ИСПРАВЛЕНО: Защита от torn write через CRC и checksum в имени файла
// ИСПРАВЛЕНО: Обработка частично записанных WAL сегментов при восстановлении
// ИСПРАВЛЕНО: Добавлена валидация целостности снапшотов перед загрузкой
// ИСПРАВЛЕНО: Частичное применение транзакции теперь откатывается
// ИСПРАВЛЕНО: Drop перед Create при restore защищён (только после успешного создания)
// Назначение: Персистентное хранение данных на диске с поддержкой checkpoint и recovery.
//
// ИСПРАВЛЕНО: защита от torn write через CRC и checksum в имени файла.
// ИСПРАВЛЕНО: обработка частично записанных WAL сегментов при восстановлении.
// ИСПРАВЛЕНО: добавлена валидация целостности снапшотов перед загрузкой.
// ИСПРАВЛЕНО: частичное применение транзакции теперь откатывается.
// ИСПРАВЛЕНО: Drop перед Create при restore защищён (только после успешного создания).
//
// ИСПРАВЛЕНО (2026-10, переход ACID → Batch):
// 1. applyWALEntry больше не читает поле State (его нет в storage.Batch).
// Формат WAL-записи Type=1 — это JSON-сериализованный storage.Batch.
// Раньше txRecord.State всегда был 0 и функция молча возвращала nil
// на всех записях — WAL не применялся при восстановлении.
// 2. ReadSegment теперь читает 8-байтовый заголовок записи WAL
// (4 байта длины big-endian + 4 байта CRC big-endian), как пишет
// SegmentedWALManager. Раньше читались только 4 байта длины, и
// CRC попадал в начало JSON — Unmarshal падал на всех записях.
// Плюс добавлена опциональная проверка CRC.
// 3. Сегменты WAL сортируются численно по ID, а не лексикографически.
// Раньше при 10+ сегментах wal_segment_1.log, wal_segment_10.log,
// wal_segment_2.log читались в неверном порядке, и "последний"
// сегмент оказывался не самым свежим.
// 4. compress/decompress используют gzip по умолчанию, как и раньше,
// но теперь корректно обрабатывают пустой вход и ошибки Close.
package storage
@@ -22,6 +40,7 @@ import (
"bytes"
"compress/gzip"
"crypto/sha256"
"encoding/binary"
"encoding/hex"
"encoding/json"
"fmt"
@@ -34,7 +53,11 @@ import (
"time"
)
// WALEntry представляет запись WAL
// =============================================================================
// ТИПЫ WAL
// =============================================================================
// WALEntry представляет запись WAL (публичная форма для читателя).
type WALEntry struct {
Index uint64 `json:"index"`
Term uint64 `json:"term"`
@@ -44,6 +67,7 @@ type WALEntry struct {
CRC uint32 `json:"crc,omitempty"`
}
// WALReader — интерфейс чтения WAL.
type WALReader interface {
ReadSince(index uint64) ([]WALEntry, error)
GetCurrentIndex() (uint64, error)
@@ -53,6 +77,12 @@ type WALReader interface {
ReadSegment(segmentPath string) ([]WALEntry, error)
}
// walReaderImpl читает сегменты WAL, записанные SegmentedWALManager.
//
// Формат сегмента:
// [4 байта длины (big-endian)] [4 байта CRC (big-endian)] [length байт данных]
//
// Данные — JSON-сериализованный WALRecord (см. transactions.go).
type walReaderImpl struct {
segmentsDir string
lastBackupIndex atomic.Uint64
@@ -60,6 +90,7 @@ type walReaderImpl struct {
logger LoggerInterface
}
// NewWALReaderImpl создаёт читатель WAL.
func NewWALReaderImpl(segmentsDir string, logger LoggerInterface) WALReader {
wr := &walReaderImpl{
segmentsDir: segmentsDir,
@@ -84,6 +115,7 @@ func (wr *walReaderImpl) loadLastBackupIndex() {
}
}
// ReadSince возвращает записи с Index > index.
func (wr *walReaderImpl) ReadSince(index uint64) ([]WALEntry, error) {
wr.mu.RLock()
defer wr.mu.RUnlock()
@@ -109,6 +141,7 @@ func (wr *walReaderImpl) ReadSince(index uint64) ([]WALEntry, error) {
return entries, nil
}
// GetCurrentIndex возвращает максимальный LSN по последнему сегменту.
func (wr *walReaderImpl) GetCurrentIndex() (uint64, error) {
segments, err := wr.GetSegments()
if err != nil {
@@ -140,6 +173,11 @@ func (wr *walReaderImpl) SetLastBackupIndex(index uint64) error {
return os.WriteFile(indexPath, data, 0644)
}
// GetSegments возвращает сегменты WAL, отсортированные по числовому ID.
//
// ИСПРАВЛЕНО: числовая сортировка вместо лексикографической, потому что
// имена файлов wal_segment_1.log, wal_segment_10.log, wal_segment_2.log
// лексикографически упорядочены как 1, 10, 2 — неверный порядок.
func (wr *walReaderImpl) GetSegments() ([]string, error) {
if wr.segmentsDir == "" {
return nil, fmt.Errorf("segments directory not set")
@@ -149,10 +187,27 @@ func (wr *walReaderImpl) GetSegments() ([]string, error) {
if err != nil {
return nil, err
}
sort.Strings(files)
sort.Slice(files, func(i, j int) bool {
return walSegmentIDFromPath(files[i]) < walSegmentIDFromPath(files[j])
})
return files, nil
}
// walSegmentIDFromPath извлекает числовой ID из имени файла сегмента.
func walSegmentIDFromPath(path string) uint64 {
base := filepath.Base(path)
var id uint64
if _, err := fmt.Sscanf(base, "wal_segment_%d.log", &id); err != nil {
return 0
}
return id
}
// ReadSegment читает один сегмент WAL и возвращает записи.
//
// ИСПРАВЛЕНО: заголовок записи теперь 8 байт (4 длины + 4 CRC),
// как в SegmentedWALManager. Раньше читались только 4 байта длины,
// и CRC попадал в данные — все записи читались неверно.
func (wr *walReaderImpl) ReadSegment(segmentPath string) ([]WALEntry, error) {
data, err := os.ReadFile(segmentPath)
if err != nil {
@@ -161,54 +216,73 @@ func (wr *walReaderImpl) ReadSegment(segmentPath string) ([]WALEntry, error) {
entries := make([]WALEntry, 0)
pos := 0
for pos < len(data) {
if pos+4 > len(data) {
if pos+8 > len(data) {
if wr.logger != nil {
wr.logger.Warn(fmt.Sprintf("Truncated record header at offset %d in %s", pos, segmentPath))
}
break
}
length := int(data[pos])<<24 | int(data[pos+1])<<16 | int(data[pos+2])<<8 | int(data[pos+3])
pos += 4
length := int(binary.BigEndian.Uint32(data[pos : pos+4]))
expectedCRC := binary.BigEndian.Uint32(data[pos+4 : pos+8])
pos += 8
if length <= 0 || length > 100*1024*1024 {
if wr.logger != nil {
wr.logger.Warn(fmt.Sprintf("Invalid record length %d at offset %d in %s", length, pos-4, segmentPath))
wr.logger.Warn(fmt.Sprintf("Invalid record length %d at offset %d in %s", length, pos-8, segmentPath))
}
break
}
if pos+length > len(data) {
if wr.logger != nil {
wr.logger.Warn(fmt.Sprintf("Torn write detected: expected %d bytes at offset %d, but only %d available in %s",
wr.logger.Warn(fmt.Sprintf("Torn write detected: expected %d bytes at offset %d, only %d available in %s",
length, pos, len(data)-pos, segmentPath))
}
break
}
recordData := data[pos : pos+length]
pos += length
var record struct {
LSN uint64 `json:"lsn"`
Timestamp int64 `json:"timestamp"`
CRC uint32 `json:"crc"`
Length uint32 `json:"length"`
Type byte `json:"type"`
Data []byte `json:"data"`
CRC uint32 `json:"crc,omitempty"`
Timestamp int64 `json:"timestamp"`
LSN uint64 `json:"lsn"`
}
if err := json.Unmarshal(recordData, &record); err != nil {
if wr.logger != nil {
wr.logger.Warn(fmt.Sprintf("Failed to unmarshal record at offset %d in %s: %v", pos-length, segmentPath, err))
wr.logger.Warn(fmt.Sprintf("Failed to unmarshal record at offset %d in %s: %v",
pos-length, segmentPath, err))
}
continue
}
entry := WALEntry{
// Опциональная проверка CRC. Если expectedCRC ноль или record.CRC ноль —
// пропускаем (запись может быть из старого формата). Если оба
// ненулевые и не совпадают — пропускаем как повреждённую.
if expectedCRC != 0 && record.CRC != 0 && expectedCRC != record.CRC {
if wr.logger != nil {
wr.logger.Warn(fmt.Sprintf("CRC mismatch at offset %d in %s: expected %08x, got %08x",
pos-length, segmentPath, expectedCRC, record.CRC))
}
continue
}
entries = append(entries, WALEntry{
Index: record.LSN,
Timestamp: record.Timestamp,
Type: fmt.Sprintf("%d", record.Type),
Data: record.Data,
CRC: record.CRC,
}
entries = append(entries, entry)
})
}
return entries, nil
}
// =============================================================================
// КОНФИГУРАЦИЯ
// =============================================================================
type PersistenceConfig struct {
DataDir string `json:"data_dir"`
CheckpointInterval time.Duration `json:"checkpoint_interval"`
@@ -245,6 +319,10 @@ func DefaultPersistenceConfig() *PersistenceConfig {
}
}
// =============================================================================
// СТРУКТУРЫ
// =============================================================================
type DatabaseSnapshot struct {
Name string `json:"name"`
Collections map[string]interface{} `json:"collections"`
@@ -609,9 +687,7 @@ func (pm *PersistenceManager) LoadDatabase(dbName string) error {
if !ok {
continue
}
// ИСПРАВЛЕНО: Не удаляем существующую коллекцию сразу. Создаём новую временно.
if err := db.CreateCollection(collName); err != nil {
// Коллекция уже существует - пропускаем
if pm.logger != nil {
pm.logger.Warn(fmt.Sprintf("Collection %s already exists during restore, skipping creation", collName))
}
@@ -683,7 +759,6 @@ func (pm *PersistenceManager) LoadDatabase(dbName string) error {
}
} else {
for _, entry := range entries {
// ИСПРАВЛЕНО: applyWALEntry теперь применяет транзакцию атомарно
if err := pm.applyWALEntry(db, entry); err != nil {
if pm.logger != nil {
pm.logger.Error(fmt.Sprintf("Failed to apply WAL entry %d (rolled back): %v", entry.Index, err))
@@ -707,16 +782,23 @@ func (pm *PersistenceManager) LoadDatabase(dbName string) error {
return nil
}
// applyWALEntry применяет одну запись WAL к базе данных
// ИСПРАВЛЕНО: Частичное применение транзакции откатывается при ошибке
// applyWALEntry применяет одну запись WAL к базе данных.
//
// ИСПРАВЛЕНО (2026-10): формат WAL-записи Type=1 — это storage.Batch.
// Поле State в Batch отсутствует, поэтому прежняя проверка
// `if txRecord.State != 1` молча отбрасывала все записи. Теперь
// структура соответствует Batch, а частичное применение откатывается.
func (pm *PersistenceManager) applyWALEntry(db *Database, entry WALEntry) error {
if entry.Type != "1" {
return nil
}
var txRecord struct {
// Формат совпадает со storage.Batch (см. transactions.go).
var batch struct {
ID uint64 `json:"id"`
State int32 `json:"state"`
Timestamp int64 `json:"timestamp"`
Database string `json:"database,omitempty"`
Collection string `json:"collection,omitempty"`
Operations []struct {
Type string `json:"type"`
Database string `json:"database"`
@@ -726,19 +808,16 @@ func (pm *PersistenceManager) applyWALEntry(db *Database, entry WALEntry) error
Version uint64 `json:"version"`
} `json:"operations"`
}
if err := json.Unmarshal(entry.Data, &txRecord); err != nil {
if err := json.Unmarshal(entry.Data, &batch); err != nil {
return err
}
if txRecord.State != 1 {
return nil
}
// ИСПРАВЛЕНО: Собираем успешно применённые операции для отката
// Собираем успешно применённые операции для отката.
type appliedOp struct {
opType string
collection string
docID string
oldDoc *Document // для отката update/delete
oldDoc *Document
}
applied := make([]appliedOp, 0)
@@ -758,12 +837,15 @@ func (pm *PersistenceManager) applyWALEntry(db *Database, entry WALEntry) error
}
case "delete":
coll.RestoreDeleted(ao.docID)
case "restore":
// Откат restore — снова soft-delete.
coll.Delete(ao.docID)
}
}
}
for _, op := range txRecord.Operations {
if op.Database != db.Name() {
for _, op := range batch.Operations {
if op.Database != "" && op.Database != db.Name() {
continue
}
coll, err := db.GetCollection(op.Collection)
@@ -784,7 +866,6 @@ func (pm *PersistenceManager) applyWALEntry(db *Database, entry WALEntry) error
applied = append(applied, appliedOp{opType: "insert", collection: op.Collection, docID: op.DocumentID})
case "update":
// ИСПРАВЛЕНО: сохраняем старое состояние для отката
oldDoc, _ := coll.Find(op.DocumentID)
var oldCopy *Document
if oldDoc != nil {
@@ -802,6 +883,13 @@ func (pm *PersistenceManager) applyWALEntry(db *Database, entry WALEntry) error
return err
}
applied = append(applied, appliedOp{opType: "delete", collection: op.Collection, docID: op.DocumentID})
case "restore":
if err := coll.RestoreDeleted(op.DocumentID); err != nil {
rollback()
return err
}
applied = append(applied, appliedOp{opType: "restore", collection: op.Collection, docID: op.DocumentID})
}
}
return nil
@@ -960,7 +1048,7 @@ func (pm *PersistenceManager) GetLastCheckpointInfo() map[string]interface{} {
defer pm.mu.RUnlock()
info := map[string]interface{}{
"last_checkpoint_time": pm.lastCheckpoint,
"last_checkpoint_time_str": time.UnixMilli(pm.lastCheckpoint).Format("2006-01-02 15:04:05.000"),
"last_checkpoint_time_str": time.UnixMilli(pm.lastCheckpoint).Format("02-01-2006 15:04:05.000"),
"checkpoint_id": pm.checkpointID.Load(),
"wal_path": pm.config.WalPath,
"use_wal": pm.config.UseWalForCheckpoint,
+70 -197
View File
@@ -33,11 +33,11 @@
// Transaction — теперь эти ссылки заменены на GetBatchManager() и *Batch.
//
// ДОБАВЛЕНО (2026-10, идемпотентность SAGA + выборы лидера):
// - SagaPersistentStorage теперь ведёт персистентную таблицу
// выполненных ExecutionID (saga_executed_keys.json). Перед выполнением
// шага и перед его компенсацией оркестратор проверяет эту таблицу,
// чтобы повторный запуск SAGA после падения не приводил к дублированию
// операций на удалённых узлах.
// - SagaPersistentStorage ведёт персистентную таблицу выполненных
// ExecutionID (saga_executed_keys.json). Перед выполнением шага
// и перед его компенсацией оркестратор проверяет эту таблицу,
// чтобы повторный запуск SAGA после падения не приводил к
// дублированию операций на удалённых узлах.
// - SagaOrchestrator получил SagaLeaderElector, который берёт лидера
// из Raft (через SagaRaftAccessor). Если Raft недоступен —
// используется резервный режим «единственный узел — лидер».
@@ -48,8 +48,15 @@
// узел знал, был ли он лидером, и корректно себя вёл до первых
// выборов.
// - Добавлен интерфейс SagaRaftAccessor, чтобы storage не зависел
// напрямую от hashicorp/raft (это позволяет избежать циклических
// импортов и упрощает тестирование).
// напрямую от hashicorp/raft.
//
// ДОБАВЛЕНО (2026-10, ревизия):
// - GetAllSagas у SagaManager — читает все SAGA (включая завершённые)
// с диска через SagaPersistentStorage.ListAll(). Используется
// REPL-командой "saga list --all".
//
// ИЗМЕНЕНО (2026-10): удалён AOF. BatchManager больше не содержит
// поле aof, и Commit не пишет в AOF. Единственный источник истины — WAL.
package storage
@@ -127,14 +134,12 @@ const (
SagaCleanupInterval = 1 * time.Minute
// SagaExecutedKeysFile — персистентная таблица выполненных ExecutionID.
// Нужна для идемпотентности шагов SAGA при повторе после падения.
SagaExecutedKeysFile = "saga_executed_keys.json"
// SagaLeaderStateFile — персистентное состояние лидера SAGA.
SagaLeaderStateFile = "saga_leader_state.json"
// SagaLeaderElectionInterval — как часто опрашиваем Raft на предмет
// лидерства. Меньше — быстрее реакция, но больше нагрузка.
// SagaLeaderElectionInterval — как часто опрашиваем Raft на предмет лидерства.
SagaLeaderElectionInterval = 2 * time.Second
)
@@ -197,7 +202,7 @@ func crc32(data []byte) uint32 {
}
// =============================================================================
// BATCH MANAGER - замена TransactionManager
// BATCH MANAGER — замена TransactionManager
// =============================================================================
// BatchManager управляет batch-операциями и WAL.
@@ -207,12 +212,8 @@ type BatchManager struct {
logger LoggerInterface
walPath string
stats *BatchStats
aof *AOFManager // append-only file для мутаций вне batch
// activeBatches — реестр активных (незакоммиченных) batch'ей.
// Ключ — Batch.ID, значение — *Batch.
// Используется eviction-механизмом (runtime_limits.go), чтобы не
// вытеснять документы, участвующие в незакоммиченной batch-операции.
activeBatches sync.Map
}
@@ -262,14 +263,6 @@ func InitBatchManagerWithConfig(walPath string, config map[string]interface{}) e
if err != nil {
return
}
// Инициализируем AOF для мутаций вне batch.
globalBatchManager.aof, err = NewAOFManager(
filepath.Join(filepath.Dir(walPath), "aof"),
fsyncEnabled, nil)
if err != nil {
return
}
})
return err
}
@@ -284,9 +277,6 @@ func SetBatchLogger(logger LoggerInterface) {
if globalBatchManager.wal != nil {
globalBatchManager.wal.logger = logger
}
if globalBatchManager.aof != nil {
globalBatchManager.aof.SetLogger(logger)
}
}
}
@@ -301,7 +291,6 @@ func GetGlobalStorage() *Storage { return globalStorage }
// =============================================================================
// RegisterBatch регистрирует batch как активный.
// Вызывается в NewBatch.
func (bm *BatchManager) RegisterBatch(b *Batch) {
if bm == nil || b == nil {
return
@@ -310,7 +299,6 @@ func (bm *BatchManager) RegisterBatch(b *Batch) {
}
// UnregisterBatch снимает регистрацию batch.
// Вызывается в Commit (успешном или неуспешном).
func (bm *BatchManager) UnregisterBatch(b *Batch) {
if bm == nil || b == nil {
return
@@ -318,11 +306,7 @@ func (bm *BatchManager) UnregisterBatch(b *Batch) {
bm.activeBatches.Delete(b.ID)
}
// IsDocInActiveBatch проверяет, участвует ли документ в активной
// (незакоммиченной) batch-операции.
//
// Используется в runtime_limits.go (eviction), чтобы не вытеснять
// документы, которые в данный момент участвуют в batch.
// IsDocInActiveBatch проверяет, участвует ли документ в активной batch-операции.
func (bm *BatchManager) IsDocInActiveBatch(docID, dbName, collName string) bool {
if bm == nil {
return false
@@ -347,7 +331,6 @@ func (bm *BatchManager) IsDocInActiveBatch(docID, dbName, collName string) bool
}
// GetActiveBatchCount возвращает количество активных batch'ей.
// Полезно для метрик и отладки.
func (bm *BatchManager) GetActiveBatchCount() int {
if bm == nil {
return 0
@@ -376,8 +359,6 @@ func NewBatch(database, collection string) *Batch {
Database: database,
Collection: collection,
}
// Регистрируем batch как активный, чтобы eviction не вытеснил
// документы, участвующие в нём.
globalBatchManager.RegisterBatch(b)
return b
}
@@ -426,29 +407,17 @@ func (b *Batch) AddRestore(docID string) {
}
// Commit применяет batch атомарно.
//
// Гарантия атомарности:
// 1. Сериализуем весь batch и пишем в WAL (fsync).
// 2. Только после успешной записи в WAL применяем операции.
// 3. Если применение падает на середине — мы можем восстановить
// состояние, проиграв WAL (операции идемпотентны или
// откатываются вручную).
//
// В любом случае (успех или ошибка) batch снимается с регистрации
// в activeBatches.
func (b *Batch) Commit() error {
if globalBatchManager == nil {
return fmt.Errorf("batch manager not initialized")
}
// Снимаем регистрацию в любом случае — batch короткоживущий.
defer globalBatchManager.UnregisterBatch(b)
if len(b.Operations) == 0 {
return nil
}
// 1. Пишем batch в WAL.
data, err := json.Marshal(b)
if err != nil {
return fmt.Errorf("failed to marshal batch: %v", err)
@@ -460,37 +429,22 @@ func (b *Batch) Commit() error {
return fmt.Errorf("failed to write batch to WAL: %v", err)
}
// 2. Применяем операции.
if err := b.apply(); err != nil {
globalBatchManager.stats.TotalFailed.Add(1)
return fmt.Errorf("batch apply failed: %v", err)
}
// 3. Также пишем в AOF для быстрого восстановления (опционально).
if globalBatchManager.aof != nil {
if err := globalBatchManager.aof.AppendBatch(b); err != nil {
if globalBatchManager.logger != nil {
globalBatchManager.logger.Warn(fmt.Sprintf("AOF append failed: %v", err))
}
}
}
globalBatchManager.stats.TotalBatches.Add(1)
globalBatchManager.stats.TotalOperations.Add(uint64(len(b.Operations)))
return nil
}
// apply применяет все операции batch.
//
// Если операция падает, мы пытаемся откатить уже применённые
// (best-effort). Полная атомарность гарантируется только на уровне
// WAL: при восстановлении batch либо весь применится, либо нет.
func (b *Batch) apply() error {
if globalStorage == nil {
return fmt.Errorf("storage not initialized")
}
// Собираем откат для уже применённых операций.
type appliedOp struct {
opType string
db string
@@ -592,11 +546,7 @@ func (b *Batch) apply() error {
// TIME-TRAVEL QUERIES (WAL-based)
// =============================================================================
// GetDocumentAtTimestamp возвращает документ на указанный момент времени,
// проигрывая WAL с начала до нужного LSN.
//
// Это O(N) по количеству записей WAL — для in-memory СУБД с небольшим WAL
// это приемлемо. Для больших WAL нужен индекс по времени.
// GetDocumentAtTimestamp возвращает документ на указанный момент времени.
func GetDocumentAtTimestamp(docID string, timestamp int64) (*Document, error) {
if globalBatchManager == nil {
return nil, fmt.Errorf("batch manager not initialized")
@@ -661,8 +611,7 @@ func GetDocumentAtTimestamp(docID string, timestamp int64) (*Document, error) {
return current, nil
}
// GetDocumentVersionsSince возвращает все версии документа,
// созданные после указанного времени.
// GetDocumentVersionsSince возвращает все версии документа после указанного времени.
func GetDocumentVersionsSince(docID string, since int64) ([]*Document, error) {
if globalBatchManager == nil || globalBatchManager.wal == nil {
return nil, fmt.Errorf("batch manager not initialized")
@@ -747,8 +696,6 @@ type SegmentedWALManager struct {
logger LoggerInterface
fsyncEnabled bool
// ackMap — map LSN -> chan struct{} для синхронного подтверждения.
// Ключ — LSN записи. После записи в WAL закрываем канал.
ackMu sync.Mutex
ackMap map[uint64]chan struct{}
}
@@ -871,13 +818,9 @@ func (wm *SegmentedWALManager) Write(record *WALRecord) error {
}
// WriteSync — синхронная запись с гарантией fsync.
//
// Используем ackMap по LSN, чтобы не было race condition
// с чужими подтверждениями.
func (wm *SegmentedWALManager) WriteSync(record *WALRecord) error {
record.Timestamp = time.Now().UnixMilli()
// Присваиваем LSN под защитой mu.
wm.mu.Lock()
lsn := wm.getCurrentLSNLocked() + 1
record.LSN = lsn
@@ -981,7 +924,6 @@ func (wm *SegmentedWALManager) flushBatch(batch []*WALRecord) {
wm.mu.Unlock()
// Подтверждаем все записи из batch.
wm.ackMu.Lock()
for _, record := range batch {
if ch, ok := wm.ackMap[record.LSN]; ok {
@@ -1150,14 +1092,14 @@ func (im *WALIndexManager) load() error {
}
// =============================================================================
// SAGA — ИНТЕРФЕЙСЫ ДЛЯ RAFT (без прямого импорта hashicorp/raft)
// SAGA — ИНТЕРФЕЙСЫ ДЛЯ RAFT
// =============================================================================
// SagaRaftAccessor — минимальный интерфейс к Raft, необходимый
// оркестратору SAGA для выборов лидера.
//
// Реализуется в internal/cluster (RaftCoordinator), чтобы избежать
// циклических импортов и позволить storage не зависеть от raft напрямую.
// Реализуется в internal/cluster (SagaLeaderBridge), чтобы storage
// не зависел напрямую от hashicorp/raft.
type SagaRaftAccessor interface {
// IsSagaLeader возвращает true, если текущий узел является лидером
// в Raft-группе.
@@ -1224,13 +1166,8 @@ type SagaStepState struct {
// SagaPersistentStorage — персистентное хранилище SAGA.
//
// ДОБАВЛЕНО: executedKeys — персистентная таблица уже выполненных
// ExecutionID. Заполняется после успешного выполнения шага
// (в executeSagaInternal) и после успешной компенсации
// (в executeCompensation). Проверяется перед выполнением/компенсацией.
//
// Формат файла — JSON: { "execution_id": timestamp_ms, ... }.
// Это позволяет при рестарте быстро понять, какие шаги уже выполнены,
// и не дублировать их на удалённых узлах.
// ExecutionID. Заполняется после успешного выполнения шага и после
// успешной компенсации. Проверяется перед выполнением/компенсацией.
type SagaPersistentStorage struct {
baseDir string
mu sync.RWMutex
@@ -1241,16 +1178,11 @@ type SagaPersistentStorage struct {
fsyncMaxRetries int
fsyncRetryDelay time.Duration
// executedKeys — таблица выполненных ExecutionID.
// Ключ — ExecutionID, значение — UnixMilli, когда он был выполнен.
// Хранится в памяти + на диске (saga_executed_keys.json).
executedMu sync.RWMutex
executedKeys map[string]int64
}
// NewSagaPersistentStorageWithConfig создаёт хранилище SAGA.
//
// ДОБАВЛЕНО: при создании загружаем executedKeys с диска.
func NewSagaPersistentStorageWithConfig(cfg *config.SagaConfig, logger LoggerInterface) (*SagaPersistentStorage, error) {
baseDir := SagaStateDir
maxCache := 10000
@@ -1282,7 +1214,6 @@ func NewSagaPersistentStorageWithConfig(cfg *config.SagaConfig, logger LoggerInt
executedKeys: make(map[string]int64),
}
// Загружаем таблицу выполненных ключей.
if err := sps.loadExecutedKeys(); err != nil {
if logger != nil {
logger.Warn(fmt.Sprintf("Failed to load saga executed keys: %v", err))
@@ -1325,7 +1256,6 @@ func (sps *SagaPersistentStorage) loadExecutedKeys() error {
}
// saveExecutedKeys сохраняет таблицу выполненных ключей на диск.
// Вызывается под executedMu.RLock или без удержания блокировки.
func (sps *SagaPersistentStorage) saveExecutedKeys() error {
sps.executedMu.RLock()
data, err := json.Marshal(sps.executedKeys)
@@ -1364,8 +1294,7 @@ func (sps *SagaPersistentStorage) IsExecuted(executionID string) bool {
// MarkExecuted атомарно отмечает ExecutionID как выполненный
// и сохраняет таблицу на диск.
//
// Возвращает true, если ключ был добавлен впервые (т.е. операция
// действительно новая), и false, если ключ уже был (повтор).
// Возвращает true, если ключ был добавлен впервые.
func (sps *SagaPersistentStorage) MarkExecuted(executionID string) bool {
sps.executedMu.Lock()
if _, exists := sps.executedKeys[executionID]; exists {
@@ -1561,21 +1490,13 @@ func (sps *SagaPersistentStorage) LoadLeaderState() (*SagaLeaderState, error) {
// =============================================================================
// SagaLeaderElector — выборы лидера SAGA через Raft.
//
// Если Raft доступен, IsLeader() опрашивает RaftAccessor.
// Если Raft недоступен, используется резервный режим: узел считается
// лидером, только если он один в кластере (single-node mode).
//
// Все изменения лидерства сохраняются в SagaPersistentStorage
// (файл saga_leader_state.json), чтобы после рестарта узел знал,
// был ли он лидером.
type SagaLeaderElector struct {
nodeID string
raft SagaRaftAccessor
storage *SagaPersistentStorage
logger LoggerInterface
isLeader atomic.Bool
leaderID atomic.Value // string
leaderID atomic.Value
currentTerm atomic.Uint64
observersMu sync.RWMutex
@@ -1597,7 +1518,6 @@ func NewSagaLeaderElector(nodeID string, raft SagaRaftAccessor, storage *SagaPer
}
e.leaderID.Store("")
// Пытаемся загрузить прошлое состояние лидера.
if storage != nil {
if state, err := storage.LoadLeaderState(); err == nil && state != nil {
e.isLeader.Store(state.IsLeader)
@@ -1606,15 +1526,12 @@ func NewSagaLeaderElector(nodeID string, raft SagaRaftAccessor, storage *SagaPer
}
}
// Если Raft доступен, регистрируем наблюдателя, чтобы оперативно
// получать уведомления о смене лидера.
if raft != nil {
raft.RegisterSagaLeaderObserver(nodeID, func(isLeader bool, leaderID string) {
e.updateLeadership(isLeader, leaderID)
})
}
// Запускаем фоновый опрос Raft (на случай, если наблюдатель не сработал).
e.wg.Add(1)
go e.leaderElectionLoop()
@@ -1622,7 +1539,6 @@ func NewSagaLeaderElector(nodeID string, raft SagaRaftAccessor, storage *SagaPer
}
// updateLeadership — центральная точка обновления лидерства.
// Вызывается и наблюдателем, и фоновым опросом.
func (e *SagaLeaderElector) updateLeadership(isLeader bool, leaderID string) {
oldLeader := e.isLeader.Load()
oldLeaderID, _ := e.leaderID.Load().(string)
@@ -1636,7 +1552,6 @@ func (e *SagaLeaderElector) updateLeadership(isLeader bool, leaderID string) {
}
e.currentTerm.Store(term)
// Сохраняем на диск.
if e.storage != nil {
_ = e.storage.SaveLeaderState(&SagaLeaderState{
LeaderID: leaderID,
@@ -1647,7 +1562,6 @@ func (e *SagaLeaderElector) updateLeadership(isLeader bool, leaderID string) {
})
}
// Если что-то изменилось — уведомляем наблюдателей.
if oldLeader != isLeader || oldLeaderID != leaderID {
e.observersMu.RLock()
obs := make([]func(isLeader bool, leaderID string), 0, len(e.observers))
@@ -1732,28 +1646,24 @@ func (e *SagaLeaderElector) Stop() {
}
// =============================================================================
// SAGA ORCHESTRATOR (упрощённый, без мьютексов на горячем пути)
// SAGA ORCHESTRATOR
// =============================================================================
// SagaOrchestrator управляет SAGA.
//
// ДОБАВЛЕНО: leaderElector — выборы лидера через Raft.
// isLeader больше не выставляется безусловно в true: значение
// синхронизируется с leaderElector.
type SagaOrchestrator struct {
storage *SagaPersistentStorage
mu sync.RWMutex
logger LoggerInterface
nodeID string
stopChan chan struct{}
wg sync.WaitGroup
isLeader atomic.Bool
leaderID string
config *config.SagaConfig
metrics *SagaMetrics
activeSagas sync.Map // map[string]*SagaTransaction
leaderElector *SagaLeaderElector
raftAccessor SagaRaftAccessor
storage *SagaPersistentStorage
mu sync.RWMutex
logger LoggerInterface
nodeID string
stopChan chan struct{}
wg sync.WaitGroup
isLeader atomic.Bool
leaderID string
config *config.SagaConfig
metrics *SagaMetrics
activeSagas sync.Map
leaderElector *SagaLeaderElector
raftAccessor SagaRaftAccessor
}
// SagaMetrics — метрики SAGA.
@@ -1765,17 +1675,11 @@ type SagaMetrics struct {
TotalCompensated atomic.Uint64
ActiveCount atomic.Int64
RecoveryCount atomic.Uint64
IdempotentSkips atomic.Uint64 // сколько шагов было пропущено по идемпотентности
NotLeaderRejects atomic.Uint64 // сколько SAGA было отклонено из-за отсутствия лидерства
IdempotentSkips atomic.Uint64
NotLeaderRejects atomic.Uint64
}
// NewSagaOrchestratorWithConfig создаёт оркестратор.
//
// ДОБАВЛЕНО: второй аргумент raftAccessor может быть nil —
// тогда оркестратор работает в single-node режиме и всегда лидер.
//
// ДОБАВЛЕНО: если raftAccessor != nil, создаётся SagaLeaderElector,
// который опрашивает Raft и обновляет isLeader.
// NewSagaOrchestratorWithConfig создаёт оркестратор без Raft.
func NewSagaOrchestratorWithConfig(cfg *config.SagaConfig, nodeID string, logger LoggerInterface) (*SagaOrchestrator, error) {
return NewSagaOrchestratorWithRaft(cfg, nodeID, nil, logger)
}
@@ -1814,11 +1718,8 @@ func NewSagaOrchestratorWithRaft(cfg *config.SagaConfig, nodeID string, raftAcce
raftAccessor: raftAccessor,
}
// Создаём выборщик лидера.
o.leaderElector = NewSagaLeaderElector(nodeID, raftAccessor, storage, logger)
// Начальное значение isLeader синхронизируем с выборщиком.
// Если Raft недоступен (nil), считаем себя лидером — single-node режим.
if raftAccessor == nil {
o.isLeader.Store(true)
o.leaderID = nodeID
@@ -1828,7 +1729,6 @@ func NewSagaOrchestratorWithRaft(cfg *config.SagaConfig, nodeID string, raftAcce
o.leaderID = o.leaderElector.GetLeaderID()
}
// Подписываемся на изменения лидерства.
o.leaderElector.RegisterObserver("orchestrator", func(isLeader bool, leaderID string) {
o.mu.Lock()
o.leaderID = leaderID
@@ -1844,9 +1744,6 @@ func NewSagaOrchestratorWithRaft(cfg *config.SagaConfig, nodeID string, raftAcce
}
// IsLeader — является ли узел лидером.
//
// ДОБАВЛЕНО: теперь значение берётся из leaderElector, а не
// из жёстко выставленного флага.
func (o *SagaOrchestrator) IsLeader() bool {
if o.leaderElector == nil {
return true
@@ -1868,7 +1765,6 @@ func (o *SagaOrchestrator) GetLeaderID() string {
}
// IsExecuted проверяет, был ли ExecutionID уже выполнен.
// Делегируется в storage.
func (o *SagaOrchestrator) IsExecuted(executionID string) bool {
if o.storage == nil {
return false
@@ -1877,7 +1773,6 @@ func (o *SagaOrchestrator) IsExecuted(executionID string) bool {
}
// MarkExecuted помечает ExecutionID как выполненный.
// Возвращает true, если это первое выполнение.
func (o *SagaOrchestrator) MarkExecuted(executionID string) bool {
if o.storage == nil {
return true
@@ -1896,7 +1791,6 @@ func (o *SagaOrchestrator) recoveryLoop() {
for {
select {
case <-ticker.C:
// Восстановлением занимается только лидер.
if !o.IsLeader() {
continue
}
@@ -1914,7 +1808,6 @@ func (o *SagaOrchestrator) orphanCleanupLoop() {
for {
select {
case <-ticker.C:
// Очисткой занимается только лидер.
if !o.IsLeader() {
continue
}
@@ -1925,7 +1818,7 @@ func (o *SagaOrchestrator) orphanCleanupLoop() {
}
}
// cleanupOrphanSteps — очистка "осиротевших" SAGA.
// cleanupOrphanSteps — очистка «осиротевших» SAGA.
func (o *SagaOrchestrator) cleanupOrphanSteps() {
pending, err := o.storage.ListPending()
if err != nil {
@@ -1948,13 +1841,7 @@ func (o *SagaOrchestrator) cleanupOrphanSteps() {
}
}
// compensateOrphanSaga — компенсация "осиротевшей" SAGA.
//
// Проверяем nil на Compensate, сохраняем состояние после
// каждой компенсации, обрабатываем compensation_failed.
//
// ДОБАВЛЕНО: проверяем лидерство — компенсацию выполняет только лидер,
// чтобы два узла не компенсировали одну и ту же SAGA.
// compensateOrphanSaga — компенсация «осиротевшей» SAGA.
func (o *SagaOrchestrator) compensateOrphanSaga(state *SagaState) {
if !o.IsLeader() {
return
@@ -2015,8 +1902,6 @@ func (o *SagaOrchestrator) stateToTransaction(state *SagaState) *SagaTransaction
}
// recoverPendingSagas — восстановление незавершённых SAGA.
//
// ДОБАВЛЕНО: выполняется только лидером.
func (o *SagaOrchestrator) recoverPendingSagas() {
if !o.IsLeader() {
return
@@ -2050,9 +1935,6 @@ func (o *SagaOrchestrator) resumeSaga(state *SagaState) error {
}
// BeginSaga — начать SAGA.
//
// ДОБАВЛЕНО: если узел не лидер — возвращаем ошибку, чтобы клиент
// перенаправил запрос на лидера.
func (o *SagaOrchestrator) BeginSaga(id string) (*SagaTransaction, error) {
if !o.IsLeader() {
o.metrics.NotLeaderRejects.Add(1)
@@ -2098,8 +1980,6 @@ func (o *SagaOrchestrator) BeginSaga(id string) (*SagaTransaction, error) {
}
// Execute — выполнить SAGA.
//
// ДОБАВЛЕНО: только лидер выполняет SAGA.
func (o *SagaOrchestrator) Execute(saga *SagaTransaction) error {
if !o.IsLeader() {
o.metrics.NotLeaderRejects.Add(1)
@@ -2109,16 +1989,6 @@ func (o *SagaOrchestrator) Execute(saga *SagaTransaction) error {
}
// executeSagaInternal — внутренняя логика выполнения SAGA.
//
// Компенсация идемпотентна и корректно обрабатывает ошибки.
//
// ДОБАВЛЕНО: идемпотентность через storage.IsExecuted / storage.MarkExecuted.
// Перед выполнением шага проверяем, не выполнялся ли он уже. Если да —
// пропускаем (IdempotentSkips++). После успешного выполнения отмечаем
// ExecutionID как выполненный на диске.
//
// ДОБАВЛЕНО: if !o.IsLeader() — прерываем выполнение, если узел потерял
// лидерство в процессе (защита от двух лидеров).
func (o *SagaOrchestrator) executeSagaInternal(saga *SagaTransaction) error {
if !o.IsLeader() {
return fmt.Errorf("node is not the saga leader")
@@ -2138,8 +2008,6 @@ func (o *SagaOrchestrator) executeSagaInternal(saga *SagaTransaction) error {
}
for i := saga.CurrentStep; i < len(saga.Steps); i++ {
// Проверяем лидерство на каждой итерации: если потеряли —
// прерываемся, чтобы не конфликтовать с новым лидером.
if !o.IsLeader() {
return fmt.Errorf("lost saga leadership during execution of %s", saga.ID)
}
@@ -2155,7 +2023,6 @@ func (o *SagaOrchestrator) executeSagaInternal(saga *SagaTransaction) error {
step.StartedAt = time.Now().UnixMilli()
saga.mu.Unlock()
// ИДЕМПОТЕНТНОСТЬ: если шаг уже выполнялся (по журналу), пропускаем.
if o.IsExecuted(step.ExecutionID) || saga.HasExecuted(step.ExecutionID) {
saga.mu.Lock()
step.Status = "completed"
@@ -2219,8 +2086,6 @@ func (o *SagaOrchestrator) executeSagaInternal(saga *SagaTransaction) error {
return fmt.Errorf("saga %s aborted at step %s: %v", saga.ID, step.Name, err)
}
// ИДЕМПОТЕНТНОСТЬ: отмечаем шаг как выполненный на диске,
// чтобы при рестарте не выполнить его повторно.
o.MarkExecuted(step.ExecutionID)
saga.mu.Lock()
@@ -2276,14 +2141,6 @@ func (o *SagaOrchestrator) saveSagaState(saga *SagaTransaction) error {
}
// executeCompensation — выполнение компенсации.
//
// Идемпотентность: пропускаем уже скомпенсированные шаги.
// Сохраняем состояние после каждой компенсации.
// Обрабатываем compensation_failed корректно.
//
// ДОБАВЛЕНО: проверяем лидерство в начале и на каждой итерации.
// ДОБАВЛЕНО: идемпотентность через storage.IsExecuted на
// execution_id компенсации (используем суффикс ":compensate").
func (o *SagaOrchestrator) executeCompensation(saga *SagaTransaction, failedStep int) error {
if !o.IsLeader() {
return fmt.Errorf("node is not the saga leader")
@@ -2313,7 +2170,6 @@ func (o *SagaOrchestrator) executeCompensation(saga *SagaTransaction, failedStep
continue
}
// ИДЕМПОТЕНТНОСТЬ: проверяем, не компенсировали ли уже.
compExecID := step.ExecutionID + ":compensate"
if o.IsExecuted(compExecID) {
saga.mu.Lock()
@@ -2325,7 +2181,6 @@ func (o *SagaOrchestrator) executeCompensation(saga *SagaTransaction, failedStep
}
if step.Compensate == nil {
// Нет функции компенсации — считаем шаг скомпенсированным.
saga.mu.Lock()
step.Status = "compensated"
step.CompletedAt = time.Now().UnixMilli()
@@ -2355,7 +2210,6 @@ func (o *SagaOrchestrator) executeCompensation(saga *SagaTransaction, failedStep
o.saveSagaState(saga)
return fmt.Errorf("compensation for step %s failed: %v", step.Name, err)
}
// ИДЕМПОТЕНТНОСТЬ: отмечаем компенсацию как выполненную.
o.MarkExecuted(compExecID)
saga.mu.Lock()
step.Status = "compensated"
@@ -2387,8 +2241,6 @@ func (o *SagaOrchestrator) GetSaga(id string) (*SagaTransaction, error) {
}
// GetMetrics — метрики SAGA.
//
// ДОБАВЛЕНО: idempotent_skips, not_leader_rejects, leader_id.
func (o *SagaOrchestrator) GetMetrics() map[string]interface{} {
return map[string]interface{}{
"total_started": o.metrics.TotalStarted.Load(),
@@ -2654,6 +2506,30 @@ func (sm *SagaManager) GetActiveSagas() []*SagaTransaction {
return result
}
// GetAllSagas возвращает все SAGA транзакции — и активные, и завершённые.
//
// ДОБАВЛЕНО (2026-10): используется REPL-командой "saga list --all".
// В отличие от GetActiveSagas, читает состояние с диска через
// SagaPersistentStorage.ListAll() и конвертирует каждую запись в
// SagaTransaction через stateToTransaction.
func (sm *SagaManager) GetAllSagas() []*SagaTransaction {
result := make([]*SagaTransaction, 0)
if sm.orchestrator == nil || sm.orchestrator.storage == nil {
return result
}
states, err := sm.orchestrator.storage.ListAll()
if err != nil {
return result
}
for _, st := range states {
if st == nil {
continue
}
result = append(result, sm.orchestrator.stateToTransaction(st))
}
return result
}
// Stop — остановка.
func (sm *SagaManager) Stop() {
close(sm.stopChan)
@@ -2711,9 +2587,6 @@ func StopBatchManager() error {
globalBatchManager.wal.Sync()
globalBatchManager.wal.Close()
}
if globalBatchManager.aof != nil {
globalBatchManager.aof.Close()
}
return nil
}
@@ -2721,7 +2594,7 @@ func StopBatchManager() error {
// СОВМЕСТИМОСТЬ: псевдонимы для старого API
// =============================================================================
// GetTransactionStats — псевдоним для GetBatchStats (обратная совместимость).
// GetTransactionStats — псевдоним для GetBatchStats.
func GetTransactionStats() map[string]interface{} {
return GetBatchStats()
}