chore: improve asyncEnumerable.WithBuffer.

alexyakunin · alexyakunin · commit 99087ab2a62d · 2024-02-25T20:06:10.000-08:00
diff --git a/src/Stl/Channels/EnumerableExt.cs b/src/Stl/Channels/EnumerableExt.cs
@@ -18,6 +18,9 @@ public static IAsyncEnumerable<T> WithBuffer<T>(
         bool allowSynchronousContinuations,
         CancellationToken cancellationToken = default)
     {
+        if (bufferSize < 1)
+            return source;
+
         var buffer = source.ToBoundedChannel(new(bufferSize) {
             SingleReader = true,
             SingleWriter = true,
@@ -31,6 +34,9 @@ public static IAsyncEnumerable<T> WithBuffer<T>(
         BoundedChannelOptions options,
         CancellationToken cancellationToken = default)
     {
+        if (options.Capacity < 1)
+            return source;
+
         var buffer = source.ToBoundedChannel(options, cancellationToken);
         return buffer.Reader.ReadAllAsync(cancellationToken);
     }

Original file line number	Diff line number	Diff line change
`@@ -18,6 +18,9 @@ public static IAsyncEnumerable<T> WithBuffer<T>(`
`18`	`18`	`bool allowSynchronousContinuations,`
`19`	`19`	`CancellationToken cancellationToken = default)`
`20`	`20`	`{`
	`21`	`+ if (bufferSize < 1)`
	`22`	`+ return source;`
	`23`	`+`
`21`	`24`	`var buffer = source.ToBoundedChannel(new(bufferSize) {`
`22`	`25`	`SingleReader = true,`
`23`	`26`	`SingleWriter = true,`
`@@ -31,6 +34,9 @@ public static IAsyncEnumerable<T> WithBuffer<T>(`
`31`	`34`	`BoundedChannelOptions options,`
`32`	`35`	`CancellationToken cancellationToken = default)`
`33`	`36`	`{`
	`37`	`+ if (options.Capacity < 1)`
	`38`	`+ return source;`
	`39`	`+`
`34`	`40`	`var buffer = source.ToBoundedChannel(options, cancellationToken);`
`35`	`41`	`return buffer.Reader.ReadAllAsync(cancellationToken);`
`36`	`42`	`}`