mongodb把一个类型的三张表转成一个表

原创

mob649e8169ec5f 2024-09-05 06:04:25 ©著作权

文章标签 数据数据结构格式化输出 文章分类 MongoDB 数据库

©著作权归作者所有：来自51CTO博客作者mob649e8169ec5f的原创作品，请联系作者获取转载授权，否则将追究法律责任

MongoDB将多个集合合并为一个集合

在MongoDB中，常常会面对需要将多个集合合并为一个集合的场景。这种情况通常发生在数据结构设计阶段，或者在数据分析和展示时。本文将介绍如何利用MongoDB的聚合框架和一些基本操作将三张表（集合）合并为一张表，并提供相应的代码示例。

背景知识

在MongoDB中，集合（Collection）是数据存储的基本单位，与传统关系型数据库中的表相似。为了更好地管理数据，有时我们会把相关的数据放在不同的集合里。在某些情况下，我们可能需要把这些数据集合合并成一个更加便于查询的集合。

假设我们有三个相关的集合，分别是：

users: 存储用户的基本信息
orders: 存储用户订单信息
products: 存储产品信息

这三张表的结构如下所示：

classDiagram
    class Users {
        +ObjectId userId
        +String name
        +String email
    }

    class Orders {
        +ObjectId orderId
        +ObjectId userId
        +ObjectId productId
        +Date orderDate
    }

    class Products {
        +ObjectId productId
        +String productName
        +Number price
    }

合并操作

我们希望将这三张表的信息合并成一个新的集合userOrders，每条记录包括用户的姓名、电子邮件、订单时间和购买的产品名称及价格。

1. MongoDB聚合框架

MongoDB的聚合框架强大而灵活，常用于复杂数据处理。我们可以使用lookup来连接不同的集合，并使用project来格式化输出的数据结构。

2. 示例代码

下面是实现该功能的示例代码：

// 引入MongoDB客户端
const { MongoClient } = require('mongodb');

// MongoDB连接字符串
const uri = "mongodb://localhost:27017";

// 创建MongoDB客户端
const client = new MongoClient(uri);

// 主函数
async function mergeCollections() {
    try {
        // 连接到MongoDB
        await client.connect();
        const database = client.db('myDatabase');
        
        const usersCollection = database.collection('users');
        const ordersCollection = database.collection('orders');
        const productsCollection = database.collection('products');
        
        // 聚合查询
        const mergedData = await ordersCollection.aggregate([
            {
                $lookup: {
                    from: 'users',
                    localField: 'userId',
                    foreignField: 'userId',
                    as: 'userInfo'
                }
            },
            {
                $unwind: "$userInfo"
            },
            {
                $lookup: {
                    from: 'products',
                    localField: 'productId',
                    foreignField: 'productId',
                    as: 'productInfo'
                }
            },
            {
                $unwind: "$productInfo"
            },
            {
                $project: {
                    _id: 0,
                    userName: "$userInfo.name",
                    userEmail: "$userInfo.email",
                    orderDate: "$orderDate",
                    productName: "$productInfo.productName",
                    productPrice: "$productInfo.price"
                }
            }
        ]).toArray();
        
        console.log(JSON.stringify(mergedData, null, 2));
        
        // 将结果插入新的集合
        const userOrdersCollection = database.collection('userOrders');
        await userOrdersCollection.insertMany(mergedData);

        console.log("数据合并完成，已插入userOrders集合。");
        
    } finally {
        // 关闭MongoDB连接
        await client.close();
    }
}

// 执行主函数
mergeCollections().catch(console.error);