Martynas Pumputis
0ff37d3492
We cannot do it from weaveexec, because the latter is distributed with an old version of Docker client which does not support Plugins API.
118 lines
3.0 KiB
Go
118 lines
3.0 KiB
Go
/* various weave docker network plugin operations */
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/docker/docker/client"
|
|
docker "github.com/fsouza/go-dockerclient"
|
|
|
|
"github.com/weaveworks/weave/plugin/net"
|
|
)
|
|
|
|
func createPluginNetwork(args []string) error {
|
|
if len(args) != 3 {
|
|
cmdUsage("create-plugin-network", "<network-name> <driver-name> <default-subnet>")
|
|
}
|
|
networkName := args[0]
|
|
driverName := args[1]
|
|
subnet := args[2]
|
|
d, err := newDockerClient()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = d.CreateNetwork(
|
|
docker.CreateNetworkOptions{
|
|
Name: networkName,
|
|
CheckDuplicate: true,
|
|
Driver: driverName,
|
|
IPAM: docker.IPAMOptions{
|
|
Driver: driverName,
|
|
Config: []docker.IPAMConfig{{Subnet: subnet}},
|
|
},
|
|
Options: map[string]interface{}{plugin.MulticastOption: "true"},
|
|
})
|
|
if err != docker.ErrNetworkAlreadyExists && err != nil {
|
|
// Despite appearances to the contrary, CreateNetwork does
|
|
// sometimes(always?) *not* return ErrNetworkAlreadyExists
|
|
// when the network already exists. Hence we need to check for
|
|
// this explicitly.
|
|
if _, err2 := d.NetworkInfo(networkName); err2 != nil {
|
|
return fmt.Errorf("unable to create network: %s", err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func removePluginNetwork(args []string) error {
|
|
if len(args) != 1 {
|
|
cmdUsage("remove-plugin-network", "<network-name>")
|
|
}
|
|
networkName := args[0]
|
|
d, err := newDockerClient()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = d.RemoveNetwork(networkName)
|
|
if _, ok := err.(*docker.NoSuchNetwork); !ok && err != nil {
|
|
if info, err2 := d.NetworkInfo(networkName); err2 == nil {
|
|
if len(info.Containers) > 0 {
|
|
containers := ""
|
|
for container := range info.Containers {
|
|
containers += fmt.Sprintf(" %.12s ", container)
|
|
}
|
|
return fmt.Errorf(`WARNING: the following containers are still attached to network %q:
|
|
%s
|
|
Docker operations involving those containers may pause or fail
|
|
while Weave is not running`, networkName, containers)
|
|
}
|
|
}
|
|
return fmt.Errorf("unable to remove network: %s", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Exits with 0 if the given plugin (v2) is enabled.
|
|
func isDockerPluginEnabled(args []string) error {
|
|
if len(args) != 1 {
|
|
cmdUsage("is-docker-plugin-enabled", "<plugin-name>")
|
|
}
|
|
|
|
pluginName := args[0]
|
|
|
|
// This is messed up: we are using docker/docker/client instead of
|
|
// fsouza/go-dockerclient because the latter does not support plugins.
|
|
c, err := client.NewEnvClient()
|
|
if err != nil {
|
|
return fmt.Errorf("unable to connect to docker: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
p, _, err := c.PluginInspectWithRaw(ctx, pluginName)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !p.Enabled {
|
|
return fmt.Errorf("plugin %q is disabled", pluginName)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func newDockerClient() (*docker.Client, error) {
|
|
// API 1.21 is the first version that supports docker network
|
|
// commands
|
|
c, err := docker.NewVersionedClientFromEnv("1.21")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("unable to connect to docker: %s", err)
|
|
}
|
|
_, err = c.Version()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("unable to connect to docker: %s", err)
|
|
}
|
|
|
|
return c, nil
|
|
}
|