fix conditional for truncate

santisq · Sep 6, 2024 · 83ee2a8 · 83ee2a8
1 parent d6ef8a1
commit 83ee2a8
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/src/PSADTree/PSADTreeCmdletBase.cs b/src/PSADTree/PSADTreeCmdletBase.cs
@@ -98,17 +98,15 @@ protected override void BeginProcessing()
 
     protected void Push(GroupPrincipal? groupPrincipal, TreeGroup treeGroup)
     {
-        if (Recursive.IsPresent || treeGroup.Depth <= Depth)
+        if (!Recursive.IsPresent && treeGroup.Depth >= Depth && !_truncateWritten)
         {
-            _stack.Push((groupPrincipal, treeGroup));
+            this.WriteWarning($"Result is truncated as enumeration has exceeded the set depth of {Depth}.");
+            _truncateWritten = true;
         }
-        else
+
+        if (Recursive.IsPresent || treeGroup.Depth <= Depth)
         {
-            if (!_truncateWritten)
-            {
-                this.WriteWarning($"Result is truncated as enumeration has exceeded the set depth of {Depth}.");
-                _truncateWritten = true;
-            }
+            _stack.Push((groupPrincipal, treeGroup));
         }
     }