多线程生产者/消费者

时间:2012-09-04 18:50:07

标签: c# producer-consumer

我有一个我用C#编写的代理服务器。我还有一个Java applet,它通过这个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是否变为假,在这种情况下我可以中止读取线程(这是我知道中断阻塞流读取的唯一方法)。这是一个可行的解决方案?如果是这样,我是否应该不断轮询队列中的可用数据,还是应该创建一个事件?我很想听听有关这个问题的任何想法!提前谢谢。

1 个答案:

答案 0 :(得分:2)

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

以下是套接字阻塞读取的示例。我将阻止读取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

你自己在这里回答了这个问题:

  

我的想法是在另一个线程

上安装removeServerResponseStream.Read

这正是你应该做的。当您知道应用程序已关闭时,请关闭活动线程中的套接字。这个免费是被阻止的线程,你可以优雅地结束。

这里的模式是你通常为特定套接字启动一个线程,你维护一个活动线程,它是一种请求的“控制器”线程。