From a5189d8bd1df044167b23b3ff05abb4aefd176a8 Mon Sep 17 00:00:00 2001
From: Yang Meng <yangm3ng@gmail.com>
Date: Mon, 7 Mar 2016 18:00:06 +0800
Subject: [PATCH] update mountpath of volume after migration

---
 pkg/volume/flocker/plugin.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/pkg/volume/flocker/plugin.go b/pkg/volume/flocker/plugin.go
index ae1dee31864..472ffbf0132 100644
--- a/pkg/volume/flocker/plugin.go
+++ b/pkg/volume/flocker/plugin.go
@@ -199,9 +199,15 @@ func (b flockerBuilder) SetUpAt(dir string, fsGroup *int64) error {
 		if err := b.updateDatasetPrimary(datasetID, primaryUUID); err != nil {
 			return err
 		}
+		newState, err := b.client.GetDatasetState(datasetID)
+		if err != nil {
+			return fmt.Errorf("The volume '%s' migrated unsuccessfully.", datasetID)
+		}
+		b.flocker.path = newState.Path
+	} else {
+		b.flocker.path = s.Path
 	}
 
-	b.flocker.path = s.Path
 	volumeutil.SetReady(b.getMetaDir())
 	return nil
 }