mirror of
https://github.com/strapi/strapi.git
synced 2025-10-20 04:24:23 +00:00
start factorizing code
This commit is contained in:
Pierre Noël
parent
e7428cdf83
commit
5d0eb61fae
@ -21,13 +21,12 @@ const { createField } = require('../fields');
|
|||||||
const { createQueryBuilder } = require('../query');
|
const { createQueryBuilder } = require('../query');
|
||||||
const { createRepository } = require('./entity-repository');
|
const { createRepository } = require('./entity-repository');
|
||||||
const { deleteRelatedMorphOneRelationsAfterMorphToManyUpdate } = require('./morph-relations');
|
const { deleteRelatedMorphOneRelationsAfterMorphToManyUpdate } = require('./morph-relations');
|
||||||
|
const { isBidirectional, isManyToAny, isAnyToOne, isAnyToMany } = require('../metadata/relations');
|
||||||
const {
|
const {
|
||||||
isBidirectional,
|
deletePreviousOneToAnyRelations,
|
||||||
isOneToAny,
|
deletePreviousAnyToOneRelations,
|
||||||
isManyToAny,
|
deleteAllRelations,
|
||||||
isAnyToOne,
|
} = require('./utils');
|
||||||
isAnyToMany,
|
|
||||||
} = require('../metadata/relations');
|
|
||||||
|
|
||||||
const toId = (value) => value.id || value;
|
const toId = (value) => value.id || value;
|
||||||
const toIds = (value) => castArray(value || []).map(toId);
|
const toIds = (value) => castArray(value || []).map(toId);
|
||||||
@ -493,42 +492,7 @@ const createEntityManager = (db) => {
|
|||||||
const relsToAdd = uniqBy('id', cleanRelationData.connect || cleanRelationData);
|
const relsToAdd = uniqBy('id', cleanRelationData.connect || cleanRelationData);
|
||||||
const relIdsToadd = toIds(relsToAdd);
|
const relIdsToadd = toIds(relsToAdd);
|
||||||
|
|
||||||
// need to delete the previous relations for oneToAny relations
|
await deletePreviousOneToAnyRelations({ id, attribute, joinTable, relIdsToadd, db });
|
||||||
if (isBidirectional(attribute) && isOneToAny(attribute)) {
|
|
||||||
// update orders for previous oneToAny relations that will be deleted if it has order (oneToMany)
|
|
||||||
if (isAnyToMany(attribute)) {
|
|
||||||
const currentRelsToDelete = await this.createQueryBuilder(joinTable.name)
|
|
||||||
.select(select)
|
|
||||||
.where({
|
|
||||||
[inverseJoinColumn.name]: relIdsToadd,
|
|
||||||
[joinColumn.name]: { $ne: id },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
|
|
||||||
currentRelsToDelete.sort((a, b) => b[orderColumnName] - a[orderColumnName]);
|
|
||||||
|
|
||||||
for (const relToDelete of currentRelsToDelete) {
|
|
||||||
if (relToDelete[orderColumnName] !== null) {
|
|
||||||
await this.createQueryBuilder(joinTable.name)
|
|
||||||
.decrement(orderColumnName, 1)
|
|
||||||
.where({
|
|
||||||
[joinColumn.name]: relToDelete[joinColumn.name],
|
|
||||||
[orderColumnName]: { $gt: relToDelete[orderColumnName] },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// delete previous oneToAny relations
|
|
||||||
await this.createQueryBuilder(joinTable.name)
|
|
||||||
.delete()
|
|
||||||
.where({ [inverseJoinColumn.name]: relIdsToadd })
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
}
|
|
||||||
|
|
||||||
// prepare new relations to insert
|
// prepare new relations to insert
|
||||||
const insert = relsToAdd.map((data) => {
|
const insert = relsToAdd.map((data) => {
|
||||||
@ -738,50 +702,7 @@ const createEntityManager = (db) => {
|
|||||||
|
|
||||||
// only delete relations
|
// only delete relations
|
||||||
if (isNull(data[attributeName])) {
|
if (isNull(data[attributeName])) {
|
||||||
// INVERSE ORDER UPDATE
|
await deleteAllRelations({ id, attribute, joinTable, db });
|
||||||
if (isBidirectional(attribute) && isManyToAny(attribute)) {
|
|
||||||
let lastId = 0;
|
|
||||||
let done = false;
|
|
||||||
const batchSize = 100;
|
|
||||||
while (!done) {
|
|
||||||
const relsToDelete = await this.createQueryBuilder(joinTable.name)
|
|
||||||
.select(select)
|
|
||||||
.where({
|
|
||||||
[joinColumn.name]: id,
|
|
||||||
id: { $gt: lastId },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.orderBy('id')
|
|
||||||
.limit(batchSize)
|
|
||||||
.execute();
|
|
||||||
// TODO: cannot put pivot here...
|
|
||||||
done = relsToDelete.length < batchSize;
|
|
||||||
lastId = relsToDelete[relsToDelete.length - 1]?.id;
|
|
||||||
|
|
||||||
const updateInverseOrderPromises = [];
|
|
||||||
for (const relToDelete of relsToDelete) {
|
|
||||||
if (relToDelete[inverseOrderColumnName] !== null) {
|
|
||||||
const updatePromise = this.createQueryBuilder(joinTable.name)
|
|
||||||
.decrement(inverseOrderColumnName, 1)
|
|
||||||
.where({
|
|
||||||
[inverseJoinColumn.name]: relToDelete[inverseJoinColumn.name],
|
|
||||||
[inverseOrderColumnName]: { $gt: relToDelete[inverseOrderColumnName] },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
updateInverseOrderPromises.push(updatePromise);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
await Promise.all(updateInverseOrderPromises);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
await this.createQueryBuilder(joinTable.name)
|
|
||||||
.delete()
|
|
||||||
.where({ [joinColumn.name]: id })
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
} else {
|
} else {
|
||||||
const cleanRelationData = toAssocs(data[attributeName]);
|
const cleanRelationData = toAssocs(data[attributeName]);
|
||||||
const isPartialUpdate =
|
const isPartialUpdate =
|
||||||
@ -800,70 +721,7 @@ const createEntityManager = (db) => {
|
|||||||
// DELETE relations in disconnect
|
// DELETE relations in disconnect
|
||||||
const relIdsToDelete = toIds(differenceBy('id', disconnect, connect));
|
const relIdsToDelete = toIds(differenceBy('id', disconnect, connect));
|
||||||
|
|
||||||
// UPDATE RELEVANT ORDERS
|
await deleteAllRelations({ id, attribute, joinTable, onlyFor: relIdsToDelete, db });
|
||||||
if (
|
|
||||||
isAnyToMany(attribute) ||
|
|
||||||
(isBidirectional(attribute) && isManyToAny(attribute))
|
|
||||||
) {
|
|
||||||
const relsToDelete = await this.createQueryBuilder(joinTable.name)
|
|
||||||
.select(select)
|
|
||||||
.where({
|
|
||||||
[joinColumn.name]: id,
|
|
||||||
[inverseJoinColumn.name]: { $in: relIdsToDelete },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
|
|
||||||
// ORDER UPDATE
|
|
||||||
if (isAnyToMany(attribute)) {
|
|
||||||
// sort by order DESC so that the order updates are done in the correct order
|
|
||||||
// avoiding one to interfere with the others
|
|
||||||
relsToDelete.sort((a, b) => b[orderColumnName] - a[orderColumnName]);
|
|
||||||
|
|
||||||
for (const relToDelete of relsToDelete) {
|
|
||||||
if (relToDelete[orderColumnName] !== null) {
|
|
||||||
await this.createQueryBuilder(joinTable.name)
|
|
||||||
.decrement(orderColumnName, 1)
|
|
||||||
.where({
|
|
||||||
[joinColumn.name]: id,
|
|
||||||
[orderColumnName]: { $gt: relToDelete[orderColumnName] },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// INVERSE ORDER UPDATE
|
|
||||||
if (isBidirectional(attribute) && isManyToAny(attribute)) {
|
|
||||||
const updateInverseOrderPromises = [];
|
|
||||||
for (const relToDelete of relsToDelete) {
|
|
||||||
if (relToDelete[inverseOrderColumnName] !== null) {
|
|
||||||
const updatePromise = this.createQueryBuilder(joinTable.name)
|
|
||||||
.decrement(inverseOrderColumnName, 1)
|
|
||||||
.where({
|
|
||||||
[inverseJoinColumn.name]: relToDelete[inverseJoinColumn.name],
|
|
||||||
[inverseOrderColumnName]: { $gt: relToDelete[inverseOrderColumnName] },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
updateInverseOrderPromises.push(updatePromise);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
await Promise.all(updateInverseOrderPromises);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// DELETE
|
|
||||||
await this.createQueryBuilder(joinTable.name)
|
|
||||||
.delete()
|
|
||||||
.where({
|
|
||||||
[joinColumn.name]: id,
|
|
||||||
[inverseJoinColumn.name]: { $in: relIdsToDelete },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
|
|
||||||
// add/move
|
// add/move
|
||||||
let max;
|
let max;
|
||||||
@ -950,80 +808,7 @@ const createEntityManager = (db) => {
|
|||||||
// overwrite all relations
|
// overwrite all relations
|
||||||
const relsToAdd = uniqBy('id', cleanRelationData);
|
const relsToAdd = uniqBy('id', cleanRelationData);
|
||||||
relIdsToaddOrMove = toIds(relsToAdd);
|
relIdsToaddOrMove = toIds(relsToAdd);
|
||||||
|
await deleteAllRelations({ id, attribute, joinTable, except: relIdsToaddOrMove, db });
|
||||||
// UPDATE RELEVANT ORDERS BEFORE DELETE
|
|
||||||
if (isAnyToMany(attribute) || isManyToAny(attribute)) {
|
|
||||||
let lastId = 0;
|
|
||||||
let done = false;
|
|
||||||
const batchSize = 100;
|
|
||||||
while (!done) {
|
|
||||||
const relsToDelete = await this.createQueryBuilder(joinTable.name)
|
|
||||||
.select(select)
|
|
||||||
.where({
|
|
||||||
[joinColumn.name]: id,
|
|
||||||
[inverseJoinColumn.name]: { $notIn: relIdsToaddOrMove },
|
|
||||||
id: { $gt: lastId },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.orderBy('id')
|
|
||||||
.limit(batchSize)
|
|
||||||
.execute();
|
|
||||||
|
|
||||||
done = relsToDelete.length < batchSize;
|
|
||||||
lastId = relsToDelete[relsToDelete.length - 1]?.id;
|
|
||||||
|
|
||||||
// ORDER UPDATE
|
|
||||||
if (isAnyToMany(attribute)) {
|
|
||||||
// sort by order DESC so that the order updates are done in the correct order
|
|
||||||
// avoiding one to interfere with the others
|
|
||||||
relsToDelete.sort((a, b) => b[orderColumnName] - a[orderColumnName]);
|
|
||||||
|
|
||||||
for (const relToDelete of relsToDelete) {
|
|
||||||
if (relToDelete[orderColumnName] !== null) {
|
|
||||||
await this.createQueryBuilder(joinTable.name)
|
|
||||||
.decrement(orderColumnName, 1)
|
|
||||||
.where({
|
|
||||||
[joinColumn.name]: id,
|
|
||||||
[orderColumnName]: { $gt: relToDelete[orderColumnName] },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
// manque le pivot ici
|
|
||||||
.execute();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// INVERSE ORDER UPDATE
|
|
||||||
if (isBidirectional(attribute) && isManyToAny(attribute)) {
|
|
||||||
const updateInverseOrderPromises = [];
|
|
||||||
for (const relToDelete of relsToDelete) {
|
|
||||||
if (relToDelete[inverseOrderColumnName] !== null) {
|
|
||||||
const updatePromise = this.createQueryBuilder(joinTable.name)
|
|
||||||
.decrement(inverseOrderColumnName, 1)
|
|
||||||
.where({
|
|
||||||
[inverseJoinColumn.name]: relToDelete[inverseJoinColumn.name],
|
|
||||||
[inverseOrderColumnName]: { $gt: relToDelete[inverseOrderColumnName] },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
updateInverseOrderPromises.push(updatePromise);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
await Promise.all(updateInverseOrderPromises);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// DELETE
|
|
||||||
await this.createQueryBuilder(joinTable.name)
|
|
||||||
.delete()
|
|
||||||
.where({
|
|
||||||
[joinColumn.name]: id,
|
|
||||||
[inverseJoinColumn.name]: { $notIn: relIdsToaddOrMove },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
|
|
||||||
const currentMovingRels = await this.createQueryBuilder(joinTable.name)
|
const currentMovingRels = await this.createQueryBuilder(joinTable.name)
|
||||||
.select(select)
|
.select(select)
|
||||||
@ -1053,6 +838,7 @@ const createEntityManager = (db) => {
|
|||||||
const insert = {
|
const insert = {
|
||||||
[joinColumn.name]: id,
|
[joinColumn.name]: id,
|
||||||
[inverseJoinColumn.name]: relToAdd.id,
|
[inverseJoinColumn.name]: relToAdd.id,
|
||||||
|
...(joinTable.on || {}),
|
||||||
...(relToAdd.__pivot || {}),
|
...(relToAdd.__pivot || {}),
|
||||||
};
|
};
|
||||||
|
|
||||||
@ -1078,82 +864,22 @@ const createEntityManager = (db) => {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Delete the previous relations for oneToAny relations
|
// Delete the previous relations for oneToAny relations
|
||||||
if (isBidirectional(attribute) && isOneToAny(attribute)) {
|
await deletePreviousOneToAnyRelations({
|
||||||
// update orders for previous oneToAny relations that will be deleted if it has order (oneToMany)
|
id,
|
||||||
if (isAnyToMany(attribute)) {
|
attribute,
|
||||||
const currentRelsToDelete = await this.createQueryBuilder(joinTable.name)
|
joinTable,
|
||||||
.select(select)
|
relIdsToadd: relIdsToaddOrMove,
|
||||||
.where({
|
db,
|
||||||
[inverseJoinColumn.name]: relIdsToaddOrMove,
|
});
|
||||||
[joinColumn.name]: { $ne: id },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
|
|
||||||
currentRelsToDelete.sort((a, b) => b[orderColumnName] - a[orderColumnName]);
|
|
||||||
|
|
||||||
for (const relToDelete of currentRelsToDelete) {
|
|
||||||
if (relToDelete[orderColumnName] !== null) {
|
|
||||||
await this.createQueryBuilder(joinTable.name)
|
|
||||||
.decrement(orderColumnName, 1)
|
|
||||||
.where({
|
|
||||||
[joinColumn.name]: relToDelete[joinColumn.name],
|
|
||||||
[orderColumnName]: { $gt: relToDelete[orderColumnName] },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// delete previous oneToAny relations
|
|
||||||
await this.createQueryBuilder(joinTable.name)
|
|
||||||
.delete()
|
|
||||||
.where({
|
|
||||||
[inverseJoinColumn.name]: relIdsToaddOrMove,
|
|
||||||
[joinColumn.name]: { $ne: id },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
}
|
|
||||||
|
|
||||||
// Delete the previous relations for anyToOne relations
|
// Delete the previous relations for anyToOne relations
|
||||||
if (isBidirectional(attribute) && isAnyToOne(attribute)) {
|
await deletePreviousAnyToOneRelations({
|
||||||
// update orders for previous anyToOne relations that will be deleted if it has order (manyToOne)
|
id,
|
||||||
if (isManyToAny(attribute)) {
|
attribute,
|
||||||
const currentRelsToDelete = await this.createQueryBuilder(joinTable.name)
|
joinTable,
|
||||||
.select(select)
|
relIdsToadd: relIdsToaddOrMove,
|
||||||
.where({
|
db,
|
||||||
[joinColumn.name]: id,
|
});
|
||||||
[inverseJoinColumn.name]: { $notIn: relIdsToaddOrMove },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
|
|
||||||
for (const relToDelete of currentRelsToDelete) {
|
|
||||||
if (relToDelete[inverseOrderColumnName] !== null) {
|
|
||||||
await this.createQueryBuilder(joinTable.name)
|
|
||||||
.decrement(inverseOrderColumnName, 1)
|
|
||||||
.where({
|
|
||||||
[inverseJoinColumn.name]: relToDelete[inverseJoinColumn.name],
|
|
||||||
[inverseOrderColumnName]: { $gt: relToDelete[inverseOrderColumnName] },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// delete previous oneToAny relations
|
|
||||||
await this.createQueryBuilder(joinTable.name)
|
|
||||||
.delete()
|
|
||||||
.where({
|
|
||||||
[joinColumn.name]: id,
|
|
||||||
[inverseJoinColumn.name]: { $notIn: relIdsToaddOrMove },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -1270,51 +996,8 @@ const createEntityManager = (db) => {
|
|||||||
|
|
||||||
if (attribute.joinTable) {
|
if (attribute.joinTable) {
|
||||||
const { joinTable } = attribute;
|
const { joinTable } = attribute;
|
||||||
const { joinColumn, inverseJoinColumn, inverseOrderColumnName } = joinTable;
|
|
||||||
|
|
||||||
// INVERSE ORDER UPDATE
|
await deleteAllRelations({ id, attribute, joinTable, db });
|
||||||
if (isBidirectional(attribute) && isManyToAny(attribute)) {
|
|
||||||
let lastId = 0;
|
|
||||||
let done = false;
|
|
||||||
const batchSize = 100;
|
|
||||||
while (!done) {
|
|
||||||
const relsToDelete = await this.createQueryBuilder(joinTable.name)
|
|
||||||
.select(inverseJoinColumn.name, inverseOrderColumnName)
|
|
||||||
.where({
|
|
||||||
[joinColumn.name]: id,
|
|
||||||
id: { $gt: lastId },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.orderBy('id')
|
|
||||||
.limit(batchSize)
|
|
||||||
.execute();
|
|
||||||
done = relsToDelete.length < batchSize;
|
|
||||||
lastId = relsToDelete[relsToDelete.length - 1]?.id;
|
|
||||||
|
|
||||||
const updateInverseOrderPromises = [];
|
|
||||||
for (const relToDelete of relsToDelete) {
|
|
||||||
if (relToDelete[inverseOrderColumnName] !== null) {
|
|
||||||
const updatePromise = this.createQueryBuilder(joinTable.name)
|
|
||||||
.decrement(inverseOrderColumnName, 1)
|
|
||||||
.where({
|
|
||||||
[inverseJoinColumn.name]: relToDelete[inverseJoinColumn.name],
|
|
||||||
[inverseOrderColumnName]: { $gt: relToDelete[inverseOrderColumnName] },
|
|
||||||
})
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
updateInverseOrderPromises.push(updatePromise);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
await Promise.all(updateInverseOrderPromises);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
await this.createQueryBuilder(joinTable.name)
|
|
||||||
.delete()
|
|
||||||
.where({ [joinColumn.name]: id })
|
|
||||||
.where(joinTable.on || {})
|
|
||||||
.execute();
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|||||||
200
packages/core/database/lib/entity-manager/utils.js
Normal file
200
packages/core/database/lib/entity-manager/utils.js
Normal file
@ -0,0 +1,200 @@
|
|||||||
|
'use strict';
|
||||||
|
|
||||||
|
const {
|
||||||
|
isBidirectional,
|
||||||
|
isOneToAny,
|
||||||
|
isManyToAny,
|
||||||
|
isAnyToOne,
|
||||||
|
isAnyToMany,
|
||||||
|
} = require('../metadata/relations');
|
||||||
|
const { createQueryBuilder } = require('../query');
|
||||||
|
|
||||||
|
const getSelect = (joinTable, attribute) => {
|
||||||
|
const { joinColumn, orderColumnName, inverseJoinColumn, inverseOrderColumnName } = joinTable;
|
||||||
|
const select = [joinColumn.name, inverseJoinColumn.name];
|
||||||
|
if (isAnyToMany(attribute)) {
|
||||||
|
select.push(orderColumnName);
|
||||||
|
}
|
||||||
|
if (isBidirectional(attribute) && isManyToAny(attribute)) {
|
||||||
|
select.push(inverseOrderColumnName);
|
||||||
|
}
|
||||||
|
return select;
|
||||||
|
};
|
||||||
|
|
||||||
|
const deletePreviousOneToAnyRelations = async ({ id, attribute, joinTable, relIdsToadd, db }) => {
|
||||||
|
const { joinColumn, inverseJoinColumn, orderColumnName } = joinTable;
|
||||||
|
const select = getSelect(joinTable, attribute);
|
||||||
|
|
||||||
|
// need to delete the previous relations for oneToAny relations
|
||||||
|
if (isBidirectional(attribute) && isOneToAny(attribute)) {
|
||||||
|
// update orders for previous oneToAny relations that will be deleted if it has order (oneToMany)
|
||||||
|
if (isAnyToMany(attribute)) {
|
||||||
|
const currentRelsToDelete = await createQueryBuilder(joinTable.name, db)
|
||||||
|
.select(select)
|
||||||
|
.where({
|
||||||
|
[inverseJoinColumn.name]: relIdsToadd,
|
||||||
|
[joinColumn.name]: { $ne: id },
|
||||||
|
})
|
||||||
|
.where(joinTable.on || {})
|
||||||
|
.execute();
|
||||||
|
|
||||||
|
currentRelsToDelete.sort((a, b) => b[orderColumnName] - a[orderColumnName]);
|
||||||
|
|
||||||
|
for (const relToDelete of currentRelsToDelete) {
|
||||||
|
if (relToDelete[orderColumnName] !== null) {
|
||||||
|
await createQueryBuilder(joinTable.name, db)
|
||||||
|
.decrement(orderColumnName, 1)
|
||||||
|
.where({
|
||||||
|
[joinColumn.name]: relToDelete[joinColumn.name],
|
||||||
|
[orderColumnName]: { $gt: relToDelete[orderColumnName] },
|
||||||
|
})
|
||||||
|
.where(joinTable.on || {})
|
||||||
|
.execute();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// delete previous oneToAny relations
|
||||||
|
await createQueryBuilder(joinTable.name, db)
|
||||||
|
.delete()
|
||||||
|
.where({
|
||||||
|
[inverseJoinColumn.name]: relIdsToadd,
|
||||||
|
[joinColumn.name]: { $ne: id },
|
||||||
|
})
|
||||||
|
.where(joinTable.on || {})
|
||||||
|
.execute();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const deletePreviousAnyToOneRelations = async ({ id, attribute, joinTable, relIdsToadd, db }) => {
|
||||||
|
const { joinColumn, inverseJoinColumn, inverseOrderColumnName } = joinTable;
|
||||||
|
const select = getSelect(joinTable, attribute);
|
||||||
|
|
||||||
|
// Delete the previous relations for anyToOne relations
|
||||||
|
if (isBidirectional(attribute) && isAnyToOne(attribute)) {
|
||||||
|
// update orders for previous anyToOne relations that will be deleted if it has order (manyToOne)
|
||||||
|
if (isManyToAny(attribute)) {
|
||||||
|
const currentRelsToDelete = await createQueryBuilder(joinTable.name, db)
|
||||||
|
.select(select)
|
||||||
|
.where({
|
||||||
|
[joinColumn.name]: id,
|
||||||
|
[inverseJoinColumn.name]: { $notIn: relIdsToadd },
|
||||||
|
})
|
||||||
|
.where(joinTable.on || {})
|
||||||
|
.execute();
|
||||||
|
|
||||||
|
for (const relToDelete of currentRelsToDelete) {
|
||||||
|
if (relToDelete[inverseOrderColumnName] !== null) {
|
||||||
|
await createQueryBuilder(joinTable.name, db)
|
||||||
|
.decrement(inverseOrderColumnName, 1)
|
||||||
|
.where({
|
||||||
|
[inverseJoinColumn.name]: relToDelete[inverseJoinColumn.name],
|
||||||
|
[inverseOrderColumnName]: { $gt: relToDelete[inverseOrderColumnName] },
|
||||||
|
})
|
||||||
|
.where(joinTable.on || {})
|
||||||
|
.execute();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// delete previous oneToAny relations
|
||||||
|
await createQueryBuilder(joinTable.name, db)
|
||||||
|
.delete()
|
||||||
|
.where({
|
||||||
|
[joinColumn.name]: id,
|
||||||
|
[inverseJoinColumn.name]: { $notIn: relIdsToadd },
|
||||||
|
})
|
||||||
|
.where(joinTable.on || {})
|
||||||
|
.execute();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
// INVERSE ORDER UPDATE
|
||||||
|
const deleteAllRelations = async ({
|
||||||
|
id,
|
||||||
|
attribute,
|
||||||
|
joinTable,
|
||||||
|
except = undefined,
|
||||||
|
onlyFor = undefined,
|
||||||
|
db,
|
||||||
|
}) => {
|
||||||
|
const { joinColumn, inverseJoinColumn, orderColumnName, inverseOrderColumnName } = joinTable;
|
||||||
|
const select = getSelect(joinTable, attribute);
|
||||||
|
|
||||||
|
if (isAnyToMany(attribute) || (isBidirectional(attribute) && isManyToAny(attribute))) {
|
||||||
|
let lastId = 0;
|
||||||
|
let done = false;
|
||||||
|
const batchSize = 100;
|
||||||
|
while (!done) {
|
||||||
|
const relsToDelete = await createQueryBuilder(joinTable.name, db)
|
||||||
|
.select(select)
|
||||||
|
.where({
|
||||||
|
[joinColumn.name]: id,
|
||||||
|
id: { $gt: lastId },
|
||||||
|
...(except ? { [inverseJoinColumn.name]: { $notIn: except } } : {}),
|
||||||
|
...(onlyFor ? { [inverseJoinColumn.name]: { $in: onlyFor } } : {}),
|
||||||
|
})
|
||||||
|
.where(joinTable.on || {})
|
||||||
|
.orderBy('id')
|
||||||
|
.limit(batchSize)
|
||||||
|
.execute();
|
||||||
|
done = relsToDelete.length < batchSize;
|
||||||
|
lastId = relsToDelete[relsToDelete.length - 1]?.id;
|
||||||
|
|
||||||
|
// ORDER UPDATE
|
||||||
|
if (isAnyToMany(attribute)) {
|
||||||
|
// sort by order DESC so that the order updates are done in the correct order
|
||||||
|
// avoiding one to interfere with the others
|
||||||
|
relsToDelete.sort((a, b) => b[orderColumnName] - a[orderColumnName]);
|
||||||
|
|
||||||
|
for (const relToDelete of relsToDelete) {
|
||||||
|
if (relToDelete[orderColumnName] !== null) {
|
||||||
|
await createQueryBuilder(joinTable.name, db)
|
||||||
|
.decrement(orderColumnName, 1)
|
||||||
|
.where({
|
||||||
|
[joinColumn.name]: id,
|
||||||
|
[orderColumnName]: { $gt: relToDelete[orderColumnName] },
|
||||||
|
})
|
||||||
|
.where(joinTable.on || {})
|
||||||
|
// manque le pivot ici
|
||||||
|
.execute();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (isBidirectional(attribute) && isManyToAny(attribute)) {
|
||||||
|
const updateInverseOrderPromises = [];
|
||||||
|
for (const relToDelete of relsToDelete) {
|
||||||
|
if (relToDelete[inverseOrderColumnName] !== null) {
|
||||||
|
const updatePromise = createQueryBuilder(joinTable.name, db)
|
||||||
|
.decrement(inverseOrderColumnName, 1)
|
||||||
|
.where({
|
||||||
|
[inverseJoinColumn.name]: relToDelete[inverseJoinColumn.name],
|
||||||
|
[inverseOrderColumnName]: { $gt: relToDelete[inverseOrderColumnName] },
|
||||||
|
})
|
||||||
|
.where(joinTable.on || {})
|
||||||
|
.execute();
|
||||||
|
updateInverseOrderPromises.push(updatePromise);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
await Promise.all(updateInverseOrderPromises);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
await createQueryBuilder(joinTable.name, db)
|
||||||
|
.delete()
|
||||||
|
.where({
|
||||||
|
[joinColumn.name]: id,
|
||||||
|
...(except ? { [inverseJoinColumn.name]: { $notIn: except } } : {}),
|
||||||
|
...(onlyFor ? { [inverseJoinColumn.name]: { $in: onlyFor } } : {}),
|
||||||
|
})
|
||||||
|
.where(joinTable.on || {})
|
||||||
|
.execute();
|
||||||
|
};
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
deletePreviousOneToAnyRelations,
|
||||||
|
deletePreviousAnyToOneRelations,
|
||||||
|
deleteAllRelations,
|
||||||
|
};
|
||||||
Loading…
x
Reference in New Issue
Block a user