Plug through labels on docker net plug

squaremo · Apr 14, 2015 · 4e827b1 · 4e827b1
1 parent c3c2dd8
commit 4e827b1
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 7 deletions.
diff --git a/api/server/server.go b/api/server/server.go
@@ -1152,8 +1152,7 @@ func postNetworkCreate(eng *engine.Engine, version version.Version, w http.Respo
 	}
 
 	body := make(map[string]interface{})
-	decoder := json.NewDecoder(r.Body)
-	if err := decoder.Decode(&body); err != nil {
+	if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
 		return err
 	}
 
@@ -1183,8 +1182,18 @@ func postNetworkPlug(eng *engine.Engine, version version.Version, w http.Respons
 		return err
 	}
 
+	body := make(map[string]interface{})
+	if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
+		return err
+	}
+
+	labels, err := getLabels(body)
+	if err != nil {
+		return err
+	}
+
 	d := getDaemon(eng)
-	id, err := d.NetworkPlug(vars["name"], vars["network"])
+	id, err := d.NetworkPlug(vars["name"], vars["network"], labels)
 	if err != nil {
 		return err
 	}

diff --git a/daemon/network.go b/daemon/network.go
@@ -95,7 +95,7 @@ func (daemon *Daemon) NetworkDestroy(id string) error {
 	return nil
 }
 
-func (daemon *Daemon) endpointOnNetwork(namesOrId string) (*Endpoint, error) {
+func (daemon *Daemon) endpointOnNetwork(namesOrId string, labels map[string]string) (*Endpoint, error) {
 	net := daemon.networks.Get(namesOrId)
 	if net == nil {
 		return nil, fmt.Errorf("Network '%s' not found", namesOrId)
@@ -104,13 +104,14 @@ func (daemon *Daemon) endpointOnNetwork(namesOrId string) (*Endpoint, error) {
 	return &Endpoint{
 		ID:      stringid.GenerateRandomID(),
 		Network: net.ID,
+		Labels:  labels,
 	}, nil
 }
 
 func (daemon *Daemon) endpointsOnNetworks(namesOrIds []string) ([]*Endpoint, error) {
 	var result []*Endpoint
 	for _, nameOrId := range namesOrIds {
-		endpoint, err := daemon.endpointOnNetwork(nameOrId)
+		endpoint, err := daemon.endpointOnNetwork(nameOrId, nil)
 		if err != nil {
 			return nil, err
 		}
@@ -119,7 +120,7 @@ func (daemon *Daemon) endpointsOnNetworks(namesOrIds []string) ([]*Endpoint, err
 	return result, nil
 }
 
-func (daemon *Daemon) NetworkPlug(containerID, nameOrId string) (string, error) {
+func (daemon *Daemon) NetworkPlug(containerID, nameOrId string, labels map[string]string) (string, error) {
 	daemon.networks.Lock()
 	defer daemon.networks.Unlock()
 
@@ -132,7 +133,7 @@ func (daemon *Daemon) NetworkPlug(containerID, nameOrId string) (string, error)
 		return "", fmt.Errorf("Cannot plug in running container (yet)")
 	}
 
-	endpoint, err := daemon.endpointOnNetwork(nameOrId)
+	endpoint, err := daemon.endpointOnNetwork(nameOrId, labels)
 	if err != nil {
 		return "", err
 	}