forked from rancher/os
/
service_factory.go
35 lines (30 loc) · 974 Bytes
/
service_factory.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
package docker
import (
"github.com/burmilla/os/pkg/util"
composeConfig "github.com/docker/libcompose/config"
"github.com/docker/libcompose/docker"
"github.com/docker/libcompose/project"
)
type ServiceFactory struct {
Context *docker.Context
Deps map[string][]string
}
func (s *ServiceFactory) Create(project *project.Project, name string, serviceConfig *composeConfig.ServiceConfig) (project.Service, error) {
if after := serviceConfig.Labels["io.rancher.os.after"]; after != "" {
for _, dep := range util.TrimSplit(after, ",") {
if dep == "cloud-init" {
dep = "cloud-init-execute"
}
s.Deps[name] = append(s.Deps[name], dep)
}
}
if before := serviceConfig.Labels["io.rancher.os.before"]; before != "" {
for _, dep := range util.TrimSplit(before, ",") {
if dep == "cloud-init" {
dep = "cloud-init-execute"
}
s.Deps[dep] = append(s.Deps[dep], name)
}
}
return NewService(s, name, serviceConfig, s.Context, project), nil
}