-
Notifications
You must be signed in to change notification settings - Fork 0
/
service.go
117 lines (103 loc) · 3.64 KB
/
service.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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
package dockercompose
import (
"encoding/json"
"fmt"
"io/ioutil"
"strings"
log "github.com/sirupsen/logrus"
"github.com/seppo0010/vortices-dockercompose/exec"
)
type ServiceConfig struct {
Image string
Command []string `yaml:"command,omitempty"`
Privileged bool
}
type Service struct {
ServiceConfig `yaml:",inline"`
ContainerName string `yaml:"container_name"`
name string
Networks map[string]ServiceNetworkConfig `yaml:"networks"`
compose *Compose
serviceNetworksConfig []ServiceNetworkConfig
}
type ServiceNetworkConfig struct {
Network *Network `yaml:"-"`
Aliases []string `yaml:"aliases,omitempty"`
}
func (s *Service) SetNetworks(serviceNetworksConfig []ServiceNetworkConfig) {
s.serviceNetworksConfig = serviceNetworksConfig
s.Networks = make(map[string]ServiceNetworkConfig, len(s.serviceNetworksConfig))
for _, network := range s.serviceNetworksConfig {
s.Networks[network.Network.name] = network
}
}
func (s *Service) Exec(path string, args ...string) exec.Cmd {
args = append([]string{"exec", "-T", s.name, path}, args...)
cmd := s.compose.exec.New("docker-compose", args...)
cmd.SetDir(s.compose.getTmpDir())
return cmd
}
func (s *Service) SudoExec(path string, args ...string) exec.Cmd {
args = append([]string{"exec", "-T", "--privileged", s.name, path}, args...)
cmd := s.compose.exec.New("docker-compose", args...)
cmd.SetDir(s.compose.getTmpDir())
return cmd
}
func (s *Service) GetIPAddressForNetwork(network *Network) (string, error) {
networksExec := s.compose.exec.New("docker", "inspect", "-f", "{{json .NetworkSettings.Networks}}", s.name)
stdout, err := networksExec.StdoutPipe()
if err != nil {
log.Errorf("failed to pipe network settings stdout: %s", err.Error())
return "", err
}
if err := networksExec.Start(); err != nil {
log.Errorf("failed to inspect network settings: %s", err.Error())
return "", err
}
var networks map[string]map[string]interface{}
err = json.NewDecoder(stdout).Decode(&networks)
if err != nil {
log.Errorf("failed to decode network settings json: %s", err.Error())
return "", err
}
if err := networksExec.Wait(); err != nil {
log.Errorf("failed to wait inspect network settings: %s", err.Error())
return "", err
}
for network_id, data := range networks {
networkLabelExec := s.compose.exec.New("docker", "inspect", "-f", "{{range $key, $value := .Labels}}{{if eq $key \"com.docker.compose.network\"}}{{$value}}{{end}}{{end}}", network_id)
stdout, err := networkLabelExec.StdoutPipe()
if err != nil {
log.Errorf("failed to pipe network label settings stdout: %s", err.Error())
return "", err
}
if err = networkLabelExec.Start(); err != nil {
log.Errorf("failed to run network label settings: %s", err.Error())
return "", err
}
stdoutBytes, err := ioutil.ReadAll(stdout)
if err != nil {
log.Errorf("failed to read network label settings stdout: %s", err.Error())
return "", err
}
if err = networkLabelExec.Wait(); err != nil {
log.Errorf("failed to wait network label settings stdout: %s", err.Error())
return "", err
}
if strings.Trim(string(stdoutBytes), " \n") == network.name {
ip, found := data["IPAddress"]
if !found {
log.Errorf("ip address not found")
return "", fmt.Errorf("ip address not found")
}
ipStr, ok := ip.(string)
if !ok {
log.Errorf("invalid ip address, got %T", ip)
return "", fmt.Errorf("invalid ip address, got %T", ip)
}
return ipStr, nil
}
}
log.Errorf("could not find ip address for %s in network %s", s.name, network.name)
return "", fmt.Errorf("could not find ip address for %s in network %s", s.name, network.name)
}