normaliseeol

api/queue/document.go

@@ -1,53 +1,53 @@
-package queue
-
-import (
-	"context"
-	"encoding/json"
-	"gotemplate/internal/document"
-	"gotemplate/internal/queue"
-
-	"github.com/aws/aws-sdk-go-v2/service/sqs/types"
-)
-
-type DocumentController struct {
-	document document.Service
-}
-
-func NewDocumentController(svc document.Service) *DocumentController {
-	return &DocumentController{
-		document: svc,
-	}
-}
-
-type DocumentQueryEvent struct {
-	ID string `json:"id"`
-}
-
-func (s *DocumentController) Sync(ctx context.Context, config *queue.QueueConfig, msg *types.Message) error {
-	var body document.Document
-	err := json.Unmarshal([]byte(*msg.Body), &body)
-	if err != nil {
-		return err
-	}
-
-	err = s.document.Sync(ctx, body)
-	if err != nil {
-		return err
-	}
-
-	queryEvent := DocumentQueryEvent{
-		ID: body.ID,
-	}
-
-	err = queue.Send(ctx, config, "DOCQUERY", queryEvent)
-	if err != nil {
-		return err
-	}
-
-	err = queue.Delete(ctx, config, msg)
-	if err != nil {
-		return err
-	}
-
-	return nil
-}
+package queue
+
+import (
+	"context"
+	"encoding/json"
+	"gotemplate/internal/document"
+	"gotemplate/internal/queue"
+
+	"github.com/aws/aws-sdk-go-v2/service/sqs/types"
+)
+
+type DocumentController struct {
+	document document.Service
+}
+
+func NewDocumentController(svc document.Service) *DocumentController {
+	return &DocumentController{
+		document: svc,
+	}
+}
+
+type DocumentQueryEvent struct {
+	ID string `json:"id"`
+}
+
+func (s *DocumentController) Sync(ctx context.Context, config *queue.QueueConfig, msg *types.Message) error {
+	var body document.Document
+	err := json.Unmarshal([]byte(*msg.Body), &body)
+	if err != nil {
+		return err
+	}
+
+	err = s.document.Sync(ctx, body)
+	if err != nil {
+		return err
+	}
+
+	queryEvent := DocumentQueryEvent{
+		ID: body.ID,
+	}
+
+	err = queue.Send(ctx, config, "DOCQUERY", queryEvent)
+	if err != nil {
+		return err
+	}
+
+	err = queue.Delete(ctx, config, msg)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}

api/queue/poll.go

@@ -1,48 +1,48 @@
-package queue
-
-import (
-	"context"
-	"gotemplate/internal/queue"
-	"log"
-
-	"github.com/aws/aws-sdk-go-v2/service/sqs"
-)
-
-type Controllers struct {
-	Document DocumentController
-}
-
-type Queue struct {
-	Config      *queue.QueueConfig
-	Controllers *Controllers
-}
-
-func PollMessages(ctx context.Context, queue *Queue) {
-	for {
-		result, err := queue.Config.Client.ReceiveMessage(ctx, &sqs.ReceiveMessageInput{
-			QueueUrl:            &queue.Config.URL,
-			MaxNumberOfMessages: 1,
-			WaitTimeSeconds:     2,
-			VisibilityTimeout:   2,
-			MessageAttributeNames: []string{
-				"type",
-			},
-		})
-		if err != nil {
-			log.Printf("Message Fetch Fail: %v", err)
-			continue
-		}
-
-		for _, message := range result.Messages {
-			go func() {
-				toProcess, err := processMessage(ctx, queue, message)
-				if !toProcess {
-					return
-				}
-				if err != nil {
-					log.Printf("Message Process Fail: %v", err)
-				}
-			}()
-		}
-	}
-}
+package queue
+
+import (
+	"context"
+	"gotemplate/internal/queue"
+	"log"
+
+	"github.com/aws/aws-sdk-go-v2/service/sqs"
+)
+
+type Controllers struct {
+	Document DocumentController
+}
+
+type Queue struct {
+	Config      *queue.QueueConfig
+	Controllers *Controllers
+}
+
+func PollMessages(ctx context.Context, queue *Queue) {
+	for {
+		result, err := queue.Config.Client.ReceiveMessage(ctx, &sqs.ReceiveMessageInput{
+			QueueUrl:            &queue.Config.URL,
+			MaxNumberOfMessages: 1,
+			WaitTimeSeconds:     2,
+			VisibilityTimeout:   2,
+			MessageAttributeNames: []string{
+				"type",
+			},
+		})
+		if err != nil {
+			log.Printf("Message Fetch Fail: %v", err)
+			continue
+		}
+
+		for _, message := range result.Messages {
+			go func() {
+				toProcess, err := processMessage(ctx, queue, message)
+				if !toProcess {
+					return
+				}
+				if err != nil {
+					log.Printf("Message Process Fail: %v", err)
+				}
+			}()
+		}
+	}
+}

api/queue/process.go

@@ -1,18 +1,18 @@
-package queue
-
-import (
-	"context"
-
-	"github.com/aws/aws-sdk-go-v2/service/sqs/types"
-)
-
-func processMessage(ctx context.Context, queue *Queue, message types.Message) (bool, error) {
-	// Process the message here and return false if not message to process
-	switch *message.MessageAttributes["type"].StringValue {
-	case "DOCTEXT":
-		err := queue.Controllers.Document.Sync(ctx, queue.Config, &message)
-		return true, err
-	}
-
-	return false, nil
-}
+package queue
+
+import (
+	"context"
+
+	"github.com/aws/aws-sdk-go-v2/service/sqs/types"
+)
+
+func processMessage(ctx context.Context, queue *Queue, message types.Message) (bool, error) {
+	// Process the message here and return false if not message to process
+	switch *message.MessageAttributes["type"].StringValue {
+	case "DOCTEXT":
+		err := queue.Controllers.Document.Sync(ctx, queue.Config, &message)
+		return true, err
+	}
+
+	return false, nil
+}