TPL Dataflow库的几个扩展函数

TPL Dataflow是微软面向高并发应用而推出的新程序库。借助于异步消息传递与管道，它可以提供比线程池更好的控制。本身TPL库在DataflowBlock类中提供了不少扩展函数，用起来还是非常方便的，但感觉还是不够全（当然，MS没必要设计大而全的接口），前段时间写个小程序的时候用到了它，当时顺便写了几个扩展函数，这里记录一下，如果后续有扩展再继续补充。

    static class DataFlowExtension

    {

        ///

        ///同步发送所有数据至TargetBlock

        ///

        public static void PostAll(this ITargetBlock target, IEnumerable source)

        {

            var isSuccess = source.All(i => target.Post(i));

            if (!isSuccess)

            {

                throw new InvalidOperationException();

            }

            target.Complete();

        }



        ///

        ///异步发送所有数据至TargetBlock

        ///

        public static async Task PostAllAsync(this ITargetBlock target, IEnumerable source)

        {

            foreach (var item in source)

            {

                await target.SendAsync(item);

            }

            target.Complete();

        }

        

        ///

        ///同步从数据源中获取所有数据

        ///

        public static IReadOnlyList ReceiveAll(this IReceivableSourceBlock source)

        {

            IList output;

            if (!source.TryReceiveAll(out output))

            {

                throw new InvalidOperationException();

            }



            return output as IReadOnlyList;

        }



        ///

        ///异步从数据源中获取所有数据

        ///

        public static async Task<IReadOnlyList> ReceiveAllAsync(this ISourceBlock source)

        {

            List output = new List();

            while (await source.OutputAvailableAsync())

            {

                output.Add(source.Receive());

            }

            return output;

        }

    }

这几个扩展函数本身是对DataflowBlock类中的函数二次封装，没有太多的功能，基本上每个函数都只有几行，主要为了使用更加方便罢了，由于实现简单，扩充它也是非常方便的。