fix: belongsToMany relation withCount

sutandojs · May 14, 2024 · 9e4e7ce · 9e4e7ce
1 parent c232422
commit 9e4e7ce
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/relations/belongs-to-many.js b/src/relations/belongs-to-many.js
@@ -300,6 +300,27 @@ class BelongsToMany extends compose(
   //     this.touch();
   //   }
   // }
+
+  getRelationExistenceQuery(query, parentQuery, columns = ['*']) {
+    if (parentQuery.getQuery()._single.table == query.getQuery()._single.table) {
+      return this.getRelationExistenceQueryForSelfJoin(query, parentQuery, columns);
+    }
+
+    this.performJoin(query);
+
+    return super.getRelationExistenceQuery(query, parentQuery, columns);
+  }
+
+  getRelationExistenceQueryForSelfJoin(query, parentQuery, columns = ['*']) {
+    const hash = this.getRelationCountHash();
+    query.select(columns).from(this.related.getTable() + ' as ' + hash);
+
+    this.related.setTable(hash);
+
+    this.performJoin(query);
+
+    return super.getRelationExistenceQuery(query, parentQuery, columns);
+  }
 }
 
 module.exports = BelongsToMany;