fix: make sure that toJson of list of ids get expanded, issue #10264

orientechnologies · Jul 15, 2024 · 057b0c3 · 057b0c3
1 parent 4ec7466
commit 057b0c3
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 6 deletions.
diff --git a/.../src/main/java/com/orientechnologies/orient/core/sql/functions/text/OSQLMethodToJSON.java b/.../src/main/java/com/orientechnologies/orient/core/sql/functions/text/OSQLMethodToJSON.java
@@ -19,6 +19,7 @@
 import com.orientechnologies.common.collection.OMultiValue;
 import com.orientechnologies.orient.core.command.OCommandContext;
 import com.orientechnologies.orient.core.db.record.OIdentifiable;
+import com.orientechnologies.orient.core.id.ORecordId;
 import com.orientechnologies.orient.core.record.ORecord;
 import com.orientechnologies.orient.core.record.impl.ODocument;
 import com.orientechnologies.orient.core.sql.executor.OResult;
@@ -56,19 +57,27 @@ public Object execute(
     }
 
     final String format = iParams.length > 0 ? ((String) iParams[0]).replace("\"", "") : null;
-
-    if (iThis instanceof OResult) {
-      iThis = ((OResult) iThis).toElement();
+    if (iThis instanceof ORecordId) {
+      ORecord rec = iContext.getDatabase().load((ORecordId) iThis);
+      if (rec != null) {
+        return rec.toJSON(format);
+      }
     }
-    if (iThis instanceof ORecord) {
+    if (iThis instanceof OResult) {
+      if (format != null) {
+        return ((OResult) iThis).toJSON();
+      } else {
+        ((OResult) iThis).toElement().toJSON(format);
+      }
+    } else if (iThis instanceof ORecord) {
 
       final ORecord record = (ORecord) iThis;
-      return iParams.length == 1 ? record.toJSON(format) : record.toJSON();
+      return record.toJSON(format);
 
     } else if (iThis instanceof Map) {
 
       final ODocument doc = new ODocument().fromMap((Map<String, Object>) iThis);
-      return iParams.length == 1 ? doc.toJSON(format) : doc.toJSON();
+      return doc.toJSON(format);
 
     } else if (OMultiValue.isMultiValue(iThis)) {
 

diff --git a/...st/java/com/orientechnologies/orient/core/sql/executor/OSelectStatementExecutionTest.java b/...st/java/com/orientechnologies/orient/core/sql/executor/OSelectStatementExecutionTest.java
@@ -4583,6 +4583,40 @@ public void testMapToJson() {
     }
   }
 
+  @Test
+  public void testOutJson() {
+    db.command("create class SimpleV extends V ").close();
+    db.command("create vertex SimpleV set name = 'foo'").close();
+    db.command("create vertex SimpleV set name = 'foo1'").close();
+    db.command(
+            "create edge e from (select from SimpleV where name='foo') to (select from SimpleV where name='foo1')")
+        .close();
+    try (OResultSet rs =
+        db.query(
+            "select name, outs.tojson() as json from ( select name, out() as outs from SimpleV where name='foo') ")) {
+      Assert.assertTrue(rs.hasNext());
+      OResult item = rs.next();
+      Assert.assertFalse(((String) item.getProperty("json")).contains("null"));
+    }
+  }
+
+  @Test
+  public void testOutJsonEmb() {
+    db.command("create class SimpleV extends V ").close();
+    db.command("create vertex SimpleV set name = 'foo'").close();
+    db.command("create vertex SimpleV set name = 'foo1'").close();
+    db.command(
+            "create edge e from (select from SimpleV where name='foo') to (select from SimpleV where name='foo1')")
+        .close();
+    try (OResultSet rs =
+        db.query(
+            "select name, outs as json from ( select name, out().toJson() as outs from SimpleV where name='foo') ")) {
+      Assert.assertTrue(rs.hasNext());
+      OResult item = rs.next();
+      Assert.assertFalse(((String) item.getProperty("json")).contains("null"));
+    }
+  }
+
   @Test
   public void testOptimizedCountQuery() {
     String className = "testOptimizedCountQuery";