使用 DeflateStream 时避免复制压缩数据

Question

假设我们已经给出了一个 API 函数 f(Stream s) 来将流中包含的二进制数据放入数据库中。我想使用 f 将文件放入数据库，但我想提前压缩数据。因此我想我可以做到以下几点：

var fileStream= File.OpenRead(path);
using(var dstream = new DeflateStream(fileStream, CompressionLevel.Optimal))
   f(dstream);

但似乎 DeflateStream 只写入流 fileStream 但在压缩时不会从流中读取。在我找到的所有示例中，都使用流的 CopyTo 方法进行压缩或解压缩。但这意味着我必须在将压缩数据传递给 f 之前在内存中保留一份副本，例如：

var memoryStream = new MemoryStream();
using(var fileStream= File.OpenRead(path)) 
  using(var dstream = new DeflateStream(memoryStream, CompressionLevel.Optimal)) {
    fileStream.CopyTo(dstream);
    memoryStream.Seek(0, SeekOrigin.Begin);
    f(memoryStream);
  }

有什么方法可以避免使用 MemoryStream 吗？

更新为了一些评论者的坚持，我添加了一个完整的例子：

using System;
using System.IO;
using System.IO.Compression;

public class ThisWouldBeTheDatabaseClient {
  public void f(Stream s) {
    // some implementation I don't have access to
    // The only thing I know is that it reads data from the stream in some way.
    var buffer = new byte[10];
    s.Read(buffer,0,10);
  }
}

public class Program {
  public static void Main() {
    var dummyDatabaseClient = new ThisWouldBeTheDatabaseClient();
    var dataBuffer = new byte[1000];
    var fileStream= new MemoryStream( dataBuffer ); // would be "File.OpenRead(path)" in real case
    using(var dstream = new DeflateStream(fileStream, CompressionLevel.Optimal))
        dummyDatabaseClient.f(dstream);
  }
}

f 的虚拟实现中的读取操作抛出异常：InvalidOperationException：不支持从压缩流中读取。 结束评论中的讨论，我假设 DeflateStream 无法实现所需的行为，但第三方库中有替代方案。

Answer 1

DeflateStream 只是一个包装器，需要一个用于压缩数据的流。所以你必须使用两个流。

Is there any way to avoid using the MemoryStream?

是的。

您需要一个流来存储临时数据而不消耗（太多）内存。您可以使用 临时文件 而不是 MemoryStream。

对于懒惰的人（首先像我），让我们创建一个 class，它的行为主要类似于 MemoryStream

public class TempFileStream : FileStream
{
    public TempFileStream() : base(
        path: Path.GetTempFileName(),
        mode: FileMode.Open,
        access: FileAccess.ReadWrite,
        share: FileShare.None,
        bufferSize: 4096,
        options: FileOptions.DeleteOnClose | FileOptions.Asynchronous | FileOptions.Encrypted | FileOptions.RandomAccess)
    {
    }
}

这里的重要部分是 FileOptions.DeleteOnClose，它将在您处理流时删除临时文件。

然后使用它

using (var compressedStream = new TempFileStream())
{
    using (var deflateStream = new DeflateStream(
        stream: compressedStream,
        compressionLevel: CompressionLevel.Optimal,
        leaveOpen: true))
    using (var fileStream = File.OpenRead(path))
    {
        fileStream.CopyTo(deflateStream);
    }

    f(compressedStream);
}

Answer 2

您可以为此使用 SharpCompress。它的 DeflateStream 允许您即时读取压缩数据，这正是您想要的。

这是一个基于 Rufo 爵士的完整示例：

using System;
using System.IO;
using SharpCompress.Compressors;
using SharpCompress.Compressors.Deflate;
using System.Linq;

public class Program
{
    public static void Main()
    {
        var dataBuffer = Enumerable.Range(1, 50000).Select(e => (byte)(e % 256)).ToArray();

        using (var dataStream = new MemoryStream(dataBuffer))
        {
            // Note: this refers to SharpCompress.Compressors.Deflate.DeflateStream                
            using (var deflateStream = new DeflateStream(dataStream, CompressionMode.Compress))
            {
                ConsumeStream(deflateStream);
            }
        }
    }

    public static void ConsumeStream(Stream stream)
    {
        // Let's just prove we can reinflate to the original data...
        byte[] data;
        using (var decompressed = new MemoryStream())
        {
            using (var decompressor = new DeflateStream(stream, CompressionMode.Decompress))
            {
                decompressor.CopyTo(decompressed);
            }
            data = decompressed.ToArray();
        }
        Console.WriteLine("Reinflated size: " + data.Length);
        int errors = 0;
        for (int i = 0; i < data.Length; i++)
        {
            if (data[i] != (i + 1) % 256)
            {
                errors++;
            }
        }
        Console.WriteLine("Total errors: " + errors);
    }
}

或使用您的示例代码：

using System;
using System.IO;
using SharpCompress.Compressors;
using SharpCompress.Compressors.Deflate;

public class ThisWouldBeTheDatabaseClient {
  public void f(Stream s) {
    // some implementation I don't have access to
    // The only thing I know is that it reads data from the stream in some way.
    var buffer = new byte[10];
    s.Read(buffer,0,10);
  }
}

public class Program {
  public static void Main() {
    var dummyDatabaseClient = new ThisWouldBeTheDatabaseClient();
    var dataBuffer = new byte[1000];
    var fileStream= new MemoryStream( dataBuffer ); // would be "File.OpenRead(path)" in real case
    using(var dstream = new DeflateStream(
        fileStream, CompressionMode.Compress, CompressionLevel.BestCompression))
        dummyDatabaseClient.f(dstream);
  }
}

这现在不会抛出异常，并将提供压缩数据。

使用 DeflateStream 时避免复制压缩数据

Avoid copying compressed data when using DeflateStream

c#

deflatestream