Examples with ProcessCanceledException com.intellij.openapi.progress.ProcessCanceledException used on opensource projects

Example 91 with ProcessCanceledException

use of com.intellij.openapi.progress.ProcessCanceledException in project intellij-community by JetBrains.

the class LineMarkersPass method queryProviders.

private static void queryProviders(@NotNull List<PsiElement> elements, @NotNull PsiFile containingFile, @NotNull List<LineMarkerProvider> providers, @NotNull PairConsumer<PsiElement, LineMarkerInfo> consumer) {
    ApplicationManager.getApplication().assertReadAccessAllowed();
    Set<PsiFile> visitedInjectedFiles = new THashSet<>();
    //noinspection ForLoopReplaceableByForEach
    for (int i = 0; i < elements.size(); i++) {
        PsiElement element = elements.get(i);
        //noinspection ForLoopReplaceableByForEach
        for (int j = 0; j < providers.size(); j++) {
            ProgressManager.checkCanceled();
            LineMarkerProvider provider = providers.get(j);
            LineMarkerInfo info;
            try {
                info = provider.getLineMarkerInfo(element);
            } catch (ProcessCanceledException | IndexNotReadyException e) {
                throw e;
            } catch (Exception e) {
                LOG.error(e);
                continue;
            }
            if (info != null) {
                consumer.consume(element, info);
            }
        }
        queryLineMarkersForInjected(element, containingFile, visitedInjectedFiles, consumer);
    }
    List<LineMarkerInfo> slowLineMarkers = new ArrayList<>();
    //noinspection ForLoopReplaceableByForEach
    for (int j = 0; j < providers.size(); j++) {
        ProgressManager.checkCanceled();
        LineMarkerProvider provider = providers.get(j);
        try {
            provider.collectSlowLineMarkers(elements, slowLineMarkers);
        } catch (ProcessCanceledException | IndexNotReadyException e) {
            throw e;
        } catch (Exception e) {
            LOG.error(e);
            continue;
        }
        if (!slowLineMarkers.isEmpty()) {
            //noinspection ForLoopReplaceableByForEach
            for (int k = 0; k < slowLineMarkers.size(); k++) {
                LineMarkerInfo slowInfo = slowLineMarkers.get(k);
                PsiElement element = slowInfo.getElement();
                consumer.consume(element, slowInfo);
            }
            slowLineMarkers.clear();
        }
    }
}

Example 92 with ProcessCanceledException

use of com.intellij.openapi.progress.ProcessCanceledException in project intellij-community by JetBrains.

the class FindDialog method findSettingsChanged.

private void findSettingsChanged() {
    if (haveResultsPreview()) {
        final ModalityState state = ModalityState.current();
        // skip initial changes
        if (state == ModalityState.NON_MODAL)
            return;
        finishPreviousPreviewSearch();
        mySearchRescheduleOnCancellationsAlarm.cancelAllRequests();
        final FindModel findModel = myHelper.getModel().clone();
        applyTo(findModel, false);
        ValidationInfo result = getValidationInfo(findModel);
        final ProgressIndicatorBase progressIndicatorWhenSearchStarted = new ProgressIndicatorBase();
        myResultsPreviewSearchProgress = progressIndicatorWhenSearchStarted;
        final DefaultTableModel model = new DefaultTableModel() {

            @Override
            public boolean isCellEditable(int row, int column) {
                return false;
            }
        };
        // Use previously shown usage files as hint for faster search and better usage preview performance if pattern length increased
        final LinkedHashSet<VirtualFile> filesToScanInitially = new LinkedHashSet<>();
        if (myHelper.myPreviousModel != null && myHelper.myPreviousModel.getStringToFind().length() < findModel.getStringToFind().length()) {
            final DefaultTableModel previousModel = (DefaultTableModel) myResultsPreviewTable.getModel();
            for (int i = 0, len = previousModel.getRowCount(); i < len; ++i) {
                final UsageInfo2UsageAdapter usage = (UsageInfo2UsageAdapter) previousModel.getValueAt(i, 0);
                final VirtualFile file = usage.getFile();
                if (file != null)
                    filesToScanInitially.add(file);
            }
        }
        myHelper.myPreviousModel = findModel;
        model.addColumn("Usages");
        myResultsPreviewTable.setModel(model);
        if (result != null) {
            myResultsPreviewTable.getEmptyText().setText(UIBundle.message("message.nothingToShow"));
            myContent.setTitleAt(RESULTS_PREVIEW_TAB_INDEX, PREVIEW_TITLE);
            return;
        }
        myResultsPreviewTable.getColumnModel().getColumn(0).setCellRenderer(new UsageTableCellRenderer(false, true));
        myResultsPreviewTable.getEmptyText().setText("Searching...");
        myContent.setTitleAt(RESULTS_PREVIEW_TAB_INDEX, PREVIEW_TITLE);
        final Component component = myInputComboBox.getEditor().getEditorComponent();
        // (UsagePreviewPanel.highlight)
        if (component instanceof EditorTextField) {
            final Document document = ((EditorTextField) component).getDocument();
            if (document != null) {
                PsiDocumentManager.getInstance(myProject).commitDocument(document);
            }
        }
        final AtomicInteger resultsCount = new AtomicInteger();
        final AtomicInteger resultsFilesCount = new AtomicInteger();
        ProgressIndicatorUtils.scheduleWithWriteActionPriority(myResultsPreviewSearchProgress, new ReadTask() {

            @Nullable
            @Override
            public Continuation performInReadAction(@NotNull ProgressIndicator indicator) throws ProcessCanceledException {
                final UsageViewPresentation presentation = FindInProjectUtil.setupViewPresentation(FindSettings.getInstance().isShowResultsInSeparateView(), findModel);
                final boolean showPanelIfOnlyOneUsage = !FindSettings.getInstance().isSkipResultsWithOneUsage();
                final FindUsagesProcessPresentation processPresentation = FindInProjectUtil.setupProcessPresentation(myProject, showPanelIfOnlyOneUsage, presentation);
                Ref<VirtualFile> lastUsageFileRef = new Ref<>();
                FindInProjectUtil.findUsages(findModel, myProject, info -> {
                    final Usage usage = UsageInfo2UsageAdapter.CONVERTER.fun(info);
                    usage.getPresentation().getIcon();
                    VirtualFile file = lastUsageFileRef.get();
                    VirtualFile usageFile = info.getVirtualFile();
                    if (file == null || !file.equals(usageFile)) {
                        resultsFilesCount.incrementAndGet();
                        lastUsageFileRef.set(usageFile);
                    }
                    ApplicationManager.getApplication().invokeLater(() -> {
                        model.addRow(new Object[] { usage });
                    }, state);
                    return resultsCount.incrementAndGet() < ShowUsagesAction.USAGES_PAGE_SIZE;
                }, processPresentation, filesToScanInitially);
                boolean succeeded = !progressIndicatorWhenSearchStarted.isCanceled();
                if (succeeded) {
                    return new Continuation(() -> {
                        if (progressIndicatorWhenSearchStarted == myResultsPreviewSearchProgress && !myResultsPreviewSearchProgress.isCanceled()) {
                            int occurrences = resultsCount.get();
                            int filesWithOccurrences = resultsFilesCount.get();
                            if (occurrences == 0)
                                myResultsPreviewTable.getEmptyText().setText(UIBundle.message("message.nothingToShow"));
                            boolean foundAllUsages = occurrences < ShowUsagesAction.USAGES_PAGE_SIZE;
                            myContent.setTitleAt(RESULTS_PREVIEW_TAB_INDEX, PREVIEW_TITLE + " (" + (foundAllUsages ? Integer.valueOf(occurrences) : occurrences + "+") + UIBundle.message("message.matches", occurrences) + " in " + (foundAllUsages ? Integer.valueOf(filesWithOccurrences) : filesWithOccurrences + "+") + UIBundle.message("message.files", filesWithOccurrences) + ")");
                        }
                    }, state);
                }
                return null;
            }

            @Override
            public void onCanceled(@NotNull ProgressIndicator indicator) {
                if (isShowing() && progressIndicatorWhenSearchStarted == myResultsPreviewSearchProgress) {
                    scheduleResultsUpdate();
                }
            }
        });
    }
}

Example 93 with ProcessCanceledException

use of com.intellij.openapi.progress.ProcessCanceledException in project intellij-community by JetBrains.

the class FindUsagesManager method createUsageSearcher.

/**
   * @throws PsiInvalidElementAccessException when the searcher can't be created (i.e. because element was invalidated)
   */
@NotNull
private static UsageSearcher createUsageSearcher(@NotNull final PsiElement2UsageTargetAdapter[] primaryTargets, @NotNull final PsiElement2UsageTargetAdapter[] secondaryTargets, @NotNull final FindUsagesHandler handler, @NotNull FindUsagesOptions options, final PsiFile scopeFile) throws PsiInvalidElementAccessException {
    ReadAction.run(() -> {
        PsiElement[] primaryElements = PsiElement2UsageTargetAdapter.convertToPsiElements(primaryTargets);
        PsiElement[] secondaryElements = PsiElement2UsageTargetAdapter.convertToPsiElements(secondaryTargets);
        ContainerUtil.concat(primaryElements, secondaryElements).forEach(psi -> {
            if (psi == null || !psi.isValid())
                throw new PsiInvalidElementAccessException(psi);
        });
    });
    FindUsagesOptions optionsClone = options.clone();
    return processor -> {
        PsiElement[] primaryElements = ReadAction.compute(() -> PsiElement2UsageTargetAdapter.convertToPsiElements(primaryTargets));
        PsiElement[] secondaryElements = ReadAction.compute(() -> PsiElement2UsageTargetAdapter.convertToPsiElements(secondaryTargets));
        Project project = ReadAction.compute(() -> scopeFile != null ? scopeFile.getProject() : primaryElements[0].getProject());
        dropResolveCacheRegularly(ProgressManager.getInstance().getProgressIndicator(), project);
        if (scopeFile != null) {
            optionsClone.searchScope = new LocalSearchScope(scopeFile);
        }
        final Processor<UsageInfo> usageInfoProcessor = new CommonProcessors.UniqueProcessor<>(usageInfo -> {
            Usage usage = ReadAction.compute(() -> UsageInfoToUsageConverter.convert(primaryElements, usageInfo));
            return processor.process(usage);
        });
        final Iterable<PsiElement> elements = ContainerUtil.concat(primaryElements, secondaryElements);
        optionsClone.fastTrack = new SearchRequestCollector(new SearchSession());
        if (optionsClone.searchScope instanceof GlobalSearchScope) {
            optionsClone.searchScope = optionsClone.searchScope.union(GlobalSearchScope.projectScope(project));
        }
        try {
            for (final PsiElement element : elements) {
                handler.processElementUsages(element, usageInfoProcessor, optionsClone);
                for (CustomUsageSearcher searcher : Extensions.getExtensions(CustomUsageSearcher.EP_NAME)) {
                    try {
                        searcher.processElementUsages(element, processor, optionsClone);
                    } catch (IndexNotReadyException e) {
                        DumbService.getInstance(element.getProject()).showDumbModeNotification("Find usages is not available during indexing");
                    } catch (ProcessCanceledException e) {
                        throw e;
                    } catch (Exception e) {
                        LOG.error(e);
                    }
                }
            }
            PsiSearchHelper.SERVICE.getInstance(project).processRequests(optionsClone.fastTrack, ref -> {
                UsageInfo info = ReadAction.compute(() -> {
                    if (!ref.getElement().isValid())
                        return null;
                    return new UsageInfo(ref);
                });
                return info == null || usageInfoProcessor.process(info);
            });
        } finally {
            optionsClone.fastTrack = null;
        }
    };
}

Example 94 with ProcessCanceledException

use of com.intellij.openapi.progress.ProcessCanceledException in project intellij-community by JetBrains.

the class FindInProjectUtil method processUsagesInFile.

// returns number of hits
static int processUsagesInFile(@NotNull final PsiFile psiFile, @NotNull final VirtualFile virtualFile, @NotNull final FindModel findModel, @NotNull final Processor<UsageInfo> consumer) {
    if (findModel.getStringToFind().isEmpty()) {
        if (!ReadAction.compute(() -> consumer.process(new UsageInfo(psiFile)))) {
            throw new ProcessCanceledException();
        }
        return 1;
    }
    // do not decompile .class files
    if (virtualFile.getFileType().isBinary())
        return 0;
    final Document document = ReadAction.compute(() -> virtualFile.isValid() ? FileDocumentManager.getInstance().getDocument(virtualFile) : null);
    if (document == null)
        return 0;
    final int[] offset = { 0 };
    int count = 0;
    int found;
    ProgressIndicator indicator = ProgressWrapper.unwrap(ProgressManager.getInstance().getProgressIndicator());
    TooManyUsagesStatus tooManyUsagesStatus = TooManyUsagesStatus.getFrom(indicator);
    do {
        // wait for user out of read action
        tooManyUsagesStatus.pauseProcessingIfTooManyUsages();
        found = ReadAction.compute(() -> {
            if (!psiFile.isValid())
                return 0;
            return addToUsages(document, consumer, findModel, psiFile, offset, USAGES_PER_READ_ACTION);
        });
        count += found;
    } while (found != 0);
    return count;
}

Example 95 with ProcessCanceledException

use of com.intellij.openapi.progress.ProcessCanceledException in project intellij-community by JetBrains.

the class RefResolveServiceImpl method processFile.

// returns list of resolved files if updated successfully, or null if write action or dumb mode started
private int[] processFile(@NotNull final VirtualFile file, int fileId, @NotNull final ProgressIndicator indicator) {
    final TIntHashSet forward;
    try {
        forward = calcForwardRefs(file, indicator);
    } catch (IndexNotReadyException | ApplicationUtil.CannotRunReadActionException e) {
        return null;
    } catch (ProcessCanceledException e) {
        throw e;
    } catch (Exception e) {
        log(ExceptionUtil.getThrowableText(e));
        flushLog();
        return null;
    }
    int[] forwardIds = forward.toArray();
    fileIsResolved.set(fileId);
    logf("  ---- " + file.getPresentableUrl() + " processed. forwardIds: " + toVfString(forwardIds));
    for (Listener listener : myListeners) {
        listener.fileResolved(file);
    }
    return forwardIds;
}