mirror of https://github.com/portainer/portainer
49 lines
1.6 KiB
Go
49 lines
1.6 KiB
Go
package endpoints
|
|
|
|
import (
|
|
"github.com/pkg/errors"
|
|
portainer "github.com/portainer/portainer/api"
|
|
"github.com/portainer/portainer/api/dataservices"
|
|
"github.com/portainer/portainer/api/internal/edge"
|
|
"github.com/portainer/portainer/api/internal/endpointutils"
|
|
"github.com/portainer/portainer/api/internal/set"
|
|
)
|
|
|
|
// updateEdgeRelations updates the edge stacks associated to an edge endpoint
|
|
func (handler *Handler) updateEdgeRelations(tx dataservices.DataStoreTx, endpoint *portainer.Endpoint) error {
|
|
if !endpointutils.IsEdgeEndpoint(endpoint) {
|
|
return nil
|
|
}
|
|
|
|
relation, err := tx.EndpointRelation().EndpointRelation(endpoint.ID)
|
|
if err != nil {
|
|
return errors.WithMessage(err, "Unable to find environment relation inside the database")
|
|
}
|
|
|
|
endpointGroup, err := tx.EndpointGroup().EndpointGroup(endpoint.GroupID)
|
|
if err != nil {
|
|
return errors.WithMessage(err, "Unable to find environment group inside the database")
|
|
}
|
|
|
|
edgeGroups, err := tx.EdgeGroup().EdgeGroups()
|
|
if err != nil {
|
|
return errors.WithMessage(err, "Unable to retrieve edge groups from the database")
|
|
}
|
|
|
|
edgeStacks, err := tx.EdgeStack().EdgeStacks()
|
|
if err != nil {
|
|
return errors.WithMessage(err, "Unable to retrieve edge stacks from the database")
|
|
}
|
|
|
|
currentEdgeStackSet := set.ToSet(edge.EndpointRelatedEdgeStacks(endpoint, endpointGroup, edgeGroups, edgeStacks))
|
|
|
|
relation.EdgeStacks = currentEdgeStackSet
|
|
|
|
err = tx.EndpointRelation().UpdateEndpointRelation(endpoint.ID, relation)
|
|
if err != nil {
|
|
return errors.WithMessage(err, "Unable to persist environment relation changes inside the database")
|
|
}
|
|
|
|
return nil
|
|
}
|