Add endpoint persistence in configuration

eth-easl · Apr 23, 2024 · 7b18915 · 7b18915
1 parent 975327b
commit 7b18915
Show file tree

Hide file tree

Showing 10 changed files with 19 additions and 2 deletions.
diff --git a/cmd/master_node/config.yaml b/cmd/master_node/config.yaml
@@ -26,3 +26,5 @@ redis:
   password: ""
   db: 0
   fullPersistence: true
+
+endpointPersistence: false
diff --git a/cmd/master_node/config_cluster.yaml b/cmd/master_node/config_cluster.yaml
@@ -26,4 +26,6 @@ redis:
   replicas: ""
   password: ""
   db: 0
-  fullPersistence: true
+  fullPersistence: true
+
+endpointPersistence: false
diff --git a/cmd/master_node/config_cluster_raft_1.yaml b/cmd/master_node/config_cluster_raft_1.yaml
@@ -27,3 +27,5 @@ redis:
   password: ""
   db: 0
   fullPersistence: true
+
+endpointPersistence: false
diff --git a/cmd/master_node/config_cluster_raft_2.yaml b/cmd/master_node/config_cluster_raft_2.yaml
@@ -27,3 +27,5 @@ redis:
   password: ""
   db: 0
   fullPersistence: true
+
+endpointPersistence: false
diff --git a/cmd/master_node/config_cluster_raft_3.yaml b/cmd/master_node/config_cluster_raft_3.yaml
@@ -27,3 +27,5 @@ redis:
   password: ""
   db: 0
   fullPersistence: true
+
+endpointPersistence: false
diff --git a/cmd/master_node/config_raft_1.yaml b/cmd/master_node/config_raft_1.yaml
@@ -27,3 +27,5 @@ redis:
   password: ""
   db: 0
   fullPersistence: true
+
+endpointPersistence: false
diff --git a/cmd/master_node/config_raft_2.yaml b/cmd/master_node/config_raft_2.yaml
@@ -27,3 +27,5 @@ redis:
   password: ""
   db: 0
   fullPersistence: true
+
+endpointPersistence: false
diff --git a/cmd/master_node/config_raft_3.yaml b/cmd/master_node/config_raft_3.yaml
@@ -27,3 +27,5 @@ redis:
   password: ""
   db: 0
   fullPersistence: true
+
+endpointPersistence: false
diff --git a/internal/control_plane/scaling_loop.go b/internal/control_plane/scaling_loop.go
@@ -134,7 +134,7 @@ func (ss *ServiceInfoStorage) doUpscaling(toCreateCount int, loopStarted time.Ti
 
 			costPersistStart := time.Now()
 
-			if false {
+			if ss.ControlPlane.Config.EndpointPersistence {
 				ss.ControlPlane.PersistenceLayer.StoreEndpoint(context.Background(), core.Endpoint{
 					SandboxID: resp.ID,
 					URL:       fmt.Sprintf("%s:%d", node.GetIP(), resp.PortMappings.HostPort),

diff --git a/pkg/config/config.go b/pkg/config/config.go
@@ -25,6 +25,7 @@ type ControlPlaneConfig struct {
 	RemoveWorkerNode           bool           `mapstructure:"removeWorkerNode"`
 	RemoveDataplane            bool           `mapstructure:"removeDataplane"`
 	PrecreateSnapshots         bool           `mapstructure:"precreateSnapshots"`
+	EndpointPersistence        bool           `mapstructure:"endpointPersistence"`
 }
 
 type DataPlaneConfig struct {