mirror of https://github.com/portainer/portainer
85 lines
2.7 KiB
Go
85 lines
2.7 KiB
Go
package edgegroups
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
|
|
"github.com/asaskevich/govalidator"
|
|
httperror "github.com/portainer/libhttp/error"
|
|
"github.com/portainer/libhttp/request"
|
|
"github.com/portainer/libhttp/response"
|
|
"github.com/portainer/portainer/api"
|
|
)
|
|
|
|
type edgeGroupCreatePayload struct {
|
|
Name string
|
|
Dynamic bool
|
|
TagIDs []portainer.TagID
|
|
Endpoints []portainer.EndpointID
|
|
PartialMatch bool
|
|
}
|
|
|
|
func (payload *edgeGroupCreatePayload) Validate(r *http.Request) error {
|
|
if govalidator.IsNull(payload.Name) {
|
|
return errors.New("Invalid Edge group name")
|
|
}
|
|
if payload.Dynamic && (payload.TagIDs == nil || len(payload.TagIDs) == 0) {
|
|
return errors.New("TagIDs is mandatory for a dynamic Edge group")
|
|
}
|
|
if !payload.Dynamic && (payload.Endpoints == nil || len(payload.Endpoints) == 0) {
|
|
return errors.New("Endpoints is mandatory for a static Edge group")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (handler *Handler) edgeGroupCreate(w http.ResponseWriter, r *http.Request) *httperror.HandlerError {
|
|
var payload edgeGroupCreatePayload
|
|
err := request.DecodeAndValidateJSONPayload(r, &payload)
|
|
if err != nil {
|
|
return &httperror.HandlerError{http.StatusBadRequest, "Invalid request payload", err}
|
|
}
|
|
|
|
edgeGroups, err := handler.DataStore.EdgeGroup().EdgeGroups()
|
|
if err != nil {
|
|
return &httperror.HandlerError{http.StatusInternalServerError, "Unable to retrieve Edge groups from the database", err}
|
|
}
|
|
|
|
for _, edgeGroup := range edgeGroups {
|
|
if edgeGroup.Name == payload.Name {
|
|
return &httperror.HandlerError{http.StatusBadRequest, "Edge group name must be unique", errors.New("Edge group name must be unique")}
|
|
}
|
|
}
|
|
|
|
edgeGroup := &portainer.EdgeGroup{
|
|
Name: payload.Name,
|
|
Dynamic: payload.Dynamic,
|
|
TagIDs: []portainer.TagID{},
|
|
Endpoints: []portainer.EndpointID{},
|
|
PartialMatch: payload.PartialMatch,
|
|
}
|
|
|
|
if edgeGroup.Dynamic {
|
|
edgeGroup.TagIDs = payload.TagIDs
|
|
} else {
|
|
endpointIDs := []portainer.EndpointID{}
|
|
for _, endpointID := range payload.Endpoints {
|
|
endpoint, err := handler.DataStore.Endpoint().Endpoint(endpointID)
|
|
if err != nil {
|
|
return &httperror.HandlerError{http.StatusInternalServerError, "Unable to retrieve endpoint from the database", err}
|
|
}
|
|
|
|
if endpoint.Type == portainer.EdgeAgentOnDockerEnvironment || endpoint.Type == portainer.EdgeAgentOnKubernetesEnvironment {
|
|
endpointIDs = append(endpointIDs, endpoint.ID)
|
|
}
|
|
}
|
|
edgeGroup.Endpoints = endpointIDs
|
|
}
|
|
|
|
err = handler.DataStore.EdgeGroup().CreateEdgeGroup(edgeGroup)
|
|
if err != nil {
|
|
return &httperror.HandlerError{http.StatusInternalServerError, "Unable to persist the Edge group inside the database", err}
|
|
}
|
|
|
|
return response.JSON(w, edgeGroup)
|
|
}
|