json-extension: support cancellation in progress example

openlawlibrary · Jul 10, 2022 · db66295 · db66295
1 parent 8adc050
commit db66295
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/examples/json-extension/server/server.py b/examples/json-extension/server/server.py
@@ -202,9 +202,14 @@ async def progress(ls: JsonLanguageServer, *args):
     # Create
     await ls.progress.create_async(token)
     # Begin
-    ls.progress.begin(token, WorkDoneProgressBegin(title='Indexing', percentage=0))
+    ls.progress.begin(token, WorkDoneProgressBegin(title='Indexing', percentage=0, cancellable=True))
     # Report
     for i in range(1, 10):
+        # Check for cancellation from client
+        if ls.progress.is_cancelled(token):
+            # ... and stop the computation if client cancelled
+            ls.progress.end(token, WorkDoneProgressEnd(message='Cancelled'))
+            return
         ls.progress.report(
             token,
             WorkDoneProgressReport(message=f'{i * 10}%', percentage= i * 10),