set StatefulSet workloads as stateful services by default

kurtosis-tech · Oct 2, 2024 · 4e601e8 · 4e601e8
1 parent 27c2d0b
commit 4e601e8
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/kontrol-service/engine/docker.go b/kontrol-service/engine/docker.go
@@ -369,10 +369,18 @@ func newClusterTopologyServiceFromServiceConfig(
 		return clusterTopologyService, stacktrace.NewError("Service %s has no workload", serviceName)
 	}
 
+	// Set default for IsStateful to true if the workload is a StatefulSet, otherwise false
+	clusterTopologyService.IsExternal = clusterTopologyService.WorkloadSpec.IsStatefulSet()
+
+	// Override the IsStateful value by manual annotations
 	isStateful, ok := serviceAnnotations["kardinal.dev.service/stateful"]
 	if ok && isStateful == "true" {
 		clusterTopologyService.IsStateful = true
 	}
+	if ok && isStateful == "false" {
+		clusterTopologyService.IsStateful = false
+	}
+
 	isExternal, ok := serviceAnnotations["kardinal.dev.service/external"]
 	if ok && isExternal == "true" {
 		clusterTopologyService.IsExternal = true