From c8e39ae95459bcd3b1fc27a4ce86ed3edc8e182f Mon Sep 17 00:00:00 2001
From: allencloud <allen.sun@daocloud.io>
Date: Fri, 20 May 2016 02:48:19 +0800
Subject: [PATCH] refresh single volume when event emitted

Signed-off-by: allencloud <allen.sun@daocloud.io>
---
 cluster/engine.go | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/cluster/engine.go b/cluster/engine.go
index d124433576..70f5410e2e 100644
--- a/cluster/engine.go
+++ b/cluster/engine.go
@@ -690,6 +690,28 @@ func (e *Engine) RefreshVolumes() error {
 	return nil
 }
 
+// refreshVolume refreshes single volume on the engine.
+func (e *Engine) refreshVolume(IDOrName string) error {
+	volume, err := e.apiClient.VolumeInspect(context.Background(), IDOrName)
+	e.CheckConnectionErr(err)
+	if err != nil {
+		if strings.Contains(err.Error(), "No such volume") {
+			e.Lock()
+			delete(e.volumes, IDOrName)
+			e.Unlock()
+			return nil
+		} else {
+			return err
+		}
+	}
+
+	e.Lock()
+	e.volumes[volume.Name] = &Volume{Volume: volume, Engine: e}
+	e.Unlock()
+
+	return nil
+}
+
 // RefreshContainers will refresh the list and status of containers running on the engine. If `full` is
 // true, each container will be inspected.
 // FIXME: unexport this method after mesos scheduler stops using it directly
@@ -1026,7 +1048,7 @@ func (e *Engine) CreateVolume(request *types.VolumeCreateRequest) (*types.Volume
 		return nil, err
 	}
 
-	e.RefreshVolumes()
+	e.refreshVolume(volume.Name)
 
 	return &volume, err
 }
@@ -1216,7 +1238,7 @@ func (e *Engine) handler(msg events.Message) error {
 	case "network":
 		e.refreshNetwork(msg.Actor.ID)
 	case "volume":
-		e.RefreshVolumes()
+		e.refreshVolume(msg.Actor.ID)
 	case "image":
 		e.RefreshImages()
 	case "container":