From f64860431a54d609198e2d1c9f5a8e9a6030c621 Mon Sep 17 00:00:00 2001
From: Assaf Giladi <assaf@weka.io>
Date: Wed, 19 Nov 2025 15:49:42 +0200
Subject: [PATCH] feat: add support for cgroups mode for data services

---
 deploy/models.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/deploy/models.go b/deploy/models.go
index e19c489..8cc6625 100644
--- a/deploy/models.go
+++ b/deploy/models.go
@@ -50,7 +50,7 @@ func (d *DeployScriptGenerator) GetWekaInstallScript() string {
 	`
 
 	cgroupsMode := "auto"
-	if d.Params.CgroupsMode != "" && d.Params.Protocol == "" {
+	if d.Params.CgroupsMode != "" && (d.Params.Protocol == "" || d.Params.Protocol == protocol.DATA) {
 		cgroupsMode = d.Params.CgroupsMode
 	}