如何确定DataHandler.writeTo()方法当前写入的字节数

Question

我必须使用MTOM从web服务下载文件。下面的实施细节我不知道。我这样做，这是有效的(响应是从webservices返回的对象，数据是javax.activation.DataHandler类型，filePath是文件，我想保存下载的内容):

DataHandler dataHandler = response.getData();
File file = new File(filePath);
FileOutputStream fileOutputStream = new FileOutputStream(file);
dataHandler.writeTo(fileOutputStream);
fileOutputStream.flush();
fileOutputStream.close();

但是，我需要弄清楚，在下载过程中写入了多少字节。它可能需要很长时间，这种方式只是等待完成下载，然后关闭输出流并继续。

我知道，我可能需要另一个线程来读取写入的字节数并使用它做一些事情(需要它来更新SWT应用程序中的progressBar)。

我的想法是以某种方式扩展DataHandler，然后定期询问，写入了多少字节)。但是我是java新手，我不知道如何实现它。

Answer 1

如果您将所有内容读取到byte[]，则需要足够的内存来保存整个文件。现在这对您来说可能不是问题，但是最好不要创建这个坏习惯(它很容易使服务器停机)。

Apache Commons和Google Guava有一个CountingOutputStream类，它可以完全满足您的需求，只需计算写入了多少字节。如果你不能使用这个库，你可以写一个类似的类，只是实现OutputStream，在构造函数中接收另一个OutputStream，每个方法只是增加byteCount和委托。

Answer 2

我做了一些研究，之前忽略了名为getInputStream()的DataHandler方法。使用这个方法，我可以做我需要的:

DataHandler dataHandler = response.getData();
File file = new File(filePath);
FileOutputStream fileOutputStream = new FileOutputStream(file);
// got Input Stream from DataHandler
InputStream is = dataHandler.getInputStream();
// then I can manage writing file my way        
byte data[] = new byte[BUFFER_SIZE];
int br;
Long bytesWritten = 0L;
while ((br = is.read(data)) > 0) {
    fileOutputStream.write(data, 0, br);
    bytesWritten = bytesWritten + br;
    // call some method to update UI progressBar
}
// this is now useless:
// dataHandler.writeTo(fileOutputStream);
fileOutputStream.flush();
fileOutputStream.close();