Clean up the code.

api/http/handler/endpoints/endpoint_update.go

@@ -175,15 +175,11 @@ func (handler *Handler) endpointUpdate(w http.ResponseWriter, r *http.Request) *
 	}
 
 	if payload.Status != nil {
-		switch *payload.Status {
-		case 1:
+		switch portainer.EndpointStatus(*payload.Status) {
+		case portainer.EndpointStatusUp:
 			endpoint.Status = portainer.EndpointStatusUp
-			break
-		case 2:
+		case portainer.EndpointStatusDown:
 			endpoint.Status = portainer.EndpointStatusDown
-			break
-		default:
-			break
 		}
 	}
 
@@ -309,6 +305,13 @@ func (handler *Handler) endpointUpdate(w http.ResponseWriter, r *http.Request) *
 			edgeStackSet[edgeStackID] = true
 		}
 
+		for _, stack := range edgeStacks {
+			if !edgeStackSet[stack.ID] {
+				delete(stack.Status, endpoint.ID)
+				handler.DataStore.EdgeStack().UpdateEdgeStack(stack.ID, &stack)
+			}
+		}
+
 		relation.EdgeStacks = edgeStackSet
 
 		err = handler.DataStore.EndpointRelation().UpdateEndpointRelation(endpoint.ID, relation)