Add using()s around internal usage of IHtmlDocument

mganss · Jan 2, 2018 · 58a7a75 · 58a7a75
1 parent 40d0656
commit 58a7a75
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/src/HtmlSanitizer/HtmlSanitizer.cs b/src/HtmlSanitizer/HtmlSanitizer.cs
@@ -432,12 +432,13 @@ private static IEnumerable<INode> GetAllNodes(INode dom)
         /// <returns>The sanitized HTML body fragment.</returns>
         public string Sanitize(string html, string baseUrl = "", IMarkupFormatter outputFormatter = null)
         {
-            var dom = SanitizeDom(html, baseUrl);
-            var output = dom.Body.ChildNodes.ToHtml(outputFormatter ?? OutputFormatter);
-            return output;
+            using (var dom = SanitizeDom(html, baseUrl))
+            {
+                var output = dom.Body.ChildNodes.ToHtml(outputFormatter ?? OutputFormatter);
+                return output;
+            }
         }
 
-
         /// <summary>
         /// Sanitizes the specified HTML body fragment. If a document is given, only the body part will be returned.
         /// </summary>
@@ -465,13 +466,15 @@ public IHtmlDocument SanitizeDom(string html, string baseUrl = "")
         public string SanitizeDocument(string html, string baseUrl = "", IMarkupFormatter outputFormatter = null)
         {
             var parser = HtmlParserFactory();
-            var dom = parser.Parse(html);
 
-            DoSanitize(dom, dom.DocumentElement, baseUrl);
+            using (var dom = parser.Parse(html))
+            {
+                DoSanitize(dom, dom.DocumentElement, baseUrl);
 
-            var output = dom.ToHtml(outputFormatter ?? OutputFormatter);
+                var output = dom.ToHtml(outputFormatter ?? OutputFormatter);
 
-            return output;
+                return output;
+            }
         }
 
         /// <summary>