migrate getData to uiDataSnapshot: java

GitOrigin-RevId: 759993dfe9da87ef7f32232c84bfb1bd122610d7
This commit is contained in:
Gregory.Shrago
2024-07-31 04:09:26 +04:00
committed by intellij-monorepo-bot
parent 21484f1e2e
commit e33abfddfb
3 changed files with 12 additions and 23 deletions

View File

@@ -488,14 +488,11 @@ public final class CyclicDependenciesPanel extends JPanel implements Disposable,
}
}
private static class MyTree extends Tree implements DataProvider {
private static class MyTree extends Tree implements UiDataProvider {
@Override
public Object getData(@NotNull String dataId) {
public void uiDataSnapshot(@NotNull DataSink sink) {
PackageDependenciesNode node = getSelectedNode();
if (CommonDataKeys.NAVIGATABLE.is(dataId)) {
return node;
}
return null;
sink.set(CommonDataKeys.NAVIGATABLE, node);
}
public @Nullable PackageDependenciesNode getSelectedNode() {

View File

@@ -68,8 +68,6 @@ import java.util.*;
import java.util.function.Consumer;
import java.util.stream.Stream;
import static com.intellij.openapi.actionSystem.PlatformCoreDataKeys.BGT_DATA_PROVIDER;
public final class PropagateAnnotationPanel extends JPanel implements Disposable {
private final Tree myTree;
@NotNull
@@ -418,25 +416,18 @@ public final class PropagateAnnotationPanel extends JPanel implements Disposable
}
}
private static final class PropagateTree extends Tree implements DataProvider {
private static final class PropagateTree extends Tree implements UiDataProvider {
private PropagateTree(TreeModel treeModel) {
super(treeModel);
getSelectionModel().setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);
}
@Override
public @Nullable Object getData(@NotNull String dataId) {
if (BGT_DATA_PROVIDER.is(dataId)) {
return (DataProvider)slowId -> getSlowData(slowId);
}
return null;
}
private @Nullable Object getSlowData(@NotNull String dataId) {
if (!CommonDataKeys.PSI_ELEMENT.is(dataId)) return null;
public void uiDataSnapshot(@NotNull DataSink sink) {
TaintNode[] selectedNodes = getSelectedNodes(TaintNode.class, null);
if (selectedNodes.length != 1) return null;
return selectedNodes[0].getRef();
sink.lazy(CommonDataKeys.PSI_ELEMENT, () -> {
return selectedNodes.length == 1 ? selectedNodes[0].getRef() : null;
});
}
@Contract("_, _ -> new")

View File

@@ -44,8 +44,7 @@ class CoroutineDumpPanel(
consoleView: ConsoleView,
toolbarActions: DefaultActionGroup,
val dump: List<CompleteCoroutineInfoData>
) :
JPanel(BorderLayout()), DataProvider {
) : JPanel(BorderLayout()), UiDataProvider {
private var exporterToTextFile: ExporterToTextFile
private var mergedDump = ArrayList<CompleteCoroutineInfoData>()
val filterField = SearchTextField()
@@ -174,7 +173,9 @@ class CoroutineDumpPanel(
}
}
override fun getData(dataId: String): Any? = if (PlatformDataKeys.EXPORTER_TO_TEXT_FILE.`is`(dataId)) exporterToTextFile else null
override fun uiDataSnapshot(sink: DataSink) {
sink[PlatformDataKeys.EXPORTER_TO_TEXT_FILE] = exporterToTextFile
}
private fun getAttributes(infoData: CompleteCoroutineInfoData): SimpleTextAttributes {
return when {