org.eclipse.hawkbit.mgmt.rest.resource.MgmtDownloadResource.java Source code

Introduction

Here is the source code for org.eclipse.hawkbit.mgmt.rest.resource.MgmtDownloadResource.java

Source

/**
 * Copyright (c) 2015 Bosch Software Innovations GmbH and others.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 */
package org.eclipse.hawkbit.mgmt.rest.resource;

import java.io.IOException;

import org.apache.commons.io.IOUtils;
import org.eclipse.hawkbit.artifact.repository.ArtifactRepository;
import org.eclipse.hawkbit.artifact.repository.model.DbArtifact;
import org.eclipse.hawkbit.cache.CacheConstants;
import org.eclipse.hawkbit.cache.DownloadArtifactCache;
import org.eclipse.hawkbit.cache.DownloadType;
import org.eclipse.hawkbit.mgmt.rest.api.MgmtDownloadRestApi;
import org.eclipse.hawkbit.rest.util.RequestResponseContextHolder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.cache.Cache;
import org.springframework.cache.Cache.ValueWrapper;
import org.springframework.context.annotation.Scope;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.context.WebApplicationContext;

/**
 * A resource for download artifacts.
 * 
 *
 *
 */
@RestController
@Scope(value = WebApplicationContext.SCOPE_REQUEST)
public class MgmtDownloadResource implements MgmtDownloadRestApi {

    private static final Logger LOGGER = LoggerFactory.getLogger(MgmtDownloadResource.class);

    @Autowired
    private ArtifactRepository artifactRepository;

    @Autowired
    @Qualifier(CacheConstants.DOWNLOAD_ID_CACHE)
    private Cache cache;

    @Autowired
    private RequestResponseContextHolder requestResponseContextHolder;

    /**
     * Handles the GET request for downloading an artifact.
     * 
     * @param downloadId
     *            the generated download id
     * @return {@link ResponseEntity} with status {@link HttpStatus#OK} if
     *         successful
     */
    @Override
    @ResponseBody
    public ResponseEntity<Void> downloadArtifactByDownloadId(@PathVariable("downloadId") final String downloadId) {
        try {
            final ValueWrapper cacheWrapper = cache.get(downloadId);
            if (cacheWrapper == null) {
                LOGGER.warn("Download Id {} could not be found", downloadId);
                return new ResponseEntity<>(HttpStatus.NOT_FOUND);
            }

            final DownloadArtifactCache artifactCache = (DownloadArtifactCache) cacheWrapper.get();
            DbArtifact artifact = null;

            if (DownloadType.BY_SHA1.equals(artifactCache.getDownloadType())) {
                artifact = artifactRepository.getArtifactBySha1(artifactCache.getId());
            } else {
                LOGGER.warn("Download Type {} not supported", artifactCache.getDownloadType());
            }

            if (artifact == null) {
                LOGGER.warn("Artifact with cached id {} and download type {} could not be found.",
                        artifactCache.getId(), artifactCache.getDownloadType());
                return new ResponseEntity<>(HttpStatus.NOT_FOUND);
            }
            try {
                IOUtils.copy(artifact.getFileInputStream(),
                        requestResponseContextHolder.getHttpServletResponse().getOutputStream());
            } catch (final IOException e) {
                LOGGER.error("Cannot copy streams", e);
                return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR).build();
            }
        } finally {
            cache.evict(downloadId);
        }

        return ResponseEntity.ok().build();
    }
}