orchid-orm 1.59.3 → 1.59.4

package/dist/index.mjs

@@ -421,7 +421,10 @@ class BelongsToVirtualColumn extends VirtualColumn {
         q,
         (as) => {
           foreignKeys.forEach((foreignKey, i) => {
-            item[foreignKey]._sql = `(SELECT "${as}"."${primaryKeys[i]}" FROM "${as}")`;
+            item[foreignKey]._sql = selectCteColumnSql(
+              as,
+              primaryKeys[i]
+            );
           });
         },
         "create" in value ? _queryCreate(selectPKeys, value.create) : _orCreate(
@@ -436,9 +439,8 @@ class BelongsToVirtualColumn extends VirtualColumn {
       const as = getFreeAlias(q.q.withShapes, "q");
       _with(q, as, query.select(...primaryKeys).findBy(value.connect));
       foreignKeys.map((foreignKey, i) => {
-        const selectColumn = `(SELECT "${as}"."${primaryKeys[i]}" FROM "${as}")`;
         item[foreignKey] = new RawSql(
-          i === 0 ? `CASE WHEN (SELECT count(*) FROM "${as}") = 0 AND (SELECT 'not-found')::int = 0 THEN NULL ELSE ${selectColumn} END` : selectColumn
+          selectCteColumnMustExistSql(i, as, primaryKeys[i])
         );
       });
       return;
@@ -542,13 +544,18 @@ const nestedUpdate$3 = ({ query, primaryKeys, foreignKeys, len }) => {
           }
         }
         if (loadPrimaryKeys) {
-          const record = await _queryFindBy(
-            query.select(...loadPrimaryKeys),
-            params.set
-          );
           for (let i = 0, len2 = loadPrimaryKeys.length; i < len2; i++) {
-            update[loadForeignKeys[i]] = record[loadPrimaryKeys[i]];
+            update[loadForeignKeys[i]] = new RawSql("");
           }
+          _with(
+            q2,
+            (as) => {
+              for (let i = 0, len2 = loadPrimaryKeys.length; i < len2; i++) {
+                update[loadForeignKeys[i]]._sql = selectCteColumnMustExistSql(i, as, loadPrimaryKeys[i]);
+              }
+            },
+            _queryFindBy(query.select(...loadPrimaryKeys), params.set)
+          );
         }
       } else if (params.create) {
         const q3 = query.clone();
@@ -645,6 +652,11 @@ const nestedUpdate$3 = ({ query, primaryKeys, foreignKeys, len }) => {
     }
   };
 };
+const selectCteColumnSql = (cteAs, column) => `(SELECT "${cteAs}"."${column}" FROM "${cteAs}")`;
+const selectCteColumnMustExistSql = (i, cteAs, column) => {
+  const selectColumn = selectCteColumnSql(cteAs, column);
+  return i === 0 ? `CASE WHEN (SELECT count(*) FROM "${cteAs}") = 0 AND (SELECT 'not-found')::int = 0 THEN NULL ELSE ${selectColumn} END` : selectColumn;
+};
 
 class HasOneVirtualColumn extends VirtualColumn {
   constructor(schema, key, state) {