feat(master): add socket

tranthaison1231 · Nov 23, 2023 · 9735e36 · 9735e36 · vercel · Nov 23, 2023
1 parent 2769560
commit 9735e36
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/apps/api/ws/main.go b/apps/api/ws/main.go
@@ -11,8 +11,7 @@ import (
 )
 
 type Body struct {
-	action  string
-	payload interface{}
+	Action string `json:"action"`
 }
 
 func Handler(ctx context.Context, event events.APIGatewayWebsocketProxyRequest) (events.APIGatewayProxyResponse, error) {
@@ -21,7 +20,8 @@ func Handler(ctx context.Context, event events.APIGatewayWebsocketProxyRequest)
 		return events.APIGatewayProxyResponse{}, err
 	}
 
-	body := Body{}
+	var body Body
+
 	err = json.Unmarshal([]byte(event.Body), &body)
 	if err != nil {
 		return events.APIGatewayProxyResponse{}, err
@@ -34,8 +34,8 @@ func Handler(ctx context.Context, event events.APIGatewayWebsocketProxyRequest)
 		return handlers.Disconnect(ctx, event)
 	case "$default":
 		fmt.Print(event.Body)
-		fmt.Println(body.action)
-		switch body.action {
+		fmt.Println(body.Action)
+		switch body.Action {
 		case "SEND_MESSAGE":
 			return handlers.SendMessageSocket(ctx, event)
 		default: