Входные и выходные двоичные потоки с использованием JERSEY?

111

Я использую Джерси для реализации RESTful API, который в основном извлекает и обслуживает данные в кодировке JSON. Но у меня бывают ситуации, когда мне нужно выполнить следующее:

  • Экспортируйте загружаемые документы, например PDF, XLS, ZIP или другие двоичные файлы.
  • Получение составных данных, таких как JSON плюс загруженный файл XLS

У меня есть одностраничный веб-клиент на основе JQuery, который создает вызовы AJAX для этой веб-службы. На данный момент он не отправляет формы и использует GET и POST (с объектом JSON). Должен ли я использовать сообщение формы для отправки данных и прикрепленного двоичного файла, или я могу создать многостраничный запрос с JSON плюс двоичный файл?

Уровень обслуживания моего приложения в настоящее время создает ByteArrayOutputStream при создании файла PDF. Как лучше всего вывести этот поток клиенту через Джерси? Я создал MessageBodyWriter, но не знаю, как его использовать из ресурса Джерси. Это правильный подход?

Я просматривал образцы, включенные в Jersey, но пока не нашел ничего, что иллюстрирует, как сделать что-то из этого. Если это важно, я использую Джерси с Джексоном для выполнения Object-> JSON без шага XML и на самом деле не использую JAX-RS.

Таурен
источник

Ответы:

109

Мне удалось получить ZIP-файл или PDF-файл, расширив StreamingOutputобъект. Вот пример кода:

@Path("PDF-file.pdf/")
@GET
@Produces({"application/pdf"})
public StreamingOutput getPDF() throws Exception {
    return new StreamingOutput() {
        public void write(OutputStream output) throws IOException, WebApplicationException {
            try {
                PDFGenerator generator = new PDFGenerator(getEntity());
                generator.generatePDF(output);
            } catch (Exception e) {
                throw new WebApplicationException(e);
            }
        }
    };
}

Класс PDFGenerator (мой собственный класс для создания PDF-файла) берет выходной поток из метода записи и записывает в него вместо вновь созданного выходного потока.

Не знаю, лучший ли это способ сделать, но он работает.

MikeTheReader
источник
33
Также возможно вернуть StreamingOutput как объект в Responseобъект. Таким образом, вы можете легко управлять типом медиа, кодом ответа HTTP и т. Д. Сообщите мне, если вы хотите, чтобы я разместил код.
Hank
3
@MyTitle: см. Пример
Хэнк,
3
Я использовал примеры кода в этом потоке в качестве справочника и обнаружил, что мне нужно очистить OutputStream в StreamingOutput.write (), чтобы клиент мог надежно получать выходные данные. В противном случае я иногда получал бы «Content-Length: 0» в заголовках и без тела, хотя журналы говорили мне, что StreamingOutput выполняется.
Джон Стюарт
@JonStewart - Мне кажется, я выполнял промывку в методе generatePDF.
MikeTheReader
1
@ Dante617. Не могли бы вы опубликовать код на стороне клиента, как клиент Джерси отправляет двоичный поток на сервер (с jersey 2.x)?
Дебора
29

Мне пришлось вернуть файл rtf, и это сработало для меня.

// create a byte array of the file in correct format
byte[] docStream = createDoc(fragments); 

return Response
            .ok(docStream, MediaType.APPLICATION_OCTET_STREAM)
            .header("content-disposition","attachment; filename = doc.rtf")
            .build();
Абхишек Ракшит
источник
26
Не очень хорошо, потому что вывод отправляется только после того, как он полностью подготовлен. Байт [] не является потоком.
java.is.for.desktop
7
Это потребляет все байты в памяти, что означает, что большие файлы могут вывести из строя сервер. Цель потоковой передачи - избежать использования всех байтов в памяти.
Роберт Кристиан
22

Я использую этот код для экспорта файла Excel (xlsx) (Apache Poi) в трикотаж в качестве вложения.

@GET
@Path("/{id}/contributions/excel")
@Produces("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
public Response exportExcel(@PathParam("id") Long id)  throws Exception  {

    Resource resource = new ClassPathResource("/xls/template.xlsx");

    final InputStream inp = resource.getInputStream();
    final Workbook wb = WorkbookFactory.create(inp);
    Sheet sheet = wb.getSheetAt(0);

    Row row = CellUtil.getRow(7, sheet);
    Cell cell = CellUtil.getCell(row, 0);
    cell.setCellValue("TITRE TEST");

    [...]

    StreamingOutput stream = new StreamingOutput() {
        public void write(OutputStream output) throws IOException, WebApplicationException {
            try {
                wb.write(output);
            } catch (Exception e) {
                throw new WebApplicationException(e);
            }
        }
    };


    return Response.ok(stream).header("content-disposition","attachment; filename = export.xlsx").build();

}
Грегори
источник
15

Вот еще один пример. Я создаю QRCode как PNG через файл ByteArrayOutputStream. Ресурс возвращает Responseобъект, а данные потока являются сущностью.

Чтобы проиллюстрировать обработку кода ответа, я добавил обработку заголовков кеша ( If-modified-since, If-none-matchesи т. Д.).

@Path("{externalId}.png")
@GET
@Produces({"image/png"})
public Response getAsImage(@PathParam("externalId") String externalId, 
        @Context Request request) throws WebApplicationException {

    ByteArrayOutputStream stream = new ByteArrayOutputStream();
    // do something with externalId, maybe retrieve an object from the
    // db, then calculate data, size, expirationTimestamp, etc

    try {
        // create a QRCode as PNG from data     
        BitMatrix bitMatrix = new QRCodeWriter().encode(
                data, 
                BarcodeFormat.QR_CODE, 
                size, 
                size
        );
        MatrixToImageWriter.writeToStream(bitMatrix, "png", stream);

    } catch (Exception e) {
        // ExceptionMapper will return HTTP 500 
        throw new WebApplicationException("Something went wrong …")
    }

    CacheControl cc = new CacheControl();
    cc.setNoTransform(true);
    cc.setMustRevalidate(false);
    cc.setNoCache(false);
    cc.setMaxAge(3600);

    EntityTag etag = new EntityTag(HelperBean.md5(data));

    Response.ResponseBuilder responseBuilder = request.evaluatePreconditions(
            updateTimestamp,
            etag
    );
    if (responseBuilder != null) {
        // Preconditions are not met, returning HTTP 304 'not-modified'
        return responseBuilder
                .cacheControl(cc)
                .build();
    }

    Response response = Response
            .ok()
            .cacheControl(cc)
            .tag(etag)
            .lastModified(updateTimestamp)
            .expires(expirationTimestamp)
            .type("image/png")
            .entity(stream.toByteArray())
            .build();
    return response;
}   

Пожалуйста, не бейте меня, если у меня stream.toByteArray()нет памяти :) Это работает с моими файлами PNG размером <1 КБ ...

Хэнк
источник
6
Я думаю, что это плохой пример потоковой передачи, поскольку возвращаемый объект на выходе представляет собой массив байтов, а не поток.
AlikElzin-kilaka 03
Хороший пример для создания ответа на запрос ресурса GET, а не для потока. Это вообще не ручей.
Роберт Кристиан
14

Я составлял свои сервисы Jersey 1.17 следующим образом:

FileStreamingOutput

public class FileStreamingOutput implements StreamingOutput {

    private File file;

    public FileStreamingOutput(File file) {
        this.file = file;
    }

    @Override
    public void write(OutputStream output)
            throws IOException, WebApplicationException {
        FileInputStream input = new FileInputStream(file);
        try {
            int bytes;
            while ((bytes = input.read()) != -1) {
                output.write(bytes);
            }
        } catch (Exception e) {
            throw new WebApplicationException(e);
        } finally {
            if (output != null) output.close();
            if (input != null) input.close();
        }
    }

}

GET

@GET
@Produces("application/pdf")
public StreamingOutput getPdf(@QueryParam(value="name") String pdfFileName) {
    if (pdfFileName == null)
        throw new WebApplicationException(Response.Status.BAD_REQUEST);
    if (!pdfFileName.endsWith(".pdf")) pdfFileName = pdfFileName + ".pdf";

    File pdf = new File(Settings.basePath, pdfFileName);
    if (!pdf.exists())
        throw new WebApplicationException(Response.Status.NOT_FOUND);

    return new FileStreamingOutput(pdf);
}

И клиент, если он вам нужен:

Client

private WebResource resource;

public InputStream getPDFStream(String filename) throws IOException {
    ClientResponse response = resource.path("pdf").queryParam("name", filename)
        .type("application/pdf").get(ClientResponse.class);
    return response.getEntityInputStream();
}
Даниэль Салай
источник
7

В этом примере показано, как опубликовать файлы журналов в JBoss через ресурс отдыха. Обратите внимание, что метод get использует интерфейс StreamingOutput для потоковой передачи содержимого файла журнала.

@Path("/logs/")
@RequestScoped
public class LogResource {

private static final Logger logger = Logger.getLogger(LogResource.class.getName());
@Context
private UriInfo uriInfo;
private static final String LOG_PATH = "jboss.server.log.dir";

public void pipe(InputStream is, OutputStream os) throws IOException {
    int n;
    byte[] buffer = new byte[1024];
    while ((n = is.read(buffer)) > -1) {
        os.write(buffer, 0, n);   // Don't allow any extra bytes to creep in, final write
    }
    os.close();
}

@GET
@Path("{logFile}")
@Produces("text/plain")
public Response getLogFile(@PathParam("logFile") String logFile) throws URISyntaxException {
    String logDirPath = System.getProperty(LOG_PATH);
    try {
        File f = new File(logDirPath + "/" + logFile);
        final FileInputStream fStream = new FileInputStream(f);
        StreamingOutput stream = new StreamingOutput() {
            @Override
            public void write(OutputStream output) throws IOException, WebApplicationException {
                try {
                    pipe(fStream, output);
                } catch (Exception e) {
                    throw new WebApplicationException(e);
                }
            }
        };
        return Response.ok(stream).build();
    } catch (Exception e) {
        return Response.status(Response.Status.CONFLICT).build();
    }
}

@POST
@Path("{logFile}")
public Response flushLogFile(@PathParam("logFile") String logFile) throws URISyntaxException {
    String logDirPath = System.getProperty(LOG_PATH);
    try {
        File file = new File(logDirPath + "/" + logFile);
        PrintWriter writer = new PrintWriter(file);
        writer.print("");
        writer.close();
        return Response.ok().build();
    } catch (Exception e) {
        return Response.status(Response.Status.CONFLICT).build();
    }
}    

}

Хайме Касеро
источник
1
Просто к сведению: вместо метода канала вы также можете использовать IOUtils.copy из ввода-вывода общего доступа Apache.
David
7

Использование Jersey 2.16. Загрузить файл очень просто.

Ниже приведен пример ZIP-файла

@GET
@Path("zipFile")
@Produces("application/zip")
public Response getFile() {
    File f = new File(ZIP_FILE_PATH);

    if (!f.exists()) {
        throw new WebApplicationException(404);
    }

    return Response.ok(f)
            .header("Content-Disposition",
                    "attachment; filename=server.zip").build();
}
апельсинжирафа
источник
1
Работает как шарм. Есть хоть какое-то представление об этих потоковых вещах, я не совсем понимаю ...
Оливер,
1
Это самый простой способ, если вы используете Джерси, спасибо
ganchito55
Можно ли сделать с @POST вместо @GET?
spr
@spr Думаю, да, это возможно. Когда страница сервера отвечает, она должна предоставить окно загрузки
orangegiraffa
5

Я нашел следующее полезным для меня, и я хотел поделиться, если это поможет вам или кому-то еще. Я хотел что-то вроде MediaType.PDF_TYPE, которого не существует, но этот код делает то же самое:

DefaultMediaTypePredictor.CommonMediaTypes.
        getMediaTypeFromFileName("anything.pdf")

См. Http://jersey.java.net/nonav/apidocs/1.1.0-ea/contribs/jersey-multipart/com/sun/jersey/multipart/file/DefaultMediaTypePredictor.CommonMediaTypes.html

В моем случае я отправлял PDF-документ на другой сайт:

FormDataMultiPart p = new FormDataMultiPart();
p.bodyPart(new FormDataBodyPart(FormDataContentDisposition
        .name("fieldKey").fileName("document.pdf").build(),
        new File("path/to/document.pdf"),
        DefaultMediaTypePredictor.CommonMediaTypes
                .getMediaTypeFromFileName("document.pdf")));

Затем p передается в качестве второго параметра функции post ().

Эта ссылка помогла мне собрать этот фрагмент кода: http://jersey.576304.n2.nabble.com/Multipart-Post-td4252846.html

Довев Хефец
источник
4

Это отлично сработало со мной url: http://example.com/rest/muqsith/get-file?filePath=C : \ Users \ I066807 \ Desktop \ test.xml

@GET
@Produces({ MediaType.APPLICATION_OCTET_STREAM })
@Path("/get-file")
public Response getFile(@Context HttpServletRequest request){
   String filePath = request.getParameter("filePath");
   if(filePath != null && !"".equals(filePath)){
        File file = new File(filePath);
        StreamingOutput stream = null;
        try {
        final InputStream in = new FileInputStream(file);
        stream = new StreamingOutput() {
            public void write(OutputStream out) throws IOException, WebApplicationException {
                try {
                    int read = 0;
                        byte[] bytes = new byte[1024];

                        while ((read = in.read(bytes)) != -1) {
                            out.write(bytes, 0, read);
                        }
                } catch (Exception e) {
                    throw new WebApplicationException(e);
                }
            }
        };
    } catch (FileNotFoundException e) {
        e.printStackTrace();
    }
        return Response.ok(stream).header("content-disposition","attachment; filename = "+file.getName()).build();
        }
    return Response.ok("file path null").build();
}
Муксит
источник
1
Не уверен Response.ok("file path null").build();, действительно ли это нормально? Вам, вероятно, следует использовать что-то вродеResponse.status(Status.BAD_REQUEST).entity(...
Christophe Roussy
1

Еще один пример кода, в котором вы можете загрузить файл в службу REST, служба REST заархивирует файл, а клиент загрузит ZIP-файл с сервера. Это хороший пример использования двоичных входных и выходных потоков с помощью Джерси.

https://stackoverflow.com/a/32253028/15789

Этот ответ был опубликован мной в другой ветке. Надеюсь это поможет.

RuntimeException
источник