नेवला | मिडलवेयर | रोलबैक ऑपरेशन प्री/पोस्ट हुक द्वारा किया जाता है जब कोई त्रुटि फेंकी जाती है

टीएलडीआर; नेवला मिडलवेयर इसके लिए नहीं बनाया गया था।

लेन-देन डालने की यह विधि वास्तव में मिडलवेयर कार्यक्षमता को पैच कर रही है, और आप अनिवार्य रूप से mongoose से पूरी तरह से अलग एक एपीआई बना रहे हैं। मिडलवेयर।

एक अलग फ़ंक्शन में आपकी हटाए गए क्वेरी के लिए तर्क को बदलना बेहतर होगा।

सरल और इच्छित समाधान

लेन-देन प्रबंधन विधि को अपना जादू करने की अनुमति दें, और अपने मूल मॉडल के लिए एक अलग निष्कासन विधि बनाएं। नेवला लपेटता है mongodb.ClientSession.prototype.withTransaction mongoose.Connection.prototype.transaction . के साथ और हमें सत्र को तत्काल या प्रबंधित करने की भी आवश्यकता नहीं है! इसकी और नीचे की लंबाई के बीच के अंतर को देखें। और आप एक अलग फ़ंक्शन की कीमत पर उस मिडलवेयर के आंतरिक भाग को याद रखने के मानसिक सिरदर्द से बचते हैं।

const parentSchema = new mongoose.Schema({
    name: String,
    children: [{ type: mongoose.Schema.Types.ObjectId, ref: "Child" }],
});

const childSchema = new mongoose.Schema({
    name: String,
    parent: { type: mongoose.Schema.Types.ObjectId, ref: "Parent" },
});

// Assume `parent` is a parent document here
async function fullRemoveParent(parent) {
    // The document's connection
    const db = parent.db;

    // This handles everything with the transaction for us, including retries
    // session, commits, aborts, etc.
    await db.transaction(async function (session) {
        // Make sure to associate all actions with the session
        await parent.remove({ session });
        await db
            .model("Child")
            .deleteMany({ _id: { $in: parent.children } })
            .session(session);
    });

    // And done!
}

छोटा एक्सटेंशन

इसे आसान बनाने का एक और तरीका है, एक मिडलवेयर पंजीकृत करना जो केवल एक सत्र iff . प्राप्त करता है _ क्वेरी में एक पंजीकृत है। यदि कोई लेन-देन प्रारंभ नहीं किया गया है, तो हो सकता है कि कोई त्रुटि हो।

const parentSchema = new mongoose.Schema({
    name: String,
    children: [{ type: mongoose.Schema.Types.ObjectId, ref: "Child" }],
});

const childSchema = new mongoose.Schema({
    name: String,
    parent: { type: mongoose.Schema.Types.ObjectId, ref: "Parent" },
});

parentSchema.pre("remove", async function () {
    // Look how easy!! Just make sure to pass a transactional 
    // session to the removal
    await this.db
        .model("Child")
        .deleteMany({ _id: { $in: parent.children } })
        .session(this.$session());

    // // If you want to: throw an error/warning if you forgot to add a session
    // // and transaction
    // if(!this.$session() || !this.$session().inTransaction()) {
    //    throw new Error("HEY YOU FORGOT A TRANSACTION.");
    // }
});

// Assume `parent` is a parent document here
async function fullRemoveParent(parent) {
    db.transaction(async function(session) {
        await parent.remove({ session });
    });
}

जोखिम भरा और जटिल समाधान

यह काम करता है, और पूरी तरह से, बहुत जटिल है। सिफारिश नहीं की गई। किसी दिन टूटने की संभावना है क्योंकि यह नेवला एपीआई की पेचीदगियों पर निर्भर करता है। मुझे नहीं पता कि मैंने इसे क्यों कोडित किया है, कृपया इसे अपनी परियोजनाओं में शामिल न करें ।

import mongoose from "mongoose";
import mongodb from "mongodb";

const parentSchema = new mongoose.Schema({
    name: String,
    children: [{ type: mongoose.Schema.Types.ObjectId, ref: "Child" }],
});

const childSchema = new mongoose.Schema({
    name: String,
    parent: { type: mongoose.Schema.Types.ObjectId, ref: "Parent" },
});

// Choose a transaction timeout
const TRANSACTION_TIMEOUT = 120000; // milliseconds

// No need for next() callback if using an async function.
parentSchema.pre("remove", async function () {
    // `this` refers to the document, not the query
    let session = this.$session();

    // Check if this op is already part of a session, and start one if not.
    if (!session) {
        // `this.db` refers to the documents's connection.
        session = await this.db.startSession();

        // Set the document's associated session.
        this.$session(session);

        // Note if you created the session, so post can clean it up.
        this.$locals.localSession = true;

        //
    }

    // Check if already in transaction.
    if (!session.inTransaction()) {
        await session.startTransaction();

        // Note if you created transaction.
        this.$locals.localTransaction = true;

        // If you want a timeout
        this.$locals.startTime = new Date();
    }

    // Let's assume that we need to remove all parent references in the
    // children. (just add session-associated ops to extend this)
    await this.db
        .model("Child") // Child model of this connection
        .updateMany(
            { _id: { $in: this.children } },
            { $unset: { parent: true } }
        )
        .session(session);
});

parentSchema.post("remove", async function (parent) {
    if (this.$locals.localTransaction) {
        // Here, there may be an error when we commit, so we need to check if it
        // is a 'retryable' error, then retry if so.
        try {
            await this.$session().commitTransaction();
        } catch (err) {
            if (
                err instanceof mongodb.MongoError &&
                err.hasErrorLabel("TransientTransactionError") &&
                new Date() - this.$locals.startTime < TRANSACTION_TIMEOUT
            ) {
                await parent.remove({ session: this.$session() });
            } else {
                throw err;
            }
        }
    }

    if (this.$locals.localSession) {
        await this.$session().endSession();
        this.$session(null);
    }
});

// Specific error handling middleware if its really time to abort (clean up
// the injections)
parentSchema.post("remove", async function (err, doc, next) {
    if (this.$locals.localTransaction) {
        await this.$session().abortTransaction();
    }

    if (this.$locals.localSession) {
        await this.$session().endSession();
        this.$session(null);
    }

    next(err);
});