make unit test happy 🎉

jiuaizxc · Jun 14, 2024 · ae62518 · ae62518
1 parent dd063ef
commit ae62518
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/LiteDB/Engine/Disk/Streams/AesStream.cs b/LiteDB/Engine/Disk/Streams/AesStream.cs
@@ -161,8 +161,8 @@ public AesStream(string password, Stream stream)
             }
             finally
             {
-                _bufferPool.Return(msBuffer);
-                _bufferPool.Return(checkBuffer);
+                _bufferPool.Return(msBuffer, true);
+                _bufferPool.Return(checkBuffer, true);
             }
         }
 

diff --git a/LiteDB/Engine/Sort/SortContainer.cs b/LiteDB/Engine/Sort/SortContainer.cs
@@ -136,7 +136,7 @@ private IEnumerable<BufferSlice> GetSourceFromStream(Stream stream)
                 yield return buffer;
             }
 
-            _bufferPool.Return(bytes);
+            _bufferPool.Return(bytes, true);
         }
 
         public void Dispose()

diff --git a/LiteDB/Engine/Sort/SortService.cs b/LiteDB/Engine/Sort/SortService.cs
@@ -71,9 +71,6 @@ public void Dispose()
                 }
             }
 
-            // return array buffer into pool
-            _bufferPool.Return(_buffer.Array);
-
             // return open strem into disk
             if (_reader.IsValueCreated)
             {