Jersey webservice可扩展的方法来下载文件并回复客户端

时间:2016-02-12 16:08:44

标签: java web-services jersey scalability nonblocking

我需要使用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会下载完整的文件,然后将其写入客户端,这不是我想要做的。 任何想法??

1 个答案:

答案 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客户端正在跟踪结果

enter image description here

从中可以看出,“远程服务器”日志记录与客户端资源的日志记录同时发生。这表明客户端不等待接收整个文件。它一开始接收就开始发送字节。

有关示例的一些注意事项:

  • 我使用了一个非常小的缓冲区大小(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());
}