我一直在使用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字节数组。
基本上,我不会像接收到的那样快地将收到的数据写入文件。我对线程有点新意,所以我不知道如何摆脱这种竞争条件。任何提示?
答案 0 :(得分:3)
这是经典的消费者/制作人。最直接/最简单的方法是使用BlockingQueue。制作人调用put()
,消费者调用take()
。
请注意,使用BlockingQueue通常会导致“我如何完成”问题。最好的方法是使用“毒丸”方法,生产者在队列上贴上一个“特殊”值,告知消费者没有更多数据。