diff --git a/src/database/schemas/management/filament.schema.js b/src/database/schemas/management/filament.schema.js index 1570a87..1642abc 100644 --- a/src/database/schemas/management/filament.schema.js +++ b/src/database/schemas/management/filament.schema.js @@ -25,19 +25,10 @@ filamentSchema.virtual('id').get(function () { filamentSchema.set('toJSON', { virtuals: true }); filamentSchema.statics.recalculate = async function (filament, user) { - const orderItemModel = mongoose.model('orderItem'); - const itemId = filament._id; - const draftOrderItems = await orderItemModel - .find({ - 'state.type': 'draft', - itemType: 'filament', - item: itemId, - }) - .populate('order') - .lean(); - - for (const orderItem of draftOrderItems) { - await orderItemModel.recalculate(orderItem, user); + const filamentSkuModel = mongoose.model('filamentSku'); + const skus = await filamentSkuModel.find({ filament: filament._id }).select('_id').lean(); + for (const sku of skus) { + await filamentSkuModel.recalculate(sku, user); } };