Repinoid
fd8ffc329d
Все операции с содержимым очереди (send/receive/delete/visibility/purge/attributes/ tags/admin/gauge/tick) переведены на per-queue RWMutex (QueueStore.GetLocked/ GetRLocked + queue.LockMu). Глобальный SyncQueues-лок остался только для map и неизменяемых полей. DLQ-переносы берут лок DLQ отдельно (порядок main→DLQ). marshalQueueMeta больше не мутирует очередь (DTO-снапшот). Устранён дедлок create_queue (setQueueAttributesV1 вызывается вне глобального лока). Добавлены конкурентные тесты; go test -race зелёный. Снимает боттлнек: синхронный Redis-I/O больше не держит глобальный лок.
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.Messages = nil
|
|
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
|
|
}
|