blob: 099ea9848acec68e94e7c3373eb82d89525744e8 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package ai.vespa.intellij.schema.findUsages;
import ai.vespa.intellij.schema.model.RankProfile;
import ai.vespa.intellij.schema.model.Schema;
import ai.vespa.intellij.schema.psi.SdRankProfileDefinition;
import com.intellij.openapi.application.ReadAction;
import com.intellij.openapi.progress.ProgressIndicatorProvider;
import com.intellij.psi.search.SearchScope;
import com.intellij.usageView.UsageInfo;
import com.intellij.util.Processor;
import java.util.HashSet;
import java.util.Set;
/**
* An instance created to find usages of a rank profile once.
*
* @author bratseth
*/
public class RankProfileUsageFinder extends UsageFinder {
private final RankProfile profile;
private final Set<RankProfile> visited = new HashSet<>();
public RankProfileUsageFinder(SdRankProfileDefinition profileDefinitionToFind, SearchScope scope, Processor<? super UsageInfo> processor) {
super(scope, processor);
this.profile = ReadAction.compute(() -> resolveSchema(profileDefinitionToFind).rankProfiles().get(profileDefinitionToFind.getName()));
}
public void findUsages() {
findUsagesBelow(profile);
}
private void findUsagesBelow(RankProfile profile) {
ProgressIndicatorProvider.checkCanceled();
if ( ! visited.add(profile)) return;
if ( ! profile.equals(this.profile))
ReadAction.compute(() -> processor().process(new UsageInfo(profile.definition())));
for (var child : ReadAction.compute(() -> profile.children()))
findUsagesBelow(child);
}
}
|