From 778bab644f091a4796da6035ffd58988a0c32b71 Mon Sep 17 00:00:00 2001
From: rony5394 <143897221+rony5394@users.noreply.github.com>
Date: Tue, 19 May 2026 14:20:22 +0200
Subject: [PATCH] Switched originalScale storage

---
 docker/docker.go |  2 --
 docker/scale.go  | 17 ++++++++++-------
 2 files changed, 10 insertions(+), 9 deletions(-)

diff --git a/docker/docker.go b/docker/docker.go
index 76cadbd..f2c18ad 100644
--- a/docker/docker.go
+++ b/docker/docker.go
@@ -8,7 +8,6 @@ import (
 	"os"
 	"os/signal"
 	"strings"
-	"sync"
 	"syscall"
 	"time"
 
@@ -23,7 +22,6 @@ import (
 
 // Add mutex.
 var ApiClient *client.Client;
-var scale sync.Map;
 var token string = "12345";
 var theConfig cfg.Config;
 
diff --git a/docker/scale.go b/docker/scale.go
index d08c3fd..f4e2ed2 100644
--- a/docker/scale.go
+++ b/docker/scale.go
@@ -3,7 +3,9 @@ package docker
 import (
 	"context"
 	"io"
+	"log/slog"
 	"net/http"
+	"strconv"
 	"time"
 
 	"encoding/json"
@@ -53,8 +55,7 @@ func scaleDown(w http.ResponseWriter, r *http.Request){
 	newScale := uint64(0);
 	updatedSpec.Mode.Replicated.Replicas = &newScale;
 	updatedSpec.Labels["blazena.scaledDown"] = "true";
-
-	scale.Store(serviceId, *originalScale);
+	updatedSpec.Labels["blazena.originalScale"] = strconv.FormatUint(*originalScale, 10);
 
 	_, err = ApiClient.ServiceUpdate(context.Background(), serviceId, inspectresoult.Version, updatedSpec, swarm.ServiceUpdateOptions{}); 
 	if(err != nil){
@@ -99,19 +100,21 @@ func scaleUp(w http.ResponseWriter, r *http.Request){
 		return;
 	}
 
-	originalScale, ok := scale.Load(serviceId);  
-	if(!ok){
+	originalScale := inspectresoult.Spec.Labels["blazena.originalScale"];
+
+	if(originalScale == ""){
 		panic("Its not okay!");
 	}
 
-	originalScaleChecked, ok := originalScale.(uint64);
-	if(!ok){
-		panic("Its very not okay!")
+	originalScaleChecked, err := strconv.ParseUint(originalScale, 10, 64);
+	if(err != nil){
+		panic("Its very not okay!"+ err.Error())
 	}
 	updatedSpec := inspectresoult.Spec;
 
 	updatedSpec.Mode.Replicated.Replicas = &originalScaleChecked;
 	delete(updatedSpec.Labels, "blazena.scaledDown");
+	delete(updatedSpec.Labels, "blazena.originalScale");
 
 	ApiClient.ServiceUpdate(context.Background(), serviceId, inspectresoult.Version, updatedSpec, swarm.ServiceUpdateOptions{});