-
Notifications
You must be signed in to change notification settings - Fork 147
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add GetQueueAttributesV1 for JSON support
- Loading branch information
1 parent
558e44a
commit 096af9a
Showing
18 changed files
with
911 additions
and
323 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,137 @@ | ||
| package gosqs | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "net/http" | ||
| "strconv" | ||
| "strings" | ||
|
|
||
| "github.com/Admiral-Piett/goaws/app" | ||
| "github.com/Admiral-Piett/goaws/app/models" | ||
| "github.com/Admiral-Piett/goaws/app/utils" | ||
| "github.com/mitchellh/copystructure" | ||
|
|
||
| "github.com/Admiral-Piett/goaws/app/interfaces" | ||
|
|
||
| log "github.com/sirupsen/logrus" | ||
| ) | ||
|
|
||
| func GetQueueAttributesV1(req *http.Request) (int, interfaces.AbstractResponseBody) { | ||
| requestBody := models.NewGetQueueAttributesRequest() | ||
| ok := utils.REQUEST_TRANSFORMER(requestBody, req) | ||
| if !ok { | ||
| log.Error("Invalid Request - GetQueueAttributesV1") | ||
| return createErrorResponseV1(ErrInvalidParameterValue.Type) | ||
| } | ||
| if requestBody.QueueUrl == "" { | ||
| log.Error("Missing QueueUrl - GetQueueAttributesV1") | ||
| return createErrorResponseV1(ErrInvalidParameterValue.Type) | ||
| } | ||
|
|
||
| requestedAttributes := func() map[string]bool { | ||
| attrs := map[string]bool{} | ||
| if len(requestBody.AttributeNames) == 0 { | ||
| return map[string]bool{"All": true} | ||
| } | ||
| for _, attr := range requestBody.AttributeNames { | ||
| if "All" == attr { | ||
| return map[string]bool{"All": true} | ||
| } | ||
| attrs[attr] = true | ||
| } | ||
| return attrs | ||
| }() | ||
|
|
||
| dupe, _ := copystructure.Copy(models.AVAILABLE_QUEUE_ATTRIBUTES) | ||
| includedAttributes, _ := dupe.(map[string]bool) | ||
| _, ok = requestedAttributes["All"] | ||
| if !ok { | ||
| for attr, _ := range includedAttributes { | ||
| _, ok := requestedAttributes[attr] | ||
| if !ok { | ||
| delete(includedAttributes, attr) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| uriSegments := strings.Split(requestBody.QueueUrl, "/") | ||
| queueName := uriSegments[len(uriSegments)-1] | ||
|
|
||
| log.Infof("Get Queue Attributes: %s", queueName) | ||
| queueAttributes := make([]models.Attribute, 0, 0) | ||
|
|
||
| app.SyncQueues.RLock() | ||
| queue, ok := app.SyncQueues.Queues[queueName] | ||
| if !ok { | ||
| log.Errorf("Get Queue URL: %s queue does not exist!!!", queueName) | ||
| app.SyncQueues.RUnlock() | ||
| return createErrorResponseV1(ErrInvalidParameterValue.Type) | ||
| } | ||
|
|
||
| if _, ok := includedAttributes["DelaySeconds"]; ok { | ||
| attr := models.Attribute{Name: "DelaySeconds", Value: strconv.Itoa(queue.DelaySeconds)} | ||
| queueAttributes = append(queueAttributes, attr) | ||
| } | ||
| if _, ok := includedAttributes["MaximumMessageSize"]; ok { | ||
| attr := models.Attribute{Name: "MaximumMessageSize", Value: strconv.Itoa(queue.MaximumMessageSize)} | ||
| queueAttributes = append(queueAttributes, attr) | ||
| } | ||
| if _, ok := includedAttributes["MessageRetentionPeriod"]; ok { | ||
| attr := models.Attribute{Name: "MessageRetentionPeriod", Value: strconv.Itoa(queue.MessageRetentionPeriod)} | ||
| queueAttributes = append(queueAttributes, attr) | ||
| } | ||
| if _, ok := includedAttributes["ReceiveMessageWaitTimeSeconds"]; ok { | ||
| attr := models.Attribute{Name: "ReceiveMessageWaitTimeSeconds", Value: strconv.Itoa(queue.ReceiveMessageWaitTimeSeconds)} | ||
| queueAttributes = append(queueAttributes, attr) | ||
| } | ||
| if _, ok := includedAttributes["VisibilityTimeout"]; ok { | ||
| attr := models.Attribute{Name: "VisibilityTimeout", Value: strconv.Itoa(queue.VisibilityTimeout)} | ||
| queueAttributes = append(queueAttributes, attr) | ||
| } | ||
| if _, ok := includedAttributes["ApproximateNumberOfMessages"]; ok { | ||
| attr := models.Attribute{Name: "ApproximateNumberOfMessages", Value: strconv.Itoa(len(queue.Messages))} | ||
| queueAttributes = append(queueAttributes, attr) | ||
| } | ||
| // TODO - implement | ||
| //if _, ok := includedAttributes["ApproximateNumberOfMessagesDelayed"]; ok { | ||
| // attr := models.Attribute{Name: "ApproximateNumberOfMessagesDelayed", Value: strconv.Itoa(len(queue.Messages))} | ||
| // queueAttributes = append(queueAttributes, attr) | ||
| //} | ||
| if _, ok := includedAttributes["ApproximateNumberOfMessagesNotVisible"]; ok { | ||
| attr := models.Attribute{Name: "ApproximateNumberOfMessagesNotVisible", Value: strconv.Itoa(numberOfHiddenMessagesInQueue(*queue))} | ||
| queueAttributes = append(queueAttributes, attr) | ||
| } | ||
| if _, ok := includedAttributes["CreatedTimestamp"]; ok { | ||
| attr := models.Attribute{Name: "CreatedTimestamp", Value: "0000000000"} | ||
| queueAttributes = append(queueAttributes, attr) | ||
| } | ||
| if _, ok := includedAttributes["LastModifiedTimestamp"]; ok { | ||
| attr := models.Attribute{Name: "LastModifiedTimestamp", Value: "0000000000"} | ||
| queueAttributes = append(queueAttributes, attr) | ||
| } | ||
| if _, ok := includedAttributes["QueueArn"]; ok { | ||
| attr := models.Attribute{Name: "QueueArn", Value: queue.Arn} | ||
| queueAttributes = append(queueAttributes, attr) | ||
| } | ||
| // TODO - implement | ||
| //if _, ok := includedAttributes["Policy"]; ok { | ||
| // attr := models.Attribute{Name: "Policy", Value: ""} | ||
| // queueAttributes = append(queueAttributes, attr) | ||
| //} | ||
| //if _, ok := includedAttributes["RedriveAllowPolicy"]; ok { | ||
| // attr := models.Attribute{Name: "RedriveAllowPolicy", Value: ""} | ||
| // queueAttributes = append(queueAttributes, attr) | ||
| //} | ||
| if _, ok := includedAttributes["RedrivePolicy"]; ok && queue.DeadLetterQueue != nil { | ||
| attr := models.Attribute{Name: "RedrivePolicy", Value: fmt.Sprintf(`{"maxReceiveCount":"%d", "deadLetterTargetArn":"%s"}`, queue.MaxReceiveCount, queue.DeadLetterQueue.Arn)} | ||
| queueAttributes = append(queueAttributes, attr) | ||
| } | ||
| app.SyncQueues.RUnlock() | ||
|
|
||
| respStruct := models.GetQueueAttributesResponse{ | ||
| models.BASE_XMLNS, | ||
| models.GetQueueAttributesResult{Attrs: queueAttributes}, | ||
| models.BASE_RESPONSE_METADATA, | ||
| } | ||
| return http.StatusOK, respStruct | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,178 @@ | ||
| package gosqs | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "net/http" | ||
| "testing" | ||
|
|
||
| "github.com/mitchellh/copystructure" | ||
|
|
||
| "github.com/Admiral-Piett/goaws/app/conf" | ||
|
|
||
| "github.com/Admiral-Piett/goaws/app/fixtures" | ||
| "github.com/Admiral-Piett/goaws/app/interfaces" | ||
| "github.com/Admiral-Piett/goaws/app/models" | ||
| "github.com/Admiral-Piett/goaws/app/utils" | ||
| "github.com/stretchr/testify/assert" | ||
| ) | ||
|
|
||
| func TestGetQueueAttributesV1_success_all(t *testing.T) { | ||
| conf.LoadYamlConfig("../conf/mock-data/mock-config.yaml", "BaseUnitTests") | ||
| defer func() { | ||
| utils.ResetApp() | ||
| utils.REQUEST_TRANSFORMER = utils.TransformRequest | ||
| }() | ||
|
|
||
| utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request) (success bool) { | ||
| v := resultingStruct.(*models.GetQueueAttributesRequest) | ||
| *v = fixtures.GetQueueAttributesRequest | ||
| return true | ||
| } | ||
|
|
||
| _, r := utils.GenerateRequestInfo("POST", "/", nil, true) | ||
| code, response := GetQueueAttributesV1(r) | ||
|
|
||
| assert.Equal(t, http.StatusOK, code) | ||
| assert.Equal(t, fixtures.GetQueueAttributesResponse, response) | ||
| } | ||
|
|
||
| func TestGetQueueAttributesV1_success_no_request_attrs_returns_all(t *testing.T) { | ||
| conf.LoadYamlConfig("../conf/mock-data/mock-config.yaml", "BaseUnitTests") | ||
| defer func() { | ||
| utils.ResetApp() | ||
| utils.REQUEST_TRANSFORMER = utils.TransformRequest | ||
| }() | ||
|
|
||
| utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request) (success bool) { | ||
| v := resultingStruct.(*models.GetQueueAttributesRequest) | ||
| *v = models.GetQueueAttributesRequest{ | ||
| QueueUrl: "unit-queue1", | ||
| } | ||
| return true | ||
| } | ||
|
|
||
| _, r := utils.GenerateRequestInfo("POST", "/", nil, true) | ||
| code, response := GetQueueAttributesV1(r) | ||
|
|
||
| assert.Equal(t, http.StatusOK, code) | ||
| assert.Equal(t, fixtures.GetQueueAttributesResponse, response) | ||
| } | ||
|
|
||
| func TestGetQueueAttributesV1_success_all_with_redrive_queue(t *testing.T) { | ||
| conf.LoadYamlConfig("../conf/mock-data/mock-config.yaml", "BaseUnitTests") | ||
| defer func() { | ||
| utils.ResetApp() | ||
| utils.REQUEST_TRANSFORMER = utils.TransformRequest | ||
| }() | ||
|
|
||
| utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request) (success bool) { | ||
| v := resultingStruct.(*models.GetQueueAttributesRequest) | ||
| *v = models.GetQueueAttributesRequest{ | ||
| QueueUrl: "unit-queue2", | ||
| AttributeNames: []string{"All"}, | ||
| } | ||
| return true | ||
| } | ||
|
|
||
| _, r := utils.GenerateRequestInfo("POST", "/", nil, true) | ||
| code, response := GetQueueAttributesV1(r) | ||
|
|
||
| dupe, _ := copystructure.Copy(fixtures.GetQueueAttributesResponse) | ||
| expectedResponse, _ := dupe.(models.GetQueueAttributesResponse) | ||
| expectedResponse.Result.Attrs[9].Value = fmt.Sprintf("%s:%s", fixtures.BASE_ARN, "unit-queue2") | ||
| expectedResponse.Result.Attrs = append(expectedResponse.Result.Attrs, | ||
| models.Attribute{ | ||
| Name: "RedrivePolicy", | ||
| Value: fmt.Sprintf(`{"maxReceiveCount":"100", "deadLetterTargetArn":"%s:%s"}`, fixtures.BASE_ARN, "other-queue1"), | ||
| }, | ||
| ) | ||
|
|
||
| assert.Equal(t, http.StatusOK, code) | ||
| assert.Equal(t, expectedResponse, response) | ||
| } | ||
|
|
||
| func TestGetQueueAttributesV1_success_specific_fields(t *testing.T) { | ||
| conf.LoadYamlConfig("../conf/mock-data/mock-config.yaml", "BaseUnitTests") | ||
| defer func() { | ||
| utils.ResetApp() | ||
| utils.REQUEST_TRANSFORMER = utils.TransformRequest | ||
| }() | ||
|
|
||
| utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request) (success bool) { | ||
| v := resultingStruct.(*models.GetQueueAttributesRequest) | ||
| *v = models.GetQueueAttributesRequest{ | ||
| QueueUrl: fmt.Sprintf("%s/unit-queue1", fixtures.BASE_URL), | ||
| AttributeNames: []string{"DelaySeconds"}, | ||
| } | ||
| return true | ||
| } | ||
|
|
||
| _, r := utils.GenerateRequestInfo("POST", "/", nil, true) | ||
| code, response := GetQueueAttributesV1(r) | ||
|
|
||
| expectedResponse := models.GetQueueAttributesResponse{ | ||
| Xmlns: models.BASE_XMLNS, | ||
| Result: models.GetQueueAttributesResult{Attrs: []models.Attribute{ | ||
| models.Attribute{ | ||
| Name: "DelaySeconds", | ||
| Value: "0", | ||
| }, | ||
| }}, | ||
| Metadata: models.BASE_RESPONSE_METADATA, | ||
| } | ||
|
|
||
| assert.Equal(t, http.StatusOK, code) | ||
| assert.Equal(t, expectedResponse, response) | ||
| } | ||
|
|
||
| func TestGetQueueAttributesV1_request_transformer_error(t *testing.T) { | ||
| defer func() { | ||
| utils.REQUEST_TRANSFORMER = utils.TransformRequest | ||
| }() | ||
|
|
||
| utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request) (success bool) { | ||
| return false | ||
| } | ||
|
|
||
| _, r := utils.GenerateRequestInfo("POST", "/", nil, true) | ||
| code, _ := GetQueueAttributesV1(r) | ||
|
|
||
| assert.Equal(t, http.StatusBadRequest, code) | ||
| } | ||
|
|
||
| func TestGetQueueAttributesV1_missing_queue_url_in_request_returns_error(t *testing.T) { | ||
| defer func() { | ||
| utils.REQUEST_TRANSFORMER = utils.TransformRequest | ||
| }() | ||
|
|
||
| utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request) (success bool) { | ||
| v := resultingStruct.(*models.GetQueueAttributesRequest) | ||
| *v = models.GetQueueAttributesRequest{ | ||
| QueueUrl: "", | ||
| AttributeNames: []string{}, | ||
| } | ||
| return true | ||
| } | ||
|
|
||
| _, r := utils.GenerateRequestInfo("POST", "/", nil, true) | ||
| code, _ := GetQueueAttributesV1(r) | ||
|
|
||
| assert.Equal(t, http.StatusBadRequest, code) | ||
| } | ||
|
|
||
| func TestGetQueueAttributesV1_missing_queue_returns_error(t *testing.T) { | ||
| defer func() { | ||
| utils.REQUEST_TRANSFORMER = utils.TransformRequest | ||
| }() | ||
|
|
||
| utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request) (success bool) { | ||
| v := resultingStruct.(*models.GetQueueAttributesRequest) | ||
| *v = fixtures.GetQueueAttributesRequest | ||
| return true | ||
| } | ||
|
|
||
| _, r := utils.GenerateRequestInfo("POST", "/", nil, true) | ||
| code, _ := GetQueueAttributesV1(r) | ||
|
|
||
| assert.Equal(t, http.StatusBadRequest, code) | ||
| } |
Oops, something went wrong.