Dynamically created tabs are shown on explain and profile queries

Metaur · Metaur · commit d28b92dd5dfd · 2017-01-17T15:25:02.000+02:00
diff --git a/database/api/src/main/java/com/neueda/jetbrains/plugin/graphdb/database/api/query/GraphQueryPlan.java b/database/api/src/main/java/com/neueda/jetbrains/plugin/graphdb/database/api/query/GraphQueryPlan.java
@@ -0,0 +1,16 @@
+package com.neueda.jetbrains.plugin.graphdb.database.api.query;
+
+import java.util.List;
+import java.util.Map;
+
+public interface GraphQueryPlan {
+
+    String getOperatorType();
+
+    Map<String, Object> getArguments();
+
+    List<String> getIdentifiers();
+
+    List<? extends GraphQueryPlan> children();
+
+}
diff --git a/database/api/src/main/java/com/neueda/jetbrains/plugin/graphdb/database/api/query/GraphQueryResult.java b/database/api/src/main/java/com/neueda/jetbrains/plugin/graphdb/database/api/query/GraphQueryResult.java
@@ -20,4 +20,10 @@ public interface GraphQueryResult {
     List<GraphRelationship> getRelationships();
 
     List<GraphQueryNotification> getNotifications();
+
+    boolean hasPlan();
+
+    boolean hasProfile();
+
+    GraphQueryPlan getPlan();
 }
diff --git a/database/neo4j/src/main/java/com/neueda/jetbrains/plugin/graphdb/database/neo4j/bolt/Neo4jBoltBuffer.java b/database/neo4j/src/main/java/com/neueda/jetbrains/plugin/graphdb/database/neo4j/bolt/Neo4jBoltBuffer.java
@@ -3,12 +3,15 @@
 import com.neueda.jetbrains.plugin.graphdb.database.api.data.GraphNode;
 import com.neueda.jetbrains.plugin.graphdb.database.api.data.GraphRelationship;
 import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryNotification;
+import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryPlan;
 import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryResultColumn;
 import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryResultRow;
 import com.neueda.jetbrains.plugin.graphdb.database.neo4j.bolt.data.Neo4jBoltQueryNotification;
+import com.neueda.jetbrains.plugin.graphdb.database.neo4j.bolt.data.Neo4jBoltQueryPlan;
 import com.neueda.jetbrains.plugin.graphdb.database.neo4j.bolt.data.Neo4jBoltQueryResultColumn;
 import com.neueda.jetbrains.plugin.graphdb.database.neo4j.bolt.data.Neo4jBoltQueryResultRow;
 import org.neo4j.driver.v1.summary.InputPosition;
+import org.neo4j.driver.v1.summary.Plan;
 import org.neo4j.driver.v1.summary.ResultSummary;
 
 import java.util.ArrayList;
@@ -18,6 +21,7 @@
 import java.util.Optional;
 
 import static java.util.stream.Collectors.toList;
+import static java.util.stream.Collectors.toMap;
 
 public class Neo4jBoltBuffer {
 
@@ -103,4 +107,39 @@ public List<GraphQueryNotification> getNotifications() {
 
         return notifications;
     }
+
+    public boolean hasPlan() {
+        return Optional.ofNullable(resultSummary)
+                .map(ResultSummary::hasPlan)
+                .orElse(false);
+    }
+
+    public boolean hasProfile() {
+        return Optional.ofNullable(resultSummary)
+                .map(ResultSummary::hasProfile)
+                .orElse(false);
+    }
+
+    public GraphQueryPlan getQueryPlan() {
+        if (!hasPlan()) {
+            return null;
+        }
+
+        Plan plan = resultSummary.plan();
+        return new Neo4jBoltQueryPlan(plan.operatorType(), getArguments(plan), plan.identifiers(),
+                getPlanChildren(plan.children()));
+    }
+
+    private static List<Neo4jBoltQueryPlan> getPlanChildren(List<? extends Plan> childrenPlans) {
+        return childrenPlans.stream()
+                .map(plan -> new Neo4jBoltQueryPlan(plan.operatorType(), getArguments(plan), plan.identifiers(),
+                        getPlanChildren(plan.children())))
+                .collect(toList());
+    }
+
+    private static Map<String, Object> getArguments(Plan plan) {
+        return plan.arguments().entrySet().stream()
+                .collect(toMap(Map.Entry::getKey, e -> e.getValue().asObject()));
+    }
+
 }
diff --git a/database/neo4j/src/main/java/com/neueda/jetbrains/plugin/graphdb/database/neo4j/bolt/data/Neo4jBoltQueryPlan.java b/database/neo4j/src/main/java/com/neueda/jetbrains/plugin/graphdb/database/neo4j/bolt/data/Neo4jBoltQueryPlan.java
@@ -0,0 +1,42 @@
+package com.neueda.jetbrains.plugin.graphdb.database.neo4j.bolt.data;
+
+import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryPlan;
+
+import java.util.List;
+import java.util.Map;
+
+public class Neo4jBoltQueryPlan implements GraphQueryPlan {
+
+    private String operatorType;
+    private Map<String, Object> arguments;
+    private List<String> identifiers;
+    private List<Neo4jBoltQueryPlan> children;
+
+    public Neo4jBoltQueryPlan(String operatorType, Map<String, Object> arguments, List<String> identifiers,
+                              List<Neo4jBoltQueryPlan> children) {
+        this.operatorType = operatorType;
+        this.arguments = arguments;
+        this.identifiers = identifiers;
+        this.children = children;
+    }
+
+    @Override
+    public String getOperatorType() {
+        return operatorType;
+    }
+
+    @Override
+    public Map<String, Object> getArguments() {
+        return arguments;
+    }
+
+    @Override
+    public List<String> getIdentifiers() {
+        return identifiers;
+    }
+
+    @Override
+    public List<Neo4jBoltQueryPlan> children() {
+        return children;
+    }
+}
diff --git a/database/neo4j/src/main/java/com/neueda/jetbrains/plugin/graphdb/database/neo4j/bolt/query/Neo4jBoltQueryResult.java b/database/neo4j/src/main/java/com/neueda/jetbrains/plugin/graphdb/database/neo4j/bolt/query/Neo4jBoltQueryResult.java
@@ -3,6 +3,7 @@
 import com.neueda.jetbrains.plugin.graphdb.database.api.data.GraphNode;
 import com.neueda.jetbrains.plugin.graphdb.database.api.data.GraphRelationship;
 import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryNotification;
+import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryPlan;
 import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryResult;
 import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryResultColumn;
 import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryResultRow;
@@ -149,6 +150,21 @@ public List<GraphQueryNotification> getNotifications() {
         return buffer.getNotifications();
     }
 
+    @Override
+    public boolean hasPlan() {
+        return buffer.hasPlan();
+    }
+
+    @Override
+    public boolean hasProfile() {
+        return buffer.hasProfile();
+    }
+
+    @Override
+    public GraphQueryPlan getPlan() {
+        return buffer.getQueryPlan();
+    }
+
     private Optional<GraphNode> findNodeById(List<GraphNode> nodes, String id) {
         return nodes.stream().filter((node) -> node.getId().equals(id)).findFirst();
     }
diff --git a/ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/database/QueryExecutionService.java b/ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/database/QueryExecutionService.java
@@ -9,6 +9,7 @@
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.component.analytics.Analytics;
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.component.datasource.state.DataSourceApi;
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.console.event.QueryExecutionProcessEvent;
+import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.console.event.QueryPlanEvent;
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.util.Notifier;
 
 import java.util.concurrent.Future;
@@ -61,6 +62,11 @@ private synchronized void executeInBackground(DataSourceApi dataSource, ExecuteQ
                     event.resultReceived(payload, result);
                     event.postResultReceived(payload);
                     event.executionCompleted(payload);
+
+                    if (result.hasPlan()) {
+                        QueryPlanEvent queryPlanEvent = messageBus.syncPublisher(QueryPlanEvent.QUERY_PLAN_EVENT);
+                        queryPlanEvent.queryPlanReceived(payload.getContent(), result);
+                    }
                 });
             } catch (Exception e) {
                 ApplicationManager.getApplication().invokeLater(() -> {
diff --git a/ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/console/GraphConsoleView.java b/ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/console/GraphConsoleView.java
@@ -1,9 +1,13 @@
 package com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.console;
 
+import com.intellij.ide.IdeEventQueue;
 import com.intellij.openapi.Disposable;
 import com.intellij.openapi.actionSystem.ActionGroup;
 import com.intellij.openapi.actionSystem.ActionManager;
+import com.intellij.openapi.actionSystem.ActionPlaces;
 import com.intellij.openapi.actionSystem.ActionToolbar;
+import com.intellij.openapi.actionSystem.AnActionEvent;
+import com.intellij.openapi.actionSystem.DefaultActionGroup;
 import com.intellij.openapi.components.ServiceManager;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.util.ActionCallback;
@@ -22,17 +26,29 @@
 import com.intellij.ui.tabs.TabInfo;
 import com.intellij.ui.tabs.impl.JBTabsImpl;
 import com.intellij.ui.treeStructure.Tree;
+import com.intellij.util.ui.UIUtil;
+import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryResult;
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.component.analytics.Analytics;
+import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.console.event.QueryPlanEvent;
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.console.graph.GraphPanel;
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.console.log.LogPanel;
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.console.params.ParametersPanel;
+import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.console.plan.QueryPlanPanel;
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.console.status.ExecutionStatusBarWidget;
 import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.console.table.TablePanel;
 import com.neueda.jetbrains.plugin.graphdb.platform.GraphConstants;
 import com.neueda.jetbrains.plugin.graphdb.visualization.services.LookAndFeelService;
 
 import javax.swing.*;
 import java.awt.*;
+import java.awt.event.MouseEvent;
+import java.awt.event.MouseListener;
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
+import java.time.format.DateTimeFormatterBuilder;
+import java.util.concurrent.atomic.AtomicInteger;
+
+import static java.time.temporal.ChronoField.*;
 
 public class GraphConsoleView implements Disposable {
 
@@ -66,6 +82,15 @@ public class GraphConsoleView implements Disposable {
     private LogPanel logPanel;
     private ParametersPanel parametersPanel;
 
+    private static final DateTimeFormatter QUERY_PLAN_TIME_FORMAT = new DateTimeFormatterBuilder()
+            .appendValue(HOUR_OF_DAY, 2)
+            .appendLiteral(':')
+            .appendValue(MINUTE_OF_HOUR, 2)
+            .optionalStart()
+            .appendLiteral(':')
+            .appendValue(SECOND_OF_MINUTE, 2)
+            .toFormatter();
+
     public GraphConsoleView() {
         initialized = false;
 
@@ -106,6 +131,10 @@ public void initToolWindow(Project project, ToolWindow toolWindow) {
                 return callback;
             });
 
+            AtomicInteger tabId = new AtomicInteger(0);
+            project.getMessageBus().connect().subscribe(QueryPlanEvent.QUERY_PLAN_EVENT,
+                    (query, result) -> createNewQueryPlanTab(query, result, tabId.incrementAndGet()));
+
             // Actions
             final ActionGroup consoleActionGroup = (ActionGroup)
                     ActionManager.getInstance().getAction(GraphConstants.Actions.CONSOLE_ACTIONS);
@@ -123,6 +152,31 @@ private void createUIComponents() {
         graphCanvas = new JPanel(new GridLayout(0, 1));
         consoleTabsPane = new JBTabsPaneImpl(null, SwingConstants.TOP, this);
         consoleTabs = (JBTabsImpl) consoleTabsPane.getTabs();
+
+        consoleTabs.addTabMouseListener(new MouseListener() {
+            @Override
+            public void mouseReleased(MouseEvent e) {
+                if (UIUtil.isCloseClick(e, MouseEvent.MOUSE_RELEASED)) {
+                    final TabInfo info = consoleTabs.findInfo(e);
+                    if (info != null && info.getText().startsWith("Query ")) {
+                        IdeEventQueue.getInstance().blockNextEvents(e);
+                        consoleTabs.removeTab(info);
+                    }
+                }
+            }
+
+            @Override
+            public void mouseClicked(MouseEvent e) {}
+
+            @Override
+            public void mousePressed(MouseEvent e) {}
+
+            @Override
+            public void mouseEntered(MouseEvent e) {}
+
+            @Override
+            public void mouseExited(MouseEvent e) {}
+        });
     }
 
     private void updateLookAndFeel() {
@@ -146,6 +200,28 @@ private void initializeWidgets(Project project) {
         statusBar.addWidget(executionStatusBarWidget, "before Position");
     }
 
+    private void createNewQueryPlanTab(String originalQuery,
+                                             GraphQueryResult result, int tabId) {
+        JPanel panel = new JPanel();
+        panel.setLayout(new BorderLayout(0, 3));
+
+        QueryPlanPanel qpPanel = new QueryPlanPanel(originalQuery, result);
+        qpPanel.initialize(panel);
+
+        TabInfo tabInfo = new TabInfo(panel);
+        DefaultActionGroup tabActions = new DefaultActionGroup(new QueryPlanPanel.CloseTab() {
+            @Override
+            public void actionPerformed(AnActionEvent e) {
+                consoleTabs.removeTab(tabInfo);
+            }
+        });
+        tabInfo.setTabLabelActions(tabActions, ActionPlaces.EDITOR_TAB);
+
+        String planType = result.hasProfile() ? "Profile" : "Explain";
+        consoleTabs.addTab(tabInfo.setText(String.format("%1s %2d - %3s", planType, tabId,
+                LocalDateTime.now().format(QUERY_PLAN_TIME_FORMAT))));
+    }
+
     public TablePanel getTablePanel() {
         return tablePanel;
     }
diff --git a/ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/console/event/QueryPlanEvent.java b/ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/console/event/QueryPlanEvent.java
@@ -0,0 +1,11 @@
+package com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.console.event;
+
+import com.intellij.util.messages.Topic;
+import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryResult;
+
+public interface QueryPlanEvent {
+
+    Topic<QueryPlanEvent> QUERY_PLAN_EVENT = Topic.create("GraphDatabaseConsole.QueryPlanEvent", QueryPlanEvent.class);
+
+    void queryPlanReceived(String query, GraphQueryResult result);
+}
diff --git a/ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/console/plan/ColumnDefinitions.java b/ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/console/plan/ColumnDefinitions.java
diff --git a/ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/console/plan/QueryPlanArgumentKeys.java b/ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/console/plan/QueryPlanArgumentKeys.java
diff --git a/ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/console/plan/QueryPlanPanel.java b/ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/console/plan/QueryPlanPanel.java
