Fix missing closing to Docker socket

kassisol · Jul 26, 2018 · d44d89f · d44d89f
1 parent 64507b1
commit d44d89f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/app.go b/app.go
@@ -67,6 +67,11 @@ func runApp(cmd *cobra.Command, args []string) {
 				continue
 			}
 
+			if err := d.Close(); err != nil {
+				log.Error(err)
+				continue
+			}
+
 			if service {
 				// Check if IP is already set
 				i := ip.NewIP(nic, ipaddr.Address, ipaddr.Netmask)

diff --git a/pkg/docker/docker.go b/pkg/docker/docker.go
@@ -23,6 +23,10 @@ func NewDockerClient() (*DockerConfig, error) {
 	return &DockerConfig{Client: cli}, nil
 }
 
+func (c *DockerConfig) Close() error {
+	return c.Client.Close()
+}
+
 func (c *DockerConfig) GetNodeID() error {
 	info, err := c.Client.Info(context.Background())
 	if err != nil {