我需要使用Jersey构建一个Web服务,从另一个服务下载一个大文件并返回给客户端。 我希望jersey将一些字节读入缓冲区并将这些字节写入客户端套接字。
我希望它使用非阻塞I / O,所以我不要让线程忙。 (这无法实现)
@GET
@Path("mypath")
public void getFile(final @Suspended AsyncResponse res) {
Client client = ClientBuilder.newClient();
WebTarget t = client.target("http://webserviceURL");
t.request()
.header("some header", "value for header")
.async().get(new InvocationCallback<byte[]>(){
public void completed(byte[] response) {
res.resume(response);
}
public void failed(Throwable throwable) {
res.resume(throwable.getMessage());
throwable.printStackTrace();
//reply with error
}
});
}
到目前为止,我有这个代码,我相信Jersey会下载完整的文件,然后将其写入客户端,这不是我想要做的。 任何想法??
答案 0 :(得分:4)
客户端异步请求对您的用例不会有太大作用。这对于“火灾和遗忘”用例来说更具意义。您可以做的只是从客户端InputStream
获取Response
并与服务器端StreamingResource
混合以流式传输结果。服务器将从其他远程资源进入时开始发送数据。
以下是一个例子。 "/file"
端点是提供文件的虚拟远程资源。 "/client"
端点使用它。
@Path("stream")
@Produces(MediaType.APPLICATION_OCTET_STREAM)
public class ClientStreamingResource {
private static final String INFILE = "Some File";
@GET
@Path("file")
public Response fileEndpoint() {
final File file = new File(INFILE);
final StreamingOutput output = new StreamingOutput() {
@Override
public void write(OutputStream out) {
try (FileInputStream in = new FileInputStream(file)) {
byte[] buf = new byte[512];
int len;
while ((len = in.read(buf)) != -1) {
out.write(buf, 0, len);
out.flush();
System.out.println("---- wrote 512 bytes file ----");
}
} catch (IOException ex) {
throw new InternalServerErrorException(ex);
}
}
};
return Response.ok(output)
.header(HttpHeaders.CONTENT_LENGTH, file.length())
.build();
}
@GET
@Path("client")
public void clientEndpoint(@Suspended final AsyncResponse asyncResponse) {
final Client client = ClientBuilder.newClient();
final WebTarget target = client.target("http://localhost:8080/stream/file");
final Response clientResponse = target.request().get();
final StreamingOutput output = new StreamingOutput() {
@Override
public void write(OutputStream out) {
try (final InputStream entityStream = clientResponse.readEntity(InputStream.class)) {
byte[] buf = new byte[512];
int len;
while ((len = entityStream.read(buf)) != -1) {
out.write(buf, 0, len);
out.flush();
System.out.println("---- wrote 512 bytes client ----");
}
} catch (IOException ex) {
throw new InternalServerErrorException(ex);
}
}
};
ResponseBuilder responseBuilder = Response.ok(output);
if (clientResponse.getHeaderString("Content-Length") != null) {
responseBuilder.header("Content-Length", clientResponse.getHeaderString("Content-Length"));
}
new Thread(() -> {
asyncResponse.resume(responseBuilder.build());
}).start();
}
}
我使用cURL
发出请求,jetty-maven-plugin
能够从命令行运行示例。当您运行它并发出请求时,您应该看到服务器记录
---- wrote 512 bytes file ----
---- wrote 512 bytes file ----
---- wrote 512 bytes client ----
---- wrote 512 bytes file ----
---- wrote 512 bytes client ----
---- wrote 512 bytes file ----
---- wrote 512 bytes client ----
---- wrote 512 bytes file ----
---- wrote 512 bytes client ----
...
而cURL
客户端正在跟踪结果
从中可以看出,“远程服务器”日志记录与客户端资源的日志记录同时发生。这表明客户端不等待接收整个文件。它一开始接收就开始发送字节。
我使用了一个非常小的缓冲区大小(512),因为我正在测试一个小的(1Mb)文件。我真的不想等待大文件进行测试。但我认为大文件应该可以正常工作。当然,您需要将缓冲区大小增加到更大的范围。
为了使用较小的缓冲区大小,您需要将Jersey属性ServerProperties.OUTBOUND_CONTENT_LENGTH_BUFFER
设置为0.原因是Jersey保留在大小为8192的内部缓冲区中,这将导致我的512字节块数据不刷新,直到缓冲8192个字节。所以我只是禁用了它。
使用AsyncResponse
时,您应该像我一样使用其他线程。您可能希望使用执行程序而不是显式创建线程。如果你不使用另一个线程,那么你仍然会从容器的线程池中占用线程。
您可以使用@ManagedAsync
注释客户端资源,而不是管理自己的线程/执行者,让泽西岛管理线程
@ManagedAsync
@GET
@Path("client")
public void clientEndpoint(@Suspended final AsyncResponse asyncResponse) {
...
asyncResponse.resume(responseBuilder.build());
}