Examples with NodeImporter org.alfresco.repo.bulkimport.NodeImporter used on opensource projects

Example 1 with NodeImporter

use of org.alfresco.repo.bulkimport.NodeImporter in project alfresco-remote-api by Alfresco.

the class BulkFilesystemImportWebScript method executeImpl.

/**
 * @see org.springframework.extensions.webscripts.DeclarativeWebScript#executeImpl(org.springframework.extensions.webscripts.WebScriptRequest, org.springframework.extensions.webscripts.Status, org.springframework.extensions.webscripts.Cache)
 */
@Override
protected Map<String, Object> executeImpl(final WebScriptRequest request, final Status status, final Cache cache) {
    Map<String, Object> model = new HashMap<String, Object>();
    String targetNodeRefStr = null;
    String targetPath = null;
    String sourceDirectoryStr = null;
    @Deprecated String replaceExistingStr = null;
    String existingFileModeStr = null;
    String batchSizeStr = null;
    String numThreadsStr = null;
    String disableRulesStr = null;
    cache.setNeverCache(true);
    try {
        if (!bulkImporter.getStatus().inProgress()) {
            NodeRef targetNodeRef = null;
            File sourceDirectory = null;
            boolean replaceExisting = false;
            BulkImportParameters.ExistingFileMode existingFileMode = null;
            int batchSize = bulkImporter.getDefaultBatchSize();
            int numThreads = bulkImporter.getDefaultNumThreads();
            boolean disableRules = false;
            // Retrieve, validate and convert parameters
            targetNodeRefStr = request.getParameter(PARAMETER_TARGET_NODEREF);
            targetPath = request.getParameter(PARAMETER_TARGET_PATH);
            sourceDirectoryStr = request.getParameter(PARAMETER_SOURCE_DIRECTORY);
            replaceExistingStr = request.getParameter(PARAMETER_REPLACE_EXISTING);
            existingFileModeStr = request.getParameter(PARAMETER_EXISTING_FILE_MODE);
            batchSizeStr = request.getParameter(PARAMETER_BATCH_SIZE);
            numThreadsStr = request.getParameter(PARAMETER_NUM_THREADS);
            disableRulesStr = request.getParameter(PARAMETER_DISABLE_RULES);
            targetNodeRef = getTargetNodeRef(targetNodeRefStr, targetPath);
            if (sourceDirectoryStr == null || sourceDirectoryStr.trim().length() == 0) {
                throw new RuntimeException("Error: mandatory parameter '" + PARAMETER_SOURCE_DIRECTORY + "' was not provided.");
            }
            sourceDirectory = new File(sourceDirectoryStr.trim());
            if (replaceExistingStr != null && existingFileModeStr != null) {
                // parameters supplied.
                throw new IllegalStateException(String.format("Only one of these parameters may be used, not both: %s, %s", PARAMETER_REPLACE_EXISTING, PARAMETER_EXISTING_FILE_MODE));
            }
            if (replaceExistingStr != null && replaceExistingStr.trim().length() > 0) {
                replaceExisting = PARAMETER_VALUE_REPLACE_EXISTING.equals(replaceExistingStr);
            }
            if (existingFileModeStr != null && existingFileModeStr.trim().length() > 0) {
                existingFileMode = BulkImportParameters.ExistingFileMode.valueOf(existingFileModeStr);
            }
            if (disableRulesStr != null && disableRulesStr.trim().length() > 0) {
                disableRules = PARAMETER_VALUE_DISABLE_RULES.equals(disableRulesStr);
            }
            // Initiate the import
            NodeImporter nodeImporter = nodeImporterFactory.getNodeImporter(sourceDirectory);
            BulkImportParameters bulkImportParameters = new BulkImportParameters();
            if (numThreadsStr != null && numThreadsStr.trim().length() > 0) {
                try {
                    numThreads = Integer.parseInt(numThreadsStr);
                    if (numThreads < 1) {
                        throw new RuntimeException("Error: parameter '" + PARAMETER_NUM_THREADS + "' must be an integer > 0.");
                    }
                    bulkImportParameters.setNumThreads(numThreads);
                } catch (NumberFormatException e) {
                    throw new RuntimeException("Error: parameter '" + PARAMETER_NUM_THREADS + "' must be an integer > 0.");
                }
            }
            if (batchSizeStr != null && batchSizeStr.trim().length() > 0) {
                try {
                    batchSize = Integer.parseInt(batchSizeStr);
                    if (batchSize < 1) {
                        throw new RuntimeException("Error: parameter '" + PARAMETER_BATCH_SIZE + "' must be an integer > 0.");
                    }
                    bulkImportParameters.setBatchSize(batchSize);
                } catch (NumberFormatException e) {
                    throw new RuntimeException("Error: parameter '" + PARAMETER_BATCH_SIZE + "' must be an integer > 0.");
                }
            }
            if (existingFileMode != null) {
                bulkImportParameters.setExistingFileMode(existingFileMode);
            } else {
                // Fall back to the old/deprecated way.
                bulkImportParameters.setReplaceExisting(replaceExisting);
            }
            bulkImportParameters.setTarget(targetNodeRef);
            bulkImportParameters.setDisableRulesService(disableRules);
            bulkImporter.asyncBulkImport(bulkImportParameters, nodeImporter);
            // ACE-3047 fix, since bulk import is started asynchronously there is a chance that client
            // will get into the status page before import is actually started.
            // In this case wrong information (for previous import) will be displayed.
            // So lets ensure that import started before redirecting client to status page.
            int i = 0;
            while (!bulkImporter.getStatus().inProgress() && i < 10) {
                Thread.sleep(100);
                i++;
            }
            // redirect to the status Web Script
            status.setCode(Status.STATUS_MOVED_TEMPORARILY);
            status.setRedirect(true);
            status.setLocation(request.getServiceContextPath() + WEB_SCRIPT_URI_BULK_FILESYSTEM_IMPORT_STATUS);
        } else {
            model.put(IMPORT_ALREADY_IN_PROGRESS_MODEL_KEY, I18NUtil.getMessage(IMPORT_ALREADY_IN_PROGRESS_ERROR_KEY));
        }
    } catch (WebScriptException wse) {
        status.setCode(Status.STATUS_BAD_REQUEST, wse.getMessage());
        status.setRedirect(true);
    } catch (FileNotFoundException fnfe) {
        status.setCode(Status.STATUS_BAD_REQUEST, "The repository path '" + targetPath + "' does not exist !");
        status.setRedirect(true);
    } catch (IllegalArgumentException iae) {
        status.setCode(Status.STATUS_BAD_REQUEST, iae.getMessage());
        status.setRedirect(true);
    } catch (Throwable t) {
        throw new WebScriptException(Status.STATUS_INTERNAL_SERVER_ERROR, buildTextMessage(t), t);
    }
    return model;
}