c# - 多线程生产者/消费者

Question

我有一个用 C# 编写的代理服务器。我还有一个 Java 小程序，它通过此 C# 代理从视频服务器接收 MJPEG 数据。我的问题是，当视频服务器当前不再提供 MJPEG 数据时，代理会陷入阻塞读取调用，我无法终止它。

// write the forwarded output
// blocking on remoteServerResponseStream.Read
while (m_running && (read = remoteServerResponseStream.Read(buffer, 0, buffer.Length)) > 0)
{
    bytesRead += read;

    output.Write(buffer, 0, read);

    output.Flush();
}

这应该由 Java 小程序关闭流(上面代码中的变量 output)来终止。但是，Java 小程序无法关闭此连接，因为代理在陷入 remoteServerResponseStream.Read 等待数据时永远不会确认关闭请求。

我已经被这个问题困扰了一个星期了。我想我可能想到了一个解决方案，但我不确定它是否有效。我很想听到对此的任何反馈。

我的想法是在另一个线程上使用removeServerResponseStream.Read并使用共享队列来传输数据。该线程将读取数据并将其放入队列中。然后，我的主线程会将队列中的任何可用数据转发到输出。这样，我可以不断检查 output.CanWrite 是否变为 false，在这种情况下我可以中止读取线程(这是我知道的中断阻塞流读取的唯一方法)。这是一个可行的解决方案吗？如果是这样，我应该不断轮询队列以获取可用数据还是应该创建一个事件？我很想听听关于这个问题的任何想法!提前致谢。

Answer 1

在这些情况下，我只是从另一个线程结束流读取器。当您关闭阅读器时，它将打破循环。通过额外的同步队列只是更多不必要的样板操作。最终你必须在某个地方阻止阻塞的读者。将值放入队列只能阻止您的消费者被阻止，但不能阻止您的阅读器。

这是一个套接字阻塞读取的示例。我将阻止读取 1 个字节，而我故意永远不会发送该字节。几秒钟后，我将处理套接字，套接字将摆脱阻塞读取，应用程序将正常退出。我将记录每个线程正在做什么以及何时发生(每个日志行前缀的数字是线程 ID，我将缩进单独的线程)。

在使用阻塞读取器的类中，您应该实现 Disposable 模式并在那里关闭/处置您的读取器。

static void Main(string[] args)
{
    SocketTest();

    Console.WriteLine("Press any key to exit");
    Console.ReadKey();
}

public static void SocketTest()
{
    int port = 22345;

    var tcpListener = new TcpListener(IPAddress.Any, port);

    tcpListener.Start();

    // Listening thread
    new Thread(() =>
    {

        Console.WriteLine(Thread.CurrentThread.ManagedThreadId + " - Waiting for connection to port");

        var socket = tcpListener.AcceptSocket();

        Console.WriteLine(Thread.CurrentThread.ManagedThreadId + " - Connection accepted");

        var stream = new NetworkStream(socket);
        var reader = new BinaryReader(stream);

        try
        {
            Console.WriteLine(Thread.CurrentThread.ManagedThreadId + " - Starting blocking read");
            var bytes = reader.ReadBytes(1);
            Console.WriteLine(Thread.CurrentThread.ManagedThreadId + " - Done blocking read, read {0} bytes", bytes.Length);
        }
        catch (Exception ex)
        {
            Console.WriteLine("Error reading " + ex);
        }
    }).Start();

    // connecting thread
    new Thread(() =>
    {
        var socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);

        Console.WriteLine("\t" + Thread.CurrentThread.ManagedThreadId + " - Connecting to local port");

        socket.Connect("127.0.0.1", port);

        Console.WriteLine("\t" + Thread.CurrentThread.ManagedThreadId + " - Connecting to local succeeded");

        Thread.Sleep(TimeSpan.FromSeconds(2));

        Console.WriteLine("\t" + Thread.CurrentThread.ManagedThreadId + " - Disposing of socket");

        socket.Dispose();

    }).Start();

    Thread.Sleep(TimeSpan.FromSeconds(5));
}

当你运行这个你会得到:

3 - Waiting for connection to port
        4 - Connecting to local port
        4 - Connecting to local succeeded
3 - Connection accepted
3 - Starting blocking read
        4 - Disposing of socket
3 - Done blocking read, read 0 bytes
Press any key to exit

不过，您自己在这里回答了这个问题:

My thought was to have the removeServerResponseStream.Read on another thread

这正是你应该做的。当您知道您的应用程序已关闭时，请关闭事件线程中的套接字。这释放了被阻塞的线程，您可以优雅地结束。

这里的模式是，您通常为特定套接字启动一个线程，并维护一个事件线程，该线程是一种用于请求的“ Controller ”线程。