Repinoid
c5b2c11f5d
- models.Queue: deadIdx/receiptIdx/uuidIdx/liveCount/deadCount + примитивы (AppendMessage/RemoveIndex/Compact/CompactIfNeeded/Reset/EnsureIndexes/ IndexOfReceipt/IndexOfUUID/MarkInFlight/ClearInFlight/InFlightIndexes/LiveCount). Удаление — tombstone за O(1), физическая чистка амортизированная. - DeleteMessage/ChangeMessageVisibility(±Batch): поиск по индексу вместо скана среза; удаление без сдвига остатка. - DeleteMessageBatch: tombstone вместо полной пересборки среза. - processQueueTick: visibility/DLQ по индексу in-flight (O(in-flight)), retention — tombstone, в конце — CompactIfNeeded. - send-пути: AppendMessage + LiveCount (OOM-лимит по живым). - admin/metrics/persistence: учёт живых сообщений, индексы после загрузки из Redis. - Redis-схема не менялась (сообщения и так в per-message HASH).
78 lines
3.1 KiB
Go
78 lines
3.1 KiB
Go
package gosqs
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"shared-sqs/app/interfaces"
|
|
"shared-sqs/app/models"
|
|
"shared-sqs/app/persistence"
|
|
"shared-sqs/app/utils"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// PurgeQueueV1 — очищает ВСЕ сообщения очереди тенанта.
|
|
//
|
|
// Логическая схема:
|
|
// 1. Разбор тела запроса → PurgeQueueRequest.
|
|
// 2. Тенант из контекста (SigV4 → AccessKey).
|
|
// 3. Имя очереди из QueueUrl (последний сегмент).
|
|
// 4. Под Lock:
|
|
// - проверка существования очереди → QueueNotFound;
|
|
// - защита от повторного purge (фикс): повторный вызов в течение 60 секунд
|
|
// → PurgeQueueInProgress (требование AWS). Раньше purge можно было
|
|
// вызывать без ограничений;
|
|
// - очистка in-memory сообщений и dedup-записей;
|
|
// - фиксация LastPurgeTime;
|
|
// - асинхронная очистка Redis (DEL хэша сообщений) + SaveQueue.
|
|
// 5. Ответ: пустой PurgeQueueResponse.
|
|
func PurgeQueueV1(req *http.Request) (int, interfaces.AbstractResponseBody) {
|
|
requestBody := models.NewPurgeQueueRequest()
|
|
ok := utils.REQUEST_TRANSFORMER(requestBody, req, false)
|
|
if !ok {
|
|
log.Error("Invalid Request - PurgeQueueV1")
|
|
return utils.CreateErrorResponseV1("InvalidParameterValue", true)
|
|
}
|
|
|
|
t := getTenantFromContext(req)
|
|
if t == nil {
|
|
return utils.CreateErrorResponseV1("InvalidClientTokenId", true)
|
|
}
|
|
|
|
uriSegments := strings.Split(requestBody.QueueUrl, "/")
|
|
queueName := uriSegments[len(uriSegments)-1]
|
|
key := tenantQueueKey(t.AccessKey, queueName)
|
|
|
|
queue := models.SyncQueues.GetLocked(key)
|
|
if queue == nil {
|
|
log.Errorf("Purge Queue: %s, queue does not exist for tenant %s", queueName, t.ID)
|
|
return utils.CreateErrorResponseV1("QueueNotFound", true)
|
|
}
|
|
defer queue.UnlockMu()
|
|
|
|
// AWS: PurgeQueue допускается не чаще одного раза в 60 секунд на очередь.
|
|
// Нулевой LastPurgeTime (purge ещё не выполнялся) → запрет не срабатывает.
|
|
if time.Since(queue.LastPurgeTime) < 60*time.Second {
|
|
log.Warnf("Purge Queue: %s rejected — purge in progress (last: %s)", queueName, queue.LastPurgeTime.Format(time.RFC3339))
|
|
return utils.CreateErrorResponseV1("PurgeQueueInProgress", true)
|
|
}
|
|
|
|
log.Infof("Purging Queue: %s (tenant: %s)", queueName, t.ID)
|
|
queue.Reset()
|
|
queue.Duplicates = make(map[string]time.Time)
|
|
// Фиксируем время purge — защита от повторного вызова в течение 60 секунд.
|
|
queue.LastPurgeTime = time.Now()
|
|
// Удаляем все сообщения из Redis одной командой DEL
|
|
persistence.PurgeMessagesPersist(key)
|
|
// Сохраняем пустые метаданные очереди
|
|
persistence.SaveQueue(key, queue)
|
|
|
|
respStruct := models.PurgeQueueResponse{
|
|
Xmlns: models.BaseXmlns,
|
|
Metadata: models.BaseResponseMetadata,
|
|
}
|
|
return http.StatusOK, respStruct
|
|
}
|