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 applet 终止(上面代码中的变量 output)。但是，Java 小程序无法关闭此连接，因为代理在 remoteServerResponseStream.Read 等待数据时从未确认关闭请求。

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

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

Answer 1

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

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

在您使用阻塞阅读器的类中，您应该实现一次性模式并在那里关闭/处理您的阅读器。

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 ”线程。