• .NET 9 中的 UnboundedPrioritizedChannel的用法
  • 发布于 2个月前
  • 205 热度
    0 评论
背景
.NET 9 Preview 5 中引入了一个支持优先级的 Channel,对于我们需要实现支持优先级的内存队列来说变得更加方便了

新的API
namespace System.Threading.Channels
{
    public partial class Channel
    {
        public static System.Threading.Channels.Channel<T> CreateUnboundedPrioritized<T>() { throw null; }
        public static System.Threading.Channels.Channel<T> CreateUnboundedPrioritized<T>(System.Threading.Channels.UnboundedPrioritizedChannelOptions<T> options) { throw null; }
    }
    public sealed partial class UnboundedPrioritizedChannelOptions<T> : System.Threading.Channels.ChannelOptions
    {
        public System.Collections.Generic.IComparer<T>? Comparer { get; set; }
    }
}
引入了新的 Channel.CreateUnboundedPrioritized 方法来创建支持优先级的 channel,就像 API 名字一样,这个 Channel 是一个 Unbounded channel,对于 BoundedChannel 的支持有一个 Github issue,感兴趣的朋友可以去看下:https://github.com/dotnet/runtime/issues/101292

例子
var c = Channel.CreateUnboundedPrioritized<int>();
//堆代码 duidaima.com
await c.Writer.WriteAsync(1);
await c.Writer.WriteAsync(5);
await c.Writer.WriteAsync(2);
await c.Writer.WriteAsync(4);
await c.Writer.WriteAsync(3);
c.Writer.Complete();

while (await c.Reader.WaitToReadAsync())
{
    while (c.Reader.TryRead(out int item))
    {
        Console.WriteLine(item);
    }
}
默认会使用默认的比较器,我们也可以实现我们自己的比较器,实现自己的优先级比较,下面是一个简单的反向输出的示例
file sealed class ReverseComparer : IComparer<int>
{
    public int Compare(int x, int y)
    {
        return y.CompareTo(x);
    }
}
var c = Channel.CreateUnboundedPrioritized<int>(new()
{
    Comparer = new ReverseComparer()
});

await c.Writer.WriteAsync(1);
await c.Writer.WriteAsync(5);
await c.Writer.WriteAsync(2);
await c.Writer.WriteAsync(4);
await c.Writer.WriteAsync(3);
c.Writer.Complete();

while (await c.Reader.WaitToReadAsync())
{
    while (c.Reader.TryRead(out int item))
    {
        Console.WriteLine(item);
    }
}

从输出结果可以看出来已经 channel 中的消息排好序了,默认从小到大,第二个示例我们使用了自定义的一个比较器变成了从大到小。
用户评论