feat: add sync checker function #4

cmd/ahorro/login/main.go

@@ -83,6 +83,7 @@ func Handler(ctx context.Context, request events.APIGatewayProxyRequest) (events
 	resp := events.APIGatewayV2HTTPResponse{
 		StatusCode:      200,
 		IsBase64Encoded: false,
+		Body:            "logged-in",
 		Headers: map[string]string{
 			"Content-Type": "application/json",
 		},

cmd/ahorro/sync/callback/main.go

@@ -37,7 +37,6 @@ func Handler(ctx context.Context, request events.APIGatewayV2HTTPRequest) (event
 
 	env := config.GetInstance()
 
-	randStateTable := env.DynamoRandTable
 	session, _ := session.NewSession()
 	svc := dynamodb.New(session)
 	input := &dynamodb.GetItemInput{
@@ -46,7 +45,7 @@ func Handler(ctx context.Context, request events.APIGatewayV2HTTPRequest) (event
 				S: aws.String(user.Id.Hex()),
 			},
 		},
-		TableName: aws.String(randStateTable),
+		TableName: aws.String(env.DynamoRandTable),
 	}
 
 	item, err := svc.GetItem(input)

cmd/ahorro/sync/checker/main.go

@@ -0,0 +1,69 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"time"
+
+	"github.com/aws/aws-lambda-go/events"
+	"github.com/aws/aws-lambda-go/lambda"
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/dynamodb"
+	"github.com/noobj/go-serverless-services/internal/config"
+	"github.com/noobj/go-serverless-services/internal/helpers/helper"
+	jwtMiddleWare "github.com/noobj/go-serverless-services/internal/middleware/jwt_auth"
+)
+
+var internalErrorhandler = func() (events.APIGatewayProxyResponse, error) {
+	return helper.GenerateErrorResponse[events.APIGatewayProxyResponse](500)
+}
+
+func Handler(ctx context.Context, request events.APIGatewayV2HTTPRequest) (events.APIGatewayProxyResponse, error) {
+	taskId, taskIdExist := request.QueryStringParameters["taskId"]
+	if !taskIdExist {
+		log.Println("Request has no task id")
+		return helper.GenerateErrorResponse[events.APIGatewayProxyResponse](400, "request query error")
+	}
+
+	env := config.GetInstance()
+
+	session, _ := session.NewSession()
+	svc := dynamodb.New(session)
+	input := &dynamodb.GetItemInput{
+		Key: map[string]*dynamodb.AttributeValue{
+			"TaskId": {
+				S: aws.String(taskId),
+			},
+		},
+		TableName: aws.String(env.DynamoTaskTable),
+	}
+
+	for i := 0; i < 10; i++ {
+		item, err := svc.GetItem(input)
+
+		if err != nil {
+			fmt.Println("Fetch task status error", err)
+			return internalErrorhandler()
+		}
+
+		if item.Item == nil {
+			return helper.GenerateErrorResponse[events.APIGatewayProxyResponse](400, "task id not found")
+		}
+
+		status := *item.Item["Completed"].N
+
+		if status != fmt.Sprint(helper.Pending) {
+			return helper.GenerateApiResponse[events.APIGatewayProxyResponse](status)
+		}
+
+		time.Sleep(time.Second)
+	}
+
+	return helper.GenerateApiResponse[events.APIGatewayProxyResponse](helper.Pending)
+}
+
+func main() {
+	lambda.Start(jwtMiddleWare.Auth(Handler))
+}

cmd/ahorro/sync/receiver/main.go

@@ -18,10 +18,6 @@ import (
 	drive "google.golang.org/api/drive/v3"
 )
 
-type Specification struct {
-	DynamoRandTable string `required:"true" split_words:"true"`
-}
-
 func Handler(ctx context.Context, request events.APIGatewayV2HTTPRequest) (events.APIGatewayProxyResponse, error) {
 	user, ok := helper.GetUserFromContext(ctx)
 	if !ok {

serverless.yml

@@ -121,7 +121,7 @@ functions:
     package:
       include:
         - bin/sync_callback
-  sync_hanlder:
+  sync_handler:
     handler: bin/sync_handler
     events:
       - sqs:
@@ -132,6 +132,15 @@ functions:
     package:
       include:
         - bin/sync_handler
+  sync_checker:
+    handler: bin/sync_checker
+    events:
+      - httpApi:
+          path: /sync/check
+          method: get
+    package:
+      include:
+        - bin/sync_checker
 resources:
   Resources:
     AhorroSyncQueue: