From af05d59196e0062d3919ec25793dbc4ef998c6a3 Mon Sep 17 00:00:00 2001
From: Soner Sayakci <s.sayakci@shopware.com>
Date: Thu, 21 Sep 2023 19:31:01 +0000
Subject: [PATCH] fix: worker running into memory limit

---
 cmd/project/project_worker.go | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/cmd/project/project_worker.go b/cmd/project/project_worker.go
index 8d39789d..3fdc86f4 100644
--- a/cmd/project/project_worker.go
+++ b/cmd/project/project_worker.go
@@ -2,7 +2,7 @@ package project
 
 import (
 	"context"
-	"github.com/FriendsOfShopware/shopware-cli/shop"
+	"fmt"
 	"os"
 	"os/exec"
 	"os/signal"
@@ -11,6 +11,8 @@ import (
 	"sync"
 	"syscall"
 
+	"github.com/FriendsOfShopware/shopware-cli/shop"
+
 	"github.com/spf13/cobra"
 
 	"github.com/FriendsOfShopware/shopware-cli/logging"
@@ -26,6 +28,7 @@ var projectWorkerCmd = &cobra.Command{
 
 		isVerbose, _ := cobraCmd.Flags().GetBool("verbose")
 		queuesToConsume, _ := cobraCmd.Flags().GetString("queue")
+		memoryLimit, _ := cobraCmd.Flags().GetString("memory-limit")
 
 		if projectRoot, err = findClosestShopwareProject(); err != nil {
 			return err
@@ -39,10 +42,14 @@ var projectWorkerCmd = &cobra.Command{
 			}
 		}
 
+		if memoryLimit == "" {
+			memoryLimit = "512M"
+		}
+
 		cancelCtx, cancel := context.WithCancel(cobraCmd.Context())
 		cancelOnTermination(cancelCtx, cancel)
 
-		consumeArgs := []string{"bin/console", "messenger:consume"}
+		consumeArgs := []string{"bin/console", "messenger:consume", fmt.Sprintf("--memory-limit=%s", memoryLimit)}
 
 		if queuesToConsume == "" {
 			if is, _ := shop.IsShopwareVersion(projectRoot, ">=6.5"); is {
@@ -83,6 +90,7 @@ func init() {
 	projectRootCmd.AddCommand(projectWorkerCmd)
 	projectWorkerCmd.PersistentFlags().Bool("verbose", false, "Enable verbose output")
 	projectWorkerCmd.PersistentFlags().String("queue", "", "Queues to consume")
+	projectWorkerCmd.PersistentFlags().String("memory-limit", "", "Memory Limit")
 }
 
 func cancelOnTermination(ctx context.Context, cancel context.CancelFunc) {