shared-sqs: Этап 4 — изоляция очередей по тенанту

2026-04-09 12:53:10 +03:00
parent f4352a17b1
commit 08053ca8e5
16 changed files with 1243 additions and 882 deletions
@@ -1,81 +1,87 @@
+// Изменено: 2026-04-09
+// ChangeMessageVisibilityV1 — меняет visibility timeout сообщения в очереди тенанта.
 package gosqs

 import (
-	"net/http"
-	"strings"
-	"time"
+"net/http"
+"strings"
+"time"

-	"shared-sqs/app/interfaces"
-	"shared-sqs/app/models"
-	"shared-sqs/app/utils"
-	"github.com/gorilla/mux"
-	log "github.com/sirupsen/logrus"
+"shared-sqs/app/interfaces"
+"shared-sqs/app/models"
+"shared-sqs/app/utils"
+"github.com/gorilla/mux"
+log "github.com/sirupsen/logrus"
 )

 func ChangeMessageVisibilityV1(req *http.Request) (int, interfaces.AbstractResponseBody) {
-	requestBody := models.NewChangeMessageVisibilityRequest()
-	ok := utils.REQUEST_TRANSFORMER(requestBody, req, false)
-	if !ok {
-		log.Error("Invalid Request - ChangeMessageVisibilityV1")
-		return utils.CreateErrorResponseV1("InvalidParameterValue", true)
-	}
-
-	vars := mux.Vars(req)
-
-	queueUrl := requestBody.QueueUrl
-	queueName := ""
-	if queueUrl == "" {
-		queueName = vars["queueName"]
-	} else {
-		uriSegments := strings.Split(queueUrl, "/")
-		queueName = uriSegments[len(uriSegments)-1]
-	}
-
-	receiptHandle := requestBody.ReceiptHandle
-
-	visibilityTimeout := requestBody.VisibilityTimeout
-	if visibilityTimeout > 43200 {
-		return utils.CreateErrorResponseV1("ValidationError", true)
-	}
-
-	if _, ok := models.SyncQueues.Queues[queueName]; !ok {
-		return utils.CreateErrorResponseV1("QueueNotFound", true)
-	}
-
-	models.SyncQueues.Lock()
-	messageFound := false
-	for i := 0; i < len(models.SyncQueues.Queues[queueName].Messages); i++ {
-		queue := models.SyncQueues.Queues[queueName]
-		msgs := queue.Messages
-		if msgs[i].ReceiptHandle == receiptHandle {
-			timeout := models.SyncQueues.Queues[queueName].VisibilityTimeout
-			if visibilityTimeout == 0 {
-				msgs[i].ReceiptTime = time.Now().UTC()
-				msgs[i].ReceiptHandle = ""
-				msgs[i].VisibilityTimeout = time.Now().Add(time.Duration(timeout) * time.Second)
-				msgs[i].Retry++
-				if queue.MaxReceiveCount > 0 &&
-					queue.DeadLetterQueue != nil &&
-					msgs[i].Retry >= queue.MaxReceiveCount {
-					queue.DeadLetterQueue.Messages = append(queue.DeadLetterQueue.Messages, msgs[i])
-					queue.Messages = append(queue.Messages[:i], queue.Messages[i+1:]...)
-				}
-			} else {
-				msgs[i].VisibilityTimeout = time.Now().Add(time.Duration(visibilityTimeout) * time.Second)
-			}
-			messageFound = true
-			break
-		}
-	}
-	models.SyncQueues.Unlock()
-	if !messageFound {
-		return utils.CreateErrorResponseV1("MessageNotInFlight", true)
-	}
-
-	respStruct := models.ChangeMessageVisibilityResult{
-		Xmlns:    models.BaseXmlns,
-		Metadata: models.BaseResponseMetadata,
-	}
-
-	return http.StatusOK, &respStruct
+requestBody := models.NewChangeMessageVisibilityRequest()
+ok := utils.REQUEST_TRANSFORMER(requestBody, req, false)
+if !ok {
+log.Error("Invalid Request - ChangeMessageVisibilityV1")
+return utils.CreateErrorResponseV1("InvalidParameterValue", true)
+}
+
+t := getTenantFromContext(req)
+if t == nil {
+return utils.CreateErrorResponseV1("InvalidClientTokenId", true)
+}
+
+vars := mux.Vars(req)
+queueUrl := requestBody.QueueUrl
+queueName := ""
+if queueUrl == "" {
+queueName = vars["queueName"]
+} else {
+uriSegments := strings.Split(queueUrl, "/")
+queueName = uriSegments[len(uriSegments)-1]
+}
+
+key := tenantQueueKey(t.AccessKey, queueName)
+receiptHandle := requestBody.ReceiptHandle
+visibilityTimeout := requestBody.VisibilityTimeout
+
+if visibilityTimeout > 43200 {
+return utils.CreateErrorResponseV1("ValidationError", true)
+}
+
+if _, ok := models.SyncQueues.Queues[key]; !ok {
+return utils.CreateErrorResponseV1("QueueNotFound", true)
+}
+
+models.SyncQueues.Lock()
+messageFound := false
+for i := 0; i < len(models.SyncQueues.Queues[key].Messages); i++ {
+queue := models.SyncQueues.Queues[key]
+msgs := queue.Messages
+if msgs[i].ReceiptHandle == receiptHandle {
+timeout := models.SyncQueues.Queues[key].VisibilityTimeout
+if visibilityTimeout == 0 {
+msgs[i].ReceiptTime = time.Now().UTC()
+msgs[i].ReceiptHandle = ""
+msgs[i].VisibilityTimeout = time.Now().Add(time.Duration(timeout) * time.Second)
+msgs[i].Retry++
+if queue.MaxReceiveCount > 0 &&
+queue.DeadLetterQueue != nil &&
+msgs[i].Retry >= queue.MaxReceiveCount {
+queue.DeadLetterQueue.Messages = append(queue.DeadLetterQueue.Messages, msgs[i])
+queue.Messages = append(queue.Messages[:i], queue.Messages[i+1:]...)
+}
+} else {
+msgs[i].VisibilityTimeout = time.Now().Add(time.Duration(visibilityTimeout) * time.Second)
+}
+messageFound = true
+break
+}
+}
+models.SyncQueues.Unlock()
+if !messageFound {
+return utils.CreateErrorResponseV1("MessageNotInFlight", true)
+}
+
+respStruct := models.ChangeMessageVisibilityResult{
+Xmlns:    models.BaseXmlns,
+Metadata: models.BaseResponseMetadata,
+}
+return http.StatusOK, &respStruct
 }