如何消除Rox NIO教程中的竞争条件

时间:2012-01-10 17:20:21

标签: java nio race-condition

我一直在使用this教程来使用套接字IO进行简单的文件传输客户端/服务器。我将响应处理程序更改为接受多个读取作为一个文件的一部分,因为我将处理大文件,可能高达500 MB。该教程没有考虑大型服务器响应,所以我有点挣扎,我已经创建了一个竞争条件。

这是响应处理程序代码:

public class RspHandler {

private byte[] rsp = null;
public synchronized boolean handleResponse(byte[] rsp) {
    this.rsp = rsp;
    this.notify();
    return true;
}

public synchronized void waitForResponse() {
    while(this.rsp == null) {
        try {
            this.wait();
        } catch (InterruptedException e) {
        }
    }
    System.out.println("Received Response : " + new String(this.rsp));
}

public synchronized void waitForFile(String filename) throws IOException {
    String filepath = "C:\\a\\received\\" + filename;
    FileOutputStream fos = new FileOutputStream(filepath);
    while(waitForFileChunk(fos) != -1){}
    fos.close();
}

private synchronized int waitForFileChunk(FileOutputStream fos) throws IOException
{
    while(this.rsp == null) {
        try {
            this.wait();
        } catch (InterruptedException e) {
        }
    }
    fos.write(this.rsp);
    int length = this.rsp.length;
    this.rsp = null;
    if(length < NioClient.READ_SIZE)//Probably a bad way to find the end of the file
    {
        return -1;
    }
    else
    {
        return length;
    }

}
}

程序的主线程在主线程上创建一个RspHandler,并将其传递给在单独线程上创建的客户端。主线程告诉客户端请求文件,然后告诉RspHandler监听响应。当客户端从服务器读取时(它现在读取大约1KB的块),它调用handleResponse(byte[] rsp)方法,填充rsp字节数组。

基本上,我不会像接收到的那样快地将收到的数据写入文件。我对线程有点新意,所以我不知道如何摆脱这种竞争条件。任何提示?

1 个答案:

答案 0 :(得分:3)

这是经典的消费者/制作人。最直接/最简单的方法是使用BlockingQueue。制作人调用put(),消费者调用take()

请注意,使用BlockingQueue通常会导致“我如何完成”问题。最好的方法是使用“毒丸”方法,生产者在队列上贴上一个“特殊”值,告知消费者没有更多数据。