more robust deletion strategy when removing endpoints (#9191)

pull/9192/head
Matt Hook 1 year ago committed by GitHub
parent afaeddb887
commit cbe23dc753
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -10,6 +10,7 @@ import (
portainer "github.com/portainer/portainer/api" portainer "github.com/portainer/portainer/api"
httperrors "github.com/portainer/portainer/api/http/errors" httperrors "github.com/portainer/portainer/api/http/errors"
"github.com/portainer/portainer/api/internal/endpointutils" "github.com/portainer/portainer/api/internal/endpointutils"
"github.com/rs/zerolog/log"
) )
// @id EndpointDelete // @id EndpointDelete
@ -22,6 +23,7 @@ import (
// @param id path int true "Environment(Endpoint) identifier" // @param id path int true "Environment(Endpoint) identifier"
// @success 204 "Success" // @success 204 "Success"
// @failure 400 "Invalid request" // @failure 400 "Invalid request"
// @failure 403 "Permission denied"
// @failure 404 "Environment(Endpoint) not found" // @failure 404 "Environment(Endpoint) not found"
// @failure 500 "Server error" // @failure 500 "Server error"
// @router /endpoints/{id} [delete] // @router /endpoints/{id} [delete]
@ -39,32 +41,27 @@ func (handler *Handler) endpointDelete(w http.ResponseWriter, r *http.Request) *
if handler.DataStore.IsErrObjectNotFound(err) { if handler.DataStore.IsErrObjectNotFound(err) {
return httperror.NotFound("Unable to find an environment with the specified identifier inside the database", err) return httperror.NotFound("Unable to find an environment with the specified identifier inside the database", err)
} else if err != nil { } else if err != nil {
return httperror.InternalServerError("Unable to find an environment with the specified identifier inside the database", err) return httperror.InternalServerError("Unable to read the environment record from the database", err)
} }
if endpoint.TLSConfig.TLS { if endpoint.TLSConfig.TLS {
folder := strconv.Itoa(endpointID) folder := strconv.Itoa(endpointID)
err = handler.FileService.DeleteTLSFiles(folder) err = handler.FileService.DeleteTLSFiles(folder)
if err != nil { if err != nil {
return httperror.InternalServerError("Unable to remove TLS files from disk", err) log.Error().Err(err).Msgf("Unable to remove TLS files from disk when deleting endpoint %d", endpointID)
} }
} }
err = handler.DataStore.Snapshot().Delete(portainer.EndpointID(endpointID)) err = handler.DataStore.Snapshot().Delete(portainer.EndpointID(endpointID))
if err != nil { if err != nil {
return httperror.InternalServerError("Unable to remove the snapshot from the database", err) log.Warn().Err(err).Msgf("Unable to remove the snapshot from the database")
}
err = handler.DataStore.Endpoint().DeleteEndpoint(portainer.EndpointID(endpointID))
if err != nil {
return httperror.InternalServerError("Unable to remove environment from the database", err)
} }
handler.ProxyManager.DeleteEndpointProxy(endpoint.ID) handler.ProxyManager.DeleteEndpointProxy(endpoint.ID)
err = handler.DataStore.EndpointRelation().DeleteEndpointRelation(endpoint.ID) err = handler.DataStore.EndpointRelation().DeleteEndpointRelation(endpoint.ID)
if err != nil { if err != nil {
return httperror.InternalServerError("Unable to remove environment relation from the database", err) log.Warn().Err(err).Msgf("Unable to remove environment relation from the database")
} }
for _, tagID := range endpoint.TagIDs { for _, tagID := range endpoint.TagIDs {
@ -73,15 +70,15 @@ func (handler *Handler) endpointDelete(w http.ResponseWriter, r *http.Request) *
}) })
if handler.DataStore.IsErrObjectNotFound(err) { if handler.DataStore.IsErrObjectNotFound(err) {
return httperror.NotFound("Unable to find tag inside the database", err) log.Warn().Err(err).Msgf("Unable to find tag inside the database")
} else if err != nil { } else if err != nil {
return httperror.InternalServerError("Unable to persist tag relation inside the database", err) log.Warn().Err(err).Msgf("Unable to delete tag relation from the database")
} }
} }
edgeGroups, err := handler.DataStore.EdgeGroup().ReadAll() edgeGroups, err := handler.DataStore.EdgeGroup().ReadAll()
if err != nil { if err != nil {
return httperror.InternalServerError("Unable to retrieve edge groups from the database", err) log.Warn().Err(err).Msgf("Unable to retrieve edge groups from the database")
} }
for _, edgeGroup := range edgeGroups { for _, edgeGroup := range edgeGroups {
@ -89,13 +86,13 @@ func (handler *Handler) endpointDelete(w http.ResponseWriter, r *http.Request) *
g.Endpoints = removeElement(g.Endpoints, endpoint.ID) g.Endpoints = removeElement(g.Endpoints, endpoint.ID)
}) })
if err != nil { if err != nil {
return httperror.InternalServerError("Unable to update edge group", err) log.Warn().Err(err).Msgf("Unable to update edge group")
} }
} }
edgeStacks, err := handler.DataStore.EdgeStack().EdgeStacks() edgeStacks, err := handler.DataStore.EdgeStack().EdgeStacks()
if err != nil { if err != nil {
return httperror.InternalServerError("Unable to retrieve edge stacks from the database", err) log.Warn().Err(err).Msgf("Unable to retrieve edge stacks from the database")
} }
for idx := range edgeStacks { for idx := range edgeStacks {
@ -104,14 +101,14 @@ func (handler *Handler) endpointDelete(w http.ResponseWriter, r *http.Request) *
delete(edgeStack.Status, endpoint.ID) delete(edgeStack.Status, endpoint.ID)
err = handler.DataStore.EdgeStack().UpdateEdgeStack(edgeStack.ID, edgeStack) err = handler.DataStore.EdgeStack().UpdateEdgeStack(edgeStack.ID, edgeStack)
if err != nil { if err != nil {
return httperror.InternalServerError("Unable to update edge stack", err) log.Warn().Err(err).Msgf("Unable to update edge stack")
} }
} }
} }
registries, err := handler.DataStore.Registry().ReadAll() registries, err := handler.DataStore.Registry().ReadAll()
if err != nil { if err != nil {
return httperror.InternalServerError("Unable to retrieve registries from the database", err) log.Warn().Err(err).Msgf("Unable to retrieve registries from the database")
} }
for idx := range registries { for idx := range registries {
@ -120,33 +117,36 @@ func (handler *Handler) endpointDelete(w http.ResponseWriter, r *http.Request) *
delete(registry.RegistryAccesses, endpoint.ID) delete(registry.RegistryAccesses, endpoint.ID)
err = handler.DataStore.Registry().Update(registry.ID, registry) err = handler.DataStore.Registry().Update(registry.ID, registry)
if err != nil { if err != nil {
return httperror.InternalServerError("Unable to update registry accesses", err) log.Warn().Err(err).Msgf("Unable to update registry accesses")
} }
} }
} }
if !endpointutils.IsEdgeEndpoint(endpoint) { if endpointutils.IsEdgeEndpoint(endpoint) {
return response.Empty(w) edgeJobs, err := handler.DataStore.EdgeJob().ReadAll()
} if err != nil {
log.Warn().Err(err).Msgf("Unable to retrieve edge jobs from the database")
edgeJobs, err := handler.DataStore.EdgeJob().ReadAll() }
if err != nil {
return httperror.InternalServerError("Unable to retrieve edge jobs from the database", err)
}
for idx := range edgeJobs { for idx := range edgeJobs {
edgeJob := &edgeJobs[idx] edgeJob := &edgeJobs[idx]
if _, ok := edgeJob.Endpoints[endpoint.ID]; ok { if _, ok := edgeJob.Endpoints[endpoint.ID]; ok {
err = handler.DataStore.EdgeJob().UpdateEdgeJobFunc(edgeJob.ID, func(j *portainer.EdgeJob) { err = handler.DataStore.EdgeJob().UpdateEdgeJobFunc(edgeJob.ID, func(j *portainer.EdgeJob) {
delete(j.Endpoints, endpoint.ID) delete(j.Endpoints, endpoint.ID)
}) })
if err != nil { if err != nil {
return httperror.InternalServerError("Unable to update edge job", err) log.Warn().Err(err).Msgf("Unable to update edge job")
}
} }
} }
} }
err = handler.DataStore.Endpoint().DeleteEndpoint(portainer.EndpointID(endpointID))
if err != nil {
return httperror.InternalServerError("Unable to remove environment from the database", err)
}
return response.Empty(w) return response.Empty(w)
} }

Loading…
Cancel
Save