fix: minor fix after upstream refactor

SAP · Jul 24, 2023 · be9e964 · be9e964
1 parent aed0b7a
commit be9e964
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/agent/src/main/java/com/orientechnologies/enterprise/server/OEnterpriseServerImpl.java b/agent/src/main/java/com/orientechnologies/enterprise/server/OEnterpriseServerImpl.java
@@ -14,6 +14,7 @@
 import com.orientechnologies.orient.core.command.OCommandRequestText;
 import com.orientechnologies.orient.core.db.*;
 import com.orientechnologies.orient.core.db.OSystemDatabase;
+import com.orientechnologies.orient.core.db.document.OQueryDatabaseState;
 import com.orientechnologies.orient.core.record.impl.ODocument;
 import com.orientechnologies.orient.core.security.OSecuritySystem;
 import com.orientechnologies.orient.core.sql.OSQLEngine;
@@ -303,7 +304,7 @@ public List<OResult> listQueries(Optional<Function<OClientConnection, Boolean>>
                             (k) -> {
                               OResultInternal internal = new OResultInternal();
                               internal.setProperty("queryId", k.getKey());
-                              OResultSet resultSet = k.getValue();
+                              OQueryDatabaseState resultSet = k.getValue();
 
                               String user = "-";
 
@@ -314,7 +315,7 @@ public List<OResult> listQueries(Optional<Function<OClientConnection, Boolean>>
                               internal.setProperty("user", user);
                               internal.setProperty("database", c.getDatabase().getName());
 
-                              Optional<QueryInfo> info = getQueryInfo(resultSet);
+                              Optional<QueryInfo> info = getQueryInfo(resultSet.getResultSet());
 
                               info.ifPresent(
                                   (it) -> {