Improved MyLongRunningCommand.

Author: rdingwall

src/ProgressDialogEx/MainWindow.xaml

@@ -72,7 +72,7 @@
                     Margin="5,5,5,5"
                     Click="ShowProgressDialogWithErrorAndCancelButtonAndProgress" />
             <Separator />
-            <Button Content="ProgressDialog with task composition example"
+            <Button Content="ProgressDialog with async task composition example"
                     HorizontalContentAlignment="Left"
                     Padding="5,0,5,0"
                     Height="26"

src/ProgressDialogEx/MyLongRunningCommand.cs

@@ -1,6 +1,7 @@
 using System;
 using System.Threading;
 using System.Threading.Tasks;
+using System.Windows;
 using System.Windows.Input;
 using ProgressDialogEx.ProgressDialog;
 
@@ -17,16 +18,18 @@ public MyLongRunningCommand(IProgressDialogService dialogService)
 
         public void Execute(object parameter)
         {
-            dialogService.Execute(DoWork, new ProgressDialogOptions { WindowTitle = "Loading files" });
+            Task<int> task = dialogService.ExecuteAsync(DoWork, new ProgressDialogOptions { WindowTitle = "Loading files" });
+
+            MessageBox.Show(String.Format("Result = {0}", task.Result));
         }
 
-        void DoWork(CancellationToken cancellationToken, IProgress<string> progress)
+        static async Task<int> DoWork(CancellationToken cancellationToken, IProgress<string> progress)
         {
-            Task.Factory.StartNew(() => progress.Report("First"), cancellationToken)
-                .ContinueWith(_ => Thread.Sleep(1000), cancellationToken)
-                .ContinueWith(_ => progress.Report("Second"), cancellationToken)
-                .ContinueWith(_ => Thread.Sleep(1000), cancellationToken)
-                .Wait();
+            return await Task.Factory.StartNew(() => progress.Report("First"), cancellationToken)
+                             .ContinueWith(_ => Thread.Sleep(1000), cancellationToken)
+                             .ContinueWith(_ => progress.Report("Second"), cancellationToken)
+                             .ContinueWith(_ => Thread.Sleep(1000), cancellationToken)
+                             .ContinueWith(_ => 42);
         }
 
         public bool CanExecute(object parameter)