2017-04-07 01:14:53 -04:00
|
|
|
package builders
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/docker/docker/api/types/swarm"
|
|
|
|
)
|
|
|
|
|
|
|
|
// Service creates a service with default values.
|
|
|
|
// Any number of service builder functions can be passed to augment it.
|
|
|
|
// Currently, only ServiceName is implemented
|
|
|
|
func Service(builders ...func(*swarm.Service)) *swarm.Service {
|
|
|
|
service := &swarm.Service{
|
|
|
|
ID: "serviceID",
|
|
|
|
Spec: swarm.ServiceSpec{
|
|
|
|
Annotations: swarm.Annotations{
|
|
|
|
Name: "defaultServiceName",
|
|
|
|
},
|
2017-06-20 14:00:01 -04:00
|
|
|
EndpointSpec: &swarm.EndpointSpec{},
|
2017-04-07 01:14:53 -04:00
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, builder := range builders {
|
|
|
|
builder(service)
|
|
|
|
}
|
|
|
|
|
|
|
|
return service
|
|
|
|
}
|
|
|
|
|
2017-06-20 14:00:01 -04:00
|
|
|
// ServiceID sets the service ID
|
|
|
|
func ServiceID(ID string) func(*swarm.Service) {
|
|
|
|
return func(service *swarm.Service) {
|
|
|
|
service.ID = ID
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-04-07 01:14:53 -04:00
|
|
|
// ServiceName sets the service name
|
|
|
|
func ServiceName(name string) func(*swarm.Service) {
|
|
|
|
return func(service *swarm.Service) {
|
|
|
|
service.Spec.Annotations.Name = name
|
|
|
|
}
|
|
|
|
}
|
2017-06-20 14:00:01 -04:00
|
|
|
|
|
|
|
// ServiceLabels sets the service's labels
|
|
|
|
func ServiceLabels(labels map[string]string) func(*swarm.Service) {
|
|
|
|
return func(service *swarm.Service) {
|
|
|
|
service.Spec.Annotations.Labels = labels
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-10-27 20:54:22 -04:00
|
|
|
// GlobalService sets the service to use "global" mode
|
|
|
|
func GlobalService() func(*swarm.Service) {
|
|
|
|
return func(service *swarm.Service) {
|
|
|
|
service.Spec.Mode = swarm.ServiceMode{Global: &swarm.GlobalService{}}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// ReplicatedService sets the service to use "replicated" mode with the specified number of replicas
|
2017-06-20 14:00:01 -04:00
|
|
|
func ReplicatedService(replicas uint64) func(*swarm.Service) {
|
|
|
|
return func(service *swarm.Service) {
|
|
|
|
service.Spec.Mode = swarm.ServiceMode{Replicated: &swarm.ReplicatedService{Replicas: &replicas}}
|
2019-10-27 20:54:22 -04:00
|
|
|
if service.ServiceStatus == nil {
|
|
|
|
service.ServiceStatus = &swarm.ServiceStatus{}
|
|
|
|
}
|
|
|
|
service.ServiceStatus.DesiredTasks = replicas
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// ServiceStatus sets the services' ServiceStatus (API v1.41 and above)
|
|
|
|
func ServiceStatus(desired, running uint64) func(*swarm.Service) {
|
|
|
|
return func(service *swarm.Service) {
|
|
|
|
service.ServiceStatus = &swarm.ServiceStatus{
|
|
|
|
RunningTasks: running,
|
|
|
|
DesiredTasks: desired,
|
|
|
|
}
|
2017-06-20 14:00:01 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// ServiceImage sets the service's image
|
|
|
|
func ServiceImage(image string) func(*swarm.Service) {
|
|
|
|
return func(service *swarm.Service) {
|
2017-08-07 05:52:40 -04:00
|
|
|
service.Spec.TaskTemplate = swarm.TaskSpec{ContainerSpec: &swarm.ContainerSpec{Image: image}}
|
2017-06-20 14:00:01 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// ServicePort sets the service's port
|
|
|
|
func ServicePort(port swarm.PortConfig) func(*swarm.Service) {
|
|
|
|
return func(service *swarm.Service) {
|
|
|
|
service.Spec.EndpointSpec.Ports = append(service.Spec.EndpointSpec.Ports, port)
|
2017-07-31 18:45:50 -04:00
|
|
|
|
|
|
|
assignedPort := port
|
|
|
|
if assignedPort.PublishedPort == 0 {
|
|
|
|
assignedPort.PublishedPort = 30000
|
|
|
|
}
|
|
|
|
service.Endpoint.Ports = append(service.Endpoint.Ports, assignedPort)
|
2017-06-20 14:00:01 -04:00
|
|
|
}
|
|
|
|
}
|