Examples with DownloadManager gov.usgs.cida.coastalhazards.jpa.DownloadManager used on opensource projects

Example 1 with DownloadManager

use of gov.usgs.cida.coastalhazards.jpa.DownloadManager in project coastal-hazards by USGS-CIDA.

the class DownloadResource method getSession.

/**
 * TODO this is in need of refactor to get it in line with item download
 * Retrieves representation of an instance of
 * gov.usgs.cida.coastalhazards.model.Item
 *
 * @param id identifier of requested item
 * @return JSON representation of the item(s)
 * @throws java.io.IOException
 * @throws java.security.NoSuchAlgorithmException
 */
@GET
@Path("/view/{id}")
@Produces("application/zip")
public Response getSession(@PathParam("id") String id) throws IOException, NoSuchAlgorithmException {
    Response response = null;
    String sessionJSON = sessionManager.load(id);
    try (DownloadManager downloadManager = new DownloadManager()) {
        if (sessionJSON == null) {
            throw new NotFoundException();
        } else {
            File zipFile = null;
            Download download;
            try {
                download = downloadManager.load(id);
                if (download != null && download.getPersistanceURI() != null) {
                    zipFile = new File(new URI(download.getPersistanceURI()));
                    if (!zipFile.exists()) {
                        throw new FileNotFoundException();
                    }
                } else {
                    throw new FileNotFoundException();
                }
            } catch (FileNotFoundException | URISyntaxException ex) {
                Session session = Session.fromJSON(sessionJSON);
                File stagingDir = DownloadUtility.createDownloadStagingArea();
                boolean staged = DownloadUtility.stageSessionDownload(session, stagingDir);
                if (staged) {
                    download = new Download();
                    download = DownloadUtility.zipStagingAreaForDownload(stagingDir, download);
                    download.setSessionId(id);
                    zipFile = download.fetchZipFile();
                    if (zipFile == null) {
                        throw new DownloadStagingUnsuccessfulException();
                    }
                    downloadManager.save(download);
                } else {
                    throw new DownloadStagingUnsuccessfulException();
                }
            }
            String contentDisposition = "attachment; filename=\"" + id + ".zip\"";
            response = Response.ok(zipFile, "application/zip").header("Content-Disposition", contentDisposition).build();
        }
    }
    return response;
}

Example 2 with DownloadManager

use of gov.usgs.cida.coastalhazards.jpa.DownloadManager in project coastal-hazards by USGS-CIDA.

the class DownloadResource method downloadItem.

/**
 * Downloads a zip file containing the contents of
 * gov.usgs.cida.coastalhazards.model.Item
 *
 * @param id identifier of requested item
 * @return JSON representation of the item(s)
 * @throws java.io.IOException
 * @throws java.lang.InterruptedException
 * @throws java.util.concurrent.ExecutionException
 */
@GET
@Path("/item/{id}")
@Produces("application/zip")
public Response downloadItem(@PathParam("id") String id) throws IOException, InterruptedException, ExecutionException {
    Response response;
    Item item;
    try (ItemManager itemManager = new ItemManager()) {
        item = itemManager.load(id);
    }
    if (item == null) {
        response = Response.status(NOT_FOUND).build();
    } else {
        Download download;
        try (DownloadManager downloadManager = new DownloadManager()) {
            download = downloadManager.load(id);
        }
        if (download == null) {
            // Download was null, so we it was not previously staged. Do so now.
            LOG.debug("Download manager could not find download for item {}. A download will be staged for this item.", id);
            DownloadUtility.stageAsyncItemDownload(id);
            response = Response.status(ACCEPTED).build();
        } else {
            LOG.debug("Download manager found a download for item id {}", id);
            if (download.isProblem()) {
                LOG.debug("Download manager found a problem with download for item id {}", id);
                response = Response.status(NOT_IMPLEMENTED).build();
            } else if (StringUtils.isBlank(download.getPersistanceURI())) {
                LOG.debug("Download manager found a download with no path id {}, Item is still being created", id);
                response = Response.status(ACCEPTED).build();
            } else {
                boolean downloadFileExists;
                try (DownloadManager downloadManager = new DownloadManager()) {
                    downloadFileExists = downloadManager.downloadFileExistsOnFilesystem(download);
                }
                // Download should be on the file system. Check that it does exist
                if (!downloadFileExists) {
                    LOG.debug("Download manager found a download path that doesn't exist for id {}. Will delete and re-stage", id);
                    try (DownloadService svc = new DownloadService()) {
                        svc.delete(id);
                    }
                    DownloadUtility.stageAsyncItemDownload(id);
                    response = Response.status(ACCEPTED).build();
                } else {
                    // File was found. Load the zip file
                    File zipFile = download.fetchZipFile();
                    if (zipFile == null) {
                        LOG.debug("Download manager found could not find the zip file that was indicated in the database for item {}. Will attempt to re-stage", id);
                        try (DownloadService svc = new DownloadService()) {
                            svc.delete(id);
                        }
                        DownloadUtility.stageAsyncItemDownload(id);
                        response = Response.status(ACCEPTED).build();
                    } else {
                        String contentDisposition = "attachment; filename=\"" + id + ".zip\"";
                        response = Response.status(OK).entity(zipFile).type("application/zip").header("Content-Disposition", contentDisposition).build();
                    }
                }
            }
        }
    }
    return response;
}

Example 3 with DownloadManager

use of gov.usgs.cida.coastalhazards.jpa.DownloadManager in project coastal-hazards by USGS-CIDA.

the class DownloadResource method checkItemAvailability.

@HEAD
@Path("/item/{id}")
public Response checkItemAvailability(@PathParam("id") String id) throws IOException {
    Response response;
    Item item;
    try (ItemManager itemManager = new ItemManager()) {
        item = itemManager.load(id);
    }
    if (item == null) {
        response = Response.status(NOT_FOUND).build();
    } else {
        Download download;
        try (DownloadManager downloadManager = new DownloadManager()) {
            download = downloadManager.load(id);
        }
        if (download != null) {
            LOG.debug("Download manager found a download for item id {}", id);
            String persistenceURI = download.getPersistanceURI();
            // If it is null or blank, the download has been accepted
            if (download.isProblem()) {
                LOG.debug("Download manager found a problem with download for item id {}", id);
                response = Response.status(INTERNAL_SERVER_ERROR).build();
            } else if (StringUtils.isBlank(persistenceURI)) {
                LOG.debug("Download manager found a download with no path id {}, Item is probably still being created", id);
                response = Response.status(ACCEPTED).build();
            } else {
                // If it is has content, check whether the file exists on the server
                // If it does not, that means this is a desynchronized entry in the datbase
                // and it should be deleted and reinitialized. Otherwise, this is
                // is good to go and send an OK response
                boolean existsOnFileSystem;
                try (DownloadManager downloadManager = new DownloadManager()) {
                    existsOnFileSystem = downloadManager.downloadFileExistsOnFilesystem(download);
                }
                if (!existsOnFileSystem) {
                    LOG.debug("Download manager found a download path that doesn't exist for id {}. Will delete and re-stage", id);
                    try (DownloadService svc = new DownloadService()) {
                        svc.delete(id);
                    }
                    LOG.debug("Download path for item {} was deleted in the database. Will not attempt to re-stage", id);
                    DownloadUtility.stageAsyncItemDownload(id);
                    response = Response.status(ACCEPTED).build();
                } else {
                    LOG.debug("Download manager found the download for item {}", id);
                    response = Response.status(OK).build();
                }
            }
        } else {
            LOG.debug("Download manager could not find download for item {}. A download will be staged for this item.", id);
            DownloadUtility.stageAsyncItemDownload(id);
            response = Response.status(ACCEPTED).build();
        }
    }
    return response;
}

Example 4 with DownloadManager

use of gov.usgs.cida.coastalhazards.jpa.DownloadManager in project coastal-hazards by USGS-CIDA.

the class DownloadResource method displayStagedItems.

@GET
@Produces("application/json")
public Response displayStagedItems() {
    String downloadJson;
    try (DownloadManager downloadManager = new DownloadManager()) {
        List<Download> allStagedDownloads = downloadManager.getAllStagedDownloads();
        Gson serializer = GsonUtil.getDefault();
        downloadJson = serializer.toJson(allStagedDownloads, ArrayList.class);
    }
    return Response.ok(downloadJson, MediaType.APPLICATION_JSON_TYPE).build();
}