Examples with TransferStreamingOutput org.commonjava.indy.core.bind.jaxrs.util.TransferStreamingOutput used on opensource projects

Example 1 with TransferStreamingOutput

use of org.commonjava.indy.core.bind.jaxrs.util.TransferStreamingOutput in project indy by Commonjava.

the class ContentAccessHandler method doGet.

public Response doGet(final String packageType, final String type, final String name, final String path, final String baseUri, final HttpServletRequest request, EventMetadata eventMetadata, final Consumer<ResponseBuilder> builderModifier) {
    if (!PackageTypes.contains(packageType)) {
        ResponseBuilder builder = Response.status(400);
        if (builderModifier != null) {
            builderModifier.accept(builder);
        }
        return builder.build();
    }
    final StoreType st = StoreType.get(type);
    final StoreKey sk = new StoreKey(packageType, st, name);
    eventMetadata = eventMetadata.set(ContentManager.ENTRY_POINT_STORE, sk);
    final AcceptInfo acceptInfo = jaxRsRequestHelper.findAccept(request, ApplicationContent.text_html);
    final String standardAccept = ApplicationContent.getStandardAccept(acceptInfo.getBaseAccept());
    Response response = null;
    logger.info("GET path: '{}' (RAW: '{}')\nIn store: '{}'\nUser addMetadata header is: '{}'\nStandard addMetadata header for that is: '{}'", path, request.getPathInfo(), sk, acceptInfo.getRawAccept(), standardAccept);
    if (path == null || path.equals("") || request.getPathInfo().endsWith("/") || path.endsWith(LISTING_HTML_FILE)) {
        try {
            logger.info("Getting listing at: {}", path);
            final String content = contentController.renderListing(standardAccept, st, name, path, baseUri, uriFormatter);
            response = formatOkResponseWithEntity(content, acceptInfo.getRawAccept(), builderModifier);
        } catch (final IndyWorkflowException e) {
            logger.error(String.format("Failed to render content listing: %s from: %s. Reason: %s", path, name, e.getMessage()), e);
            response = formatResponse(e, builderModifier);
        }
    } else {
        try {
            logger.info("START: retrieval of content: {}:{}", sk, path);
            final Transfer item = contentController.get(sk, path, eventMetadata);
            logger.info("HANDLE: retrieval of content: {}:{}", sk, path);
            if (item == null) {
                return handleMissingContentQuery(sk, path, builderModifier);
            }
            boolean handleLocking = false;
            if (!item.isWriteLocked()) {
                item.lockWrite();
                handleLocking = true;
            }
            try {
                if (!item.exists()) {
                    return handleMissingContentQuery(sk, path, builderModifier);
                } else if (item.isDirectory()) {
                    try {
                        logger.info("Getting listing at: {}", path + "/");
                        final String content = contentController.renderListing(standardAccept, st, name, path + "/", baseUri, uriFormatter);
                        response = formatOkResponseWithEntity(content, acceptInfo.getRawAccept(), builderModifier);
                    } catch (final IndyWorkflowException e) {
                        logger.error(String.format("Failed to render content listing: %s from: %s. Reason: %s", path, name, e.getMessage()), e);
                        response = formatResponse(e, builderModifier);
                    }
                } else {
                    logger.info("RETURNING: retrieval of content: {}:{}", sk, path);
                    // open the stream here to prevent deletion while waiting for the transfer back to the user to start...
                    InputStream in = item.openInputStream(true, eventMetadata);
                    final ResponseBuilder builder = Response.ok(new TransferStreamingOutput(in));
                    setInfoHeaders(builder, item, sk, path, true, contentController.getContentType(path), contentController.getHttpMetadata(item));
                    if (builderModifier != null) {
                        builderModifier.accept(builder);
                    }
                    response = builder.build();
                }
            } finally {
                if (handleLocking) {
                    item.unlock();
                }
            }
        } catch (final IOException | IndyWorkflowException e) {
            logger.error(String.format("Failed to download artifact: %s from: %s. Reason: %s", path, name, e.getMessage()), e);
            response = formatResponse(e, builderModifier);
        }
    }
    logger.info("RETURNING RESULT: {}:{}", sk, path);
    return response;
}