mirror of https://github.com/portainer/portainer
46 lines
1.3 KiB
Go
46 lines
1.3 KiB
Go
package endpointgroups
|
|
|
|
import (
|
|
portainer "github.com/portainer/portainer/api"
|
|
"github.com/portainer/portainer/api/internal/edge"
|
|
)
|
|
|
|
func (handler *Handler) updateEndpointRelations(endpoint *portainer.Endpoint, endpointGroup *portainer.EndpointGroup) error {
|
|
if endpoint.Type != portainer.EdgeAgentOnKubernetesEnvironment && endpoint.Type != portainer.EdgeAgentOnDockerEnvironment {
|
|
return nil
|
|
}
|
|
|
|
if endpointGroup == nil {
|
|
unassignedGroup, err := handler.DataStore.EndpointGroup().EndpointGroup(portainer.EndpointGroupID(1))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
endpointGroup = unassignedGroup
|
|
}
|
|
|
|
endpointRelation, err := handler.DataStore.EndpointRelation().EndpointRelation(endpoint.ID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
edgeGroups, err := handler.DataStore.EdgeGroup().EdgeGroups()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
edgeStacks, err := handler.DataStore.EdgeStack().EdgeStacks()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
endpointStacks := edge.EndpointRelatedEdgeStacks(endpoint, endpointGroup, edgeGroups, edgeStacks)
|
|
stacksSet := map[portainer.EdgeStackID]bool{}
|
|
for _, edgeStackID := range endpointStacks {
|
|
stacksSet[edgeStackID] = true
|
|
}
|
|
endpointRelation.EdgeStacks = stacksSet
|
|
|
|
return handler.DataStore.EndpointRelation().UpdateEndpointRelation(endpoint.ID, endpointRelation)
|
|
}
|