add ic debug info (#76)

vesoft-inc · Dec 9, 2022 · f0d98b4 · f0d98b4
1 parent dad7c36
commit f0d98b4
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 9 deletions.
diff --git a/...g/ldbcouncil/snb/impls/workloads/nebula/operationhandlers/NebulaListOperationHandler.java b/...g/ldbcouncil/snb/impls/workloads/nebula/operationhandlers/NebulaListOperationHandler.java
@@ -46,6 +46,14 @@ public void executeOperation(TOperation operation, NebulaDbConnectionState state
 
         } catch (Exception e) {
             throw new DbException(e);
+        } finally {
+            long threadID = Thread.currentThread().getId();
+            long consumeTime = (System.currentTimeMillis() - startTime);
+            String name = operation.getClass().getSimpleName();
+            if (name.startsWith("LdbcQuery")) {
+                // only output IC query
+                System.out.println("Query SimpleName : " + name + " threadID : " + threadID + " startTime : " + startTime / 1000 + " consumeTime : " + consumeTime + "ms");
+            }
         }
     }
 

diff --git a/...il/snb/impls/workloads/nebula/operationhandlers/NebulaMultipleUpdateOperationHandler.java b/...il/snb/impls/workloads/nebula/operationhandlers/NebulaMultipleUpdateOperationHandler.java
@@ -27,6 +27,10 @@ public void executeOperation(TOperation operation, NebulaDbConnectionState state
             }
         } catch (Exception e) {
             throw new DbException(e);
+        } finally {
+            long threadID = Thread.currentThread().getId();
+            long consumeTime = (System.currentTimeMillis() - startTime) / 1000;
+            // System.out.println("Query SimpleName : " + operation.getClass().getSimpleName() + " threadID : " + threadID + " startTime : " + startTime / 1000 + " consumeTime : " + consumeTime);
         }
         resultReporter.report(0, LdbcNoResult.INSTANCE, operation);
     }

diff --git a/...council/snb/impls/workloads/nebula/operationhandlers/NebulaSingletonOperationHandler.java b/...council/snb/impls/workloads/nebula/operationhandlers/NebulaSingletonOperationHandler.java
@@ -44,6 +44,14 @@ public void executeOperation(TOperation operation, NebulaDbConnectionState state
             resultReporter.report(resultCount, tuple, operation);
         } catch (Exception e) {
             throw new DbException(e);
+        } finally {
+            long threadID = Thread.currentThread().getId();
+            long consumeTime = (System.currentTimeMillis() - startTime);
+            String name = operation.getClass().getSimpleName();
+            if (name.startsWith("LdbcQuery")) {
+                // only output IC query
+                System.out.println("Query SimpleName : " + name + " threadID : " + threadID + " startTime : " + startTime / 1000 + " consumeTime : " + consumeTime + "ms");
+            }
         }
     }
 

diff --git a/...ldbcouncil/snb/impls/workloads/nebula/operationhandlers/NebulaUpdateOperationHandler.java b/...ldbcouncil/snb/impls/workloads/nebula/operationhandlers/NebulaUpdateOperationHandler.java
@@ -14,16 +14,21 @@ public abstract class NebulaUpdateOperationHandler <TOperation extends Operation
     @Override
     public void executeOperation(TOperation operation, NebulaDbConnectionState state, ResultReporter resultReporter) throws DbException {
         Session session = state.getSession();
-            try {
-                final String queryString = getQueryString(state, operation);
-                state.logQuery(operation.getClass().getSimpleName(), queryString);
-                ResultSet result = session.execute(queryString);
-                if (state.isPrintErrors() && !result.isSucceeded()) {
-                    System.out.println(result.getErrorMessage());
-                }
-            } catch (Exception e) {
-                throw new DbException(e);
+        long startTime = System.currentTimeMillis();
+        try {
+            final String queryString = getQueryString(state, operation);
+            state.logQuery(operation.getClass().getSimpleName(), queryString);
+            ResultSet result = session.execute(queryString);
+            if (state.isPrintErrors() && !result.isSucceeded()) {
+                System.out.println(result.getErrorMessage());
             }
+        } catch (Exception e) {
+            throw new DbException(e);
+        } finally {
+            long threadID = Thread.currentThread().getId();
+            long consumeTime = (System.currentTimeMillis() - startTime) / 1000;
+            // System.out.println("Query SimpleName : " + operation.getClass().getSimpleName() + " threadID : " + threadID + " startTime : " + startTime / 1000 + " consumeTime : " + consumeTime);
+        }
             resultReporter.report(0, LdbcNoResult.INSTANCE, operation);
     }
 }