From d20c501af4dcb8d7d02247e1f12e9645102ee6ca Mon Sep 17 00:00:00 2001
From: Selvasundaram <selvam@gluster.com>
Date: Mon, 11 Jul 2011 19:20:36 +0530
Subject: Format disk task status update bug fixes [Bug 3150] New: Brick list
 is not getting updated after migrate operation - fix

---
 .../gluster/storage/management/gui/dialogs/MigrateBrickWizard.java   | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'src/com.gluster.storage.management.gui')

diff --git a/src/com.gluster.storage.management.gui/src/com/gluster/storage/management/gui/dialogs/MigrateBrickWizard.java b/src/com.gluster.storage.management.gui/src/com/gluster/storage/management/gui/dialogs/MigrateBrickWizard.java
index 4693726b..d8f0f165 100644
--- a/src/com.gluster.storage.management.gui/src/com/gluster/storage/management/gui/dialogs/MigrateBrickWizard.java
+++ b/src/com.gluster.storage.management.gui/src/com/gluster/storage/management/gui/dialogs/MigrateBrickWizard.java
@@ -27,6 +27,7 @@ import com.gluster.storage.management.client.GlusterDataModelManager;
 import com.gluster.storage.management.client.TasksClient;
 import com.gluster.storage.management.client.VolumesClient;
 import com.gluster.storage.management.core.model.Brick;
+import com.gluster.storage.management.core.model.Cluster;
 import com.gluster.storage.management.core.model.TaskInfo;
 import com.gluster.storage.management.core.model.Volume;
 
@@ -34,6 +35,7 @@ public class MigrateBrickWizard extends Wizard {
 	private Volume volume;
 	private Brick brick;
 	private MigrateBrickPage1 page;
+	private Cluster cluster = GlusterDataModelManager.getInstance().getModel().getCluster();
 
 	public MigrateBrickWizard(Volume volume, Brick brick) {
 		setWindowTitle("Gluster Management Console - Migrate Brick [" + volume.getName() + "]");
@@ -63,7 +65,8 @@ public class MigrateBrickWizard extends Wizard {
 			TasksClient taskClient = new TasksClient();
 			TaskInfo taskInfo = taskClient.getTaskInfo(uri);
 			if (taskInfo != null && taskInfo instanceof TaskInfo) {
-				GlusterDataModelManager.getInstance().getModel().getCluster().addTaskInfo(taskInfo);
+				cluster.addTaskInfo(taskInfo);
+				GlusterDataModelManager.getInstance().refreshVolumeData(cluster.getVolume(taskInfo.getReference()));
 			}
 			MessageDialog.openInformation(getShell(), "Brick migration", "Brick migration started successfully");
 		} catch (Exception e) {
-- 
cgit