Question

我有一个输入流和一个流阅读器组件。这工作正常但现在我想记录所有流量（将副本保存在文件中）。所以我需要监视一个流。我想到的解决方案是流管道（管道）或流包装器，它将流作为输入，然后让我第一次看到流量。像这样：

    void Init(System.Net.Sockets.NetworkStream stream)
    {
        System.IO.Stream wrappedStream = new MyWrapper(stream);
        wrappedStream.ReadSpy = MyMethod;
        XmlReader reader = XmlReader.Create(wrappedStream);
    }

    // This will get called after some bytes have been read from a stream,
    // but before they get passed to the XmlReader
    byte[] MyMethod(byte[] buffer)
    {
        m_Writer.Write(buffer); // write to a file 
        return buffer; // Give to XmlReader 
    }

Answer 1

你想要什么叫做装饰模式。它是一种动态添加/修改行为的技术：

为此，您希望使用工厂的构造函数实现抽象类Stream，该构造函数接受另一个Stream实例。您为抽象类的每个方法/重载提供了一个实现，它在装饰的Stream上调用相同的方法/重载，并执行您需要的任何其他工作。

一旦你完成了这个并用你的新装饰器装饰Stream，它可以被接受Stream的任何其他东西互换使用，包括其他类似的装饰器：装饰者甚至可以嵌套，就像洋葱层一样，撰写你需要的行为。

这样的事情：

class StreamInterceptor : Stream
{

  public Stream DecoratedInstance { get; set; }

  public event Action<byte[]> BytesRead;
  public event Action<byte[]> BytesWritten;

  public StreamInterceptor( Stream instance )
  {
    if ( instance == null ) throw new ArgumentNullException("instance");
    this.DecoratedInstance = instance ;
    return ;
  }

  public override bool CanRead
  {
    get { return DecoratedInstance.CanRead; }
  }

  public override bool CanSeek
  {
    get { return DecoratedInstance.CanSeek; }
  }

  public override bool CanWrite
  {
    get { return DecoratedInstance.CanWrite; }
  }

  public override void Flush()
  {
    DecoratedInstance.Flush();
    return;
  }

  public override long Length
  {
    get { return DecoratedInstance.Length; }
  }

  public override long Position
  {
    get { return DecoratedInstance.Position; }
    set { DecoratedInstance.Position = value; }
  }

  public override int Read( byte[] buffer , int offset , int count )
  {
    int bytesRead = DecoratedInstance.Read(buffer, offset, count);

    // raise the bytes read event
    byte[] temp = new byte[bytesRead];
    Array.Copy(buffer,offset,temp,0,bytesRead);
    BytesRead(temp);

    return bytesRead;
  }

  public override long Seek( long offset , SeekOrigin origin )
  {
    return DecoratedInstance.Seek(offset, origin);
  }

  public override void SetLength( long value )
  {
    DecoratedInstance.SetLength(value);
    return;
  }

  public override void Write( byte[] buffer , int offset , int count )
  {

    // raise the bytes written event
    byte[] temp = new byte[count];
    Array.Copy(buffer,offset,temp,0,count);
    BytesWritten(temp);

    DecoratedInstance.Write(buffer, offset, count);
    return;
  }

}

一旦你有了这个，你可以这样说：

static void Main()
{
  StreamInterceptor si = new StreamInterceptor(File.Open("foo.bar.txt",FileMode.Open,FileAccess.ReadWrite,FileShare.Read));

  si.BytesRead    += (bytes) => { Console.WriteLine("{0} bytes read", bytes.Length); } ;
  si.BytesWritten += (bytes) => { Console.WriteLine("{0} bytes written", bytes.Length); } ;

  Stream s = (Stream) si ;
  DoSomethingUseful(s);

}

只要有人从流中读取或写入，就会调用您的事件处理程序。

C＃流管（Stream Spy）

1 个答案: