fix(P1.2-revue): батч-видимость не заводила в DLQ + TOCTOU лимит в SendMessageBatch

1. ChangeMessageVisibilityBatch при VisibilityTimeout=0 лишь инкрементировал
   Retry и НЕ переносил сообщение в DLQ (логика была только в одиночной версии).
   Вынесен единый метод Queue.ResetVisibilityAndTryDLQ — сброс видимости +
   DLQ-перенос. Обе версии привели к одному пути.
2. SendMessageBatch: проверка OOM-лимита была только по устаревшему снимку
   (RLock → Lock) — добавлена повторная проверка под per-queue Lock (TOCTOU).
3. Инварианты InFlightIndexes/ensureIndexes зафиксированы в док-комментариях.

Тест: TestChangeVisibilityBatch_MovesToDLQ (падает на старой логике, зелёный
на новой).
This commit is contained in:
Repinoid
2026-10-07 08:35:52 +03:00
parent f96649b14e
commit 2e1c594892
4 changed files with 67 additions and 34 deletions
+13 -10
View File
@@ -82,19 +82,22 @@ func ChangeMessageVisibilityBatchV1(req *http.Request) (int, interfaces.Abstract
messageFound := false
// O(1)-поиск по индексу ReceiptHandle (P1.2) вместо скана всего среза.
if i, ok := queue.IndexOfReceipt(entry.ReceiptHandle); ok {
msg := &queue.Messages[i]
if entry.VisibilityTimeout == 0 {
// Сброс: сообщение снова видимо, счётчик retry++
queue.ClearInFlight(i, entry.ReceiptHandle)
msg.ReceiptTime = time.Now().UTC()
msg.ReceiptHandle = ""
msg.VisibilityTimeout = time.Now().Add(time.Duration(queue.VisibilityTimeout) * time.Second)
msg.Retry++
// Сброс + DLQ-перенос при исчерпании MaxReceiveCount — единый путь
// с одиночной версией (раньше батч лишь инкрементировал Retry и не
// переносил в DLQ — счётчик и перенос могли расходиться).
uuid, moved := queue.ResetVisibilityAndTryDLQ(i, entry.ReceiptHandle)
if moved {
// Сообщение перенесено в DLQ и удалено из очереди — чистим Redis.
persistence.DeleteMessagePersist(key, uuid)
} else if j, ok := queue.IndexOfUUID(uuid); ok {
persistence.SaveMessage(key, &queue.Messages[j])
}
} else {
msg.VisibilityTimeout = time.Now().Add(time.Duration(entry.VisibilityTimeout) * time.Second)
queue.Messages[i].VisibilityTimeout = time.Now().Add(time.Duration(entry.VisibilityTimeout) * time.Second)
// Персистим изменённое сообщение отдельно
persistence.SaveMessage(key, &queue.Messages[i])
}
// Персистим изменённое сообщение отдельно
persistence.SaveMessage(key, msg)
messageFound = true
}