HGKeeper

// Convey

// This program is free software: you can redistribute it and/or modify

// it under the terms of the GNU General Public License as published by

// the Free Software Foundation, either version 3 of the License, or

// (at your option) any later version.

// This program is distributed in the hope that it will be useful,

// but WITHOUT ANY WARRANTY; without even the implied warranty of

// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License

// along with this program. If not, see <http://www.gnu.org/licenses/>.

package docker

import (

"github.com/aphistic/gomol"

"bitbucket.org/rw_grim/convey/logging"

"bitbucket.org/rw_grim/convey/state"

"bitbucket.org/rw_grim/convey/util"

)

type Network struct {

name string

logger *gomol.LogAdapter

state *state.State

}

const (

networkCreateTemplate = `network create {{.Name}}`

networkDisconnectTemplate = `network disconnect --force {{.Name}} {{.ContainerName}}`

networkDestroyTemplate = `network rm {{.Name}}`

)

func NewNetwork(st *state.State) (*Network, error) {

network := &Network{

name: util.ID(),

logger: logging.NewAdapter("network"),

state: st,

}

params := map[string]interface{}{

"Name": network.name,

}

_, _, err := DockerOutput("create network", networkCreateTemplate, params, st)

if err != nil {

return nil, err

}

network.logger.Debugf("created network: %#v", network.name)

return network, nil

}

func (network *Network) Name() string {

return network.name

}

func (network *Network) Destroy() error {

for _, name := range network.state.GetRunning() {

params := map[string]interface{}{

"Name": network.name,

"ContainerName": name,

}

network.logger.Infof("disconnecting container %s from network %s", name, network.Name())

err := Docker("disconnect container", networkDisconnectTemplate, params, network.state)

if err != nil {

network.logger.Warningf("failed to disconnect container %s from network %s", name, network.Name())

continue

}

network.logger.Infof("disconnected container %s from network %s", name, network.Name())

}

params := map[string]interface{}{

"Name": network.name,

}

return Docker("remove network", networkDestroyTemplate, params, network.state)

}