Examples with ImportMode edu.harvard.iq.dataverse.batch.jobs.importer.ImportMode used on opensource projects

Example 1 with ImportMode

use of edu.harvard.iq.dataverse.batch.jobs.importer.ImportMode in project dataverse by IQSS.

the class Datasets method receiveChecksumValidationResults.

@POST
@Path("{identifier}/dataCaptureModule/checksumValidation")
public Response receiveChecksumValidationResults(@PathParam("identifier") String id, JsonObject jsonFromDcm) {
    logger.log(Level.FINE, "jsonFromDcm: {0}", jsonFromDcm);
    AuthenticatedUser authenticatedUser = null;
    try {
        authenticatedUser = findAuthenticatedUserOrDie();
    } catch (WrappedResponse ex) {
        return error(Response.Status.BAD_REQUEST, "Authentication is required.");
    }
    if (!authenticatedUser.isSuperuser()) {
        return error(Response.Status.FORBIDDEN, "Superusers only.");
    }
    String statusMessageFromDcm = jsonFromDcm.getString("status");
    try {
        Dataset dataset = findDatasetOrDie(id);
        if ("validation passed".equals(statusMessageFromDcm)) {
            String uploadFolder = jsonFromDcm.getString("uploadFolder");
            int totalSize = jsonFromDcm.getInt("totalSize");
            ImportMode importMode = ImportMode.MERGE;
            try {
                JsonObject jsonFromImportJobKickoff = execCommand(new ImportFromFileSystemCommand(createDataverseRequest(findUserOrDie()), dataset, uploadFolder, new Long(totalSize), importMode));
                long jobId = jsonFromImportJobKickoff.getInt("executionId");
                String message = jsonFromImportJobKickoff.getString("message");
                JsonObjectBuilder job = Json.createObjectBuilder();
                job.add("jobId", jobId);
                job.add("message", message);
                return ok(job);
            } catch (WrappedResponse wr) {
                String message = wr.getMessage();
                return error(Response.Status.INTERNAL_SERVER_ERROR, "Uploaded files have passed checksum validation but something went wrong while attempting to put the files into Dataverse. Message was '" + message + "'.");
            }
        } else if ("validation failed".equals(statusMessageFromDcm)) {
            Map<String, AuthenticatedUser> distinctAuthors = permissionService.getDistinctUsersWithPermissionOn(Permission.EditDataset, dataset);
            distinctAuthors.values().forEach((value) -> {
                userNotificationService.sendNotification((AuthenticatedUser) value, new Timestamp(new Date().getTime()), UserNotification.Type.CHECKSUMFAIL, dataset.getId());
            });
            List<AuthenticatedUser> superUsers = authenticationServiceBean.findSuperUsers();
            if (superUsers != null && !superUsers.isEmpty()) {
                superUsers.forEach((au) -> {
                    userNotificationService.sendNotification(au, new Timestamp(new Date().getTime()), UserNotification.Type.CHECKSUMFAIL, dataset.getId());
                });
            }
            return ok("User notified about checksum validation failure.");
        } else {
            return error(Response.Status.BAD_REQUEST, "Unexpected status cannot be processed: " + statusMessageFromDcm);
        }
    } catch (WrappedResponse ex) {
        return ex.getResponse();
    }
}

Example 2 with ImportMode

use of edu.harvard.iq.dataverse.batch.jobs.importer.ImportMode in project dataverse by IQSS.

the class FileRecordJobResource method getFilesystemImport.

@POST
@Path("import/datasets/files/{identifier}")
@Produces(MediaType.APPLICATION_JSON)
public Response getFilesystemImport(@PathParam("identifier") String identifier, @QueryParam("mode") @DefaultValue("MERGE") String mode, /*@QueryParam("fileMode") @DefaultValue("package_file") String fileMode*/
@QueryParam("uploadFolder") String uploadFolder, @QueryParam("totalSize") Long totalSize) {
    return response(req -> {
        ImportMode importMode = ImportMode.MERGE;
        // Switch to this if you ever need to use something other than MERGE.
        // ImportMode importMode = ImportMode.valueOf(mode);
        JsonObject jsonObject = execCommand(new ImportFromFileSystemCommand(req, findDatasetOrDie(identifier), uploadFolder, totalSize, importMode));
        String returnString = jsonObject.getString("message");
        if (!returnString.equals("FileSystemImportJob in progress")) {
            return error(Response.Status.INTERNAL_SERVER_ERROR, returnString);
        }
        return ok(Json.createObjectBuilder().add("message", returnString).add("executionId", jsonObject.getInt("executionId")));
    });
}