c# stream

最新推荐文章于 2024-06-26 20:17:49 发布

原创最新推荐文章于 2024-06-26 20:17:49 发布 · 963 阅读

0 ·

CC 4.0 BY-SA版权

C# 专栏收录该内容

19 篇文章

订阅专栏

本文提供了一种方法，通过流获取指定长度的字节数组，使用循环和条件判断来确保正确读取并返回所需字节数组。

        /// <summary>
        /// 从流获取指定长度的字节数组
        /// </summary>
        public static IEnumerable<byte[]> GetBytes(this Stream stream, long length, int size = 1024)
        {
            byte[] buffer = new byte[size];
            int len, read;

            long sum = 0;
            do
            {
                long unread = length - sum;
                read = unread >= size ? size : (int)unread;

                len = stream.Read(buffer, 0, read);
                sum += len;

                if (buffer.Length == len)
                {
                    yield return buffer;
                }
                else
                {
                    yield return buffer.Take(len).ToArray();
                }
            } while (sum < length && len > 0);
        }