MongoDB中createCollection函数变量未更新返回UNKNOWN问题

我写了一个MongoCreationCollection函数，用来检测MongoDB里指定的集合是否存在。函数应该返回"EXIST"（存在）、"NOT_EXIST"（不存在），默认值是"UNKNOWN"。

运行JS程序后，函数返回"UNKNOWN"，但控制台日志显示正确的"EXIST"状态。运行日志如下：

VirtualBox:\~/PROJECT/Docker-Nodejs/internal_modules$ node mongo.js
Debugger attached.
Code running
return variable UNKNOWN
result variable : UNKNOWN
console log : EXIST

为什么函数里的variable变量没被更新？相关函数代码如下：

function MongoCreationCollection(databasename,name){
    var variable = "UNKNOWN";
    MongoClient.connect(url, function(err, client) {
        const db = client.db(databasename)

        db.createCollection(name, function (err, res) {
            if (err) {
                //console.log(err);
                if (err.codeName =="NamespaceExists") {
                    variable="EXIST"
                    console.log("console log : "+variable);
                }
            }
            else {
                console.log("Collection created! : "+name+"");
                variable="NOT_EXIST"
                console.log("console log : "+variable);
            }
        });
    });
    console.log("return variable "+variable);
    return variable;
};

调用代码：

var result = MongoCreationCollection(database,"enfants");
console.log("result variable : "+result+"");

核心原因：异步代码执行顺序问题 #

MongoClient.connect和db.createCollection都是异步操作，它们的回调函数会在主线程执行完之后才被调用。执行顺序是：

1. 函数初始化variable = "UNKNOWN"
2. 发起MongoDB连接和集合创建的异步请求，此时回调函数还未执行
3. 直接执行console.log("return variable "+variable)并返回初始值"UNKNOWN"
4. 异步操作完成后，回调函数才会修改variable，但此时函数已经返回，返回值无法更新

解决方案：用Promise/async-await处理异步逻辑 #

要让函数正确返回异步操作的结果，必须用异步模式改写：

方案1：Promise封装

function MongoCreationCollection(databasename, name) {
    return new Promise((resolve, reject) => {
        MongoClient.connect(url, function(err, client) {
            if (err) {
                reject(err);
                return;
            }
            const db = client.db(databasename);

            db.createCollection(name, function (err, res) {
                client.close(); // 务必关闭连接，避免资源泄漏
                if (err) {
                    if (err.codeName === "NamespaceExists") {
                        console.log("console log : EXIST");
                        resolve("EXIST");
                    } else {
                        reject(err);
                    }
                } else {
                    console.log("Collection created! : " + name);
                    console.log("console log : NOT_EXIST");
                    resolve("NOT_EXIST");
                }
            });
        });
    });
}

调用方式：

MongoCreationCollection(database, "enfants")
    .then(result => {
        console.log("result variable : " + result);
    })
    .catch(err => {
        console.error("出错了：", err);
    });

方案2：async/await（更简洁易读）

async function MongoCreationCollection(databasename, name) {
    let client;
    try {
        client = await MongoClient.connect(url);
        const db = client.db(databasename);
        await db.createCollection(name);
        console.log("Collection created! : " + name);
        console.log("console log : NOT_EXIST");
        return "NOT_EXIST";
    } catch (err) {
        if (err.codeName === "NamespaceExists") {
            console.log("console log : EXIST");
            return "EXIST";
        }
        throw err;
    } finally {
        if (client) {
            await client.close(); // 确保连接关闭
        }
    }
}

调用方式：

(async function() {
    try {
        const result = await MongoCreationCollection(database, "enfants");
        console.log("result variable : " + result);
    } catch (err) {
        console.error("出错了：", err);
    }
})();

额外注意事项 #

• 必须关闭MongoDB客户端连接，防止资源泄漏
• 异常处理要覆盖所有可能的错误场景，不要只处理集合已存在的情况

内容的提问来源于stack exchange，提问作者brice