Dispose DafnyTokenTagger

1 files changed, 15 insertions, 1 deletions

diff --git a/Source/DafnyExtension/TokenTagger.cs b/Source/DafnyExtension/TokenTagger.cs
index 8a01c22f..5068354a 100644
--- a/Source/DafnyExtension/TokenTagger.cs
+++ b/Source/DafnyExtension/TokenTagger.cs
@@ -70,11 +70,12 @@ namespace DafnyLanguage
     }

   }

 

-  internal sealed class DafnyTokenTagger : ITagger<DafnyTokenTag>

+  internal sealed class DafnyTokenTagger : ITagger<DafnyTokenTag>, IDisposable

   {

     ITextBuffer _buffer;

     ITextSnapshot _snapshot;

     List<TokenRegion> _regions;

+    bool _disposed;

 

     internal DafnyTokenTagger(ITextBuffer buffer) {

       _buffer = buffer;

@@ -84,6 +85,19 @@ namespace DafnyLanguage
       _buffer.Changed += new EventHandler<TextContentChangedEventArgs>(ReparseFile);

     }

 

+    public void Dispose() {

+      lock (this) {

+        if (!_disposed) {

+          _buffer.Changed -= ReparseFile;

+          _buffer = null;

+          _snapshot = null;

+          _regions = null;

+          _disposed = true;

+        }

+      }

+      GC.SuppressFinalize(this);

+    }

+

     public event EventHandler<SnapshotSpanEventArgs> TagsChanged;

 

     public IEnumerable<ITagSpan<DafnyTokenTag>> GetTags(NormalizedSnapshotSpanCollection spans) {