如何结合GraphQL、MongoDB编写AWS Lambda函数并关联集合数据？

我明白你现在的需求——不仅要从MongoDB的user_posts集合获取数据，还要关联user_profile集合，把每个关联用户的username和photo字段补充到结果里，同时还要在AWS Lambda里集成GraphQL来处理这类查询。咱们一步步来解决：

第一步：改进MongoDB查询，实现数据关联 #

你当前的Lambda只是简单查询user_posts，没有关联用户信息。这里需要用MongoDB的聚合框架，通过多次$lookup和数组映射来关联嵌套数组里的用户数据，生成你需要的输出格式。

下面是能实现需求的聚合查询逻辑：

db.collection("user_posts").aggregate([
  // 关联帖子主用户的信息，提取username和photo到根字段
  {
    $lookup: {
      from: "user_profile",
      localField: "userid",
      foreignField: "_id",
      as: "userInfo"
    }
  },
  { $unwind: "$userInfo" },
  {
    $addFields: {
      username: "$userInfo.username",
      photo: "$userInfo.photo"
    }
  },
  { $project: { userInfo: 0 } },

  // 处理like数组，给每个点赞对象关联用户信息
  {
    $lookup: {
      from: "user_profile",
      localField: "like.userid",
      foreignField: "_id",
      as: "likeUserInfo"
    }
  },
  {
    $addFields: {
      like: {
        $map: {
          input: "$like",
          as: "likeItem",
          in: {
            $mergeObjects: [
              "$$likeItem",
              {
                $arrayElemAt: [
                  "$likeUserInfo",
                  { $indexOfArray: ["$likeUserInfo._id", "$$likeItem.userid"] }
                ]
              }
            ]
          }
        }
      }
    }
  },
  { $project: { likeUserInfo: 0 } },

  // 处理comment数组，给每个评论对象关联用户信息
  {
    $lookup: {
      from: "user_profile",
      localField: "comment.userid",
      foreignField: "_id",
      as: "commentUserInfo"
    }
  },
  {
    $addFields: {
      comment: {
        $map: {
          input: "$comment",
          as: "commentItem",
          in: {
            $mergeObjects: [
              "$$commentItem",
              {
                $arrayElemAt: [
                  "$commentUserInfo",
                  { $indexOfArray: ["$commentUserInfo._id", "$$commentItem.userid"] }
                ]
              }
            ]
          }
        }
      }
    }
  },
  { $project: { commentUserInfo: 0 } },

  // 处理share数组，给每个分享对象关联用户信息
  {
    $lookup: {
      from: "user_profile",
      localField: "share.userid",
      foreignField: "_id",
      as: "shareUserInfo"
    }
  },
  {
    $addFields: {
      share: {
        $map: {
          input: "$share",
          as: "shareItem",
          in: {
            $mergeObjects: [
              "$$shareItem",
              {
                $arrayElemAt: [
                  "$shareUserInfo",
                  { $indexOfArray: ["$shareUserInfo._id", "$$shareItem.userid"] }
                ]
              }
            ]
          }
        }
      }
    }
  },
  { $project: { shareUserInfo: 0 } },

  // 过滤指定用户的帖子
  { $match: { userid: uid } }
]).toArray()

第二步：在AWS Lambda中集成GraphQL #

推荐用apollo-server-lambda（专门针对Lambda优化的GraphQL服务框架）来实现，下面是完整的代码步骤：

1. 安装依赖 #

在你的Lambda项目目录下运行：

npm install apollo-server-lambda mongodb

2. 编写完整的Lambda代码 #

const { ApolloServer, gql } = require('apollo-server-lambda');
const { MongoClient, ObjectId } = require('mongodb');

// 1. 定义GraphQL Schema，描述数据结构
const typeDefs = gql`
  type User {
    _id: ID!
    username: String!
    photo: String
  }

  type Like {
    userid: ID!
    status: String!
    username: String!
    photo: String
  }

  type Comment {
    userid: ID!
    comment: String!
    username: String!
    photo: String
  }

  type Share {
    userid: ID!
    status: String!
    username: String!
    photo: String
  }

  type Post {
    _id: ID!
    userid: ID!
    username: String!
    photo: String
    media: String!
    type: String!
    created: String
    modified: String
    like: [Like]
    comment: [Comment]
    share: [Share]
  }

  type Query {
    getUserPosts(userid: ID!): [Post]
  }
`;

// 2. 定义Resolver，处理数据查询逻辑
const resolvers = {
  Query: {
    getUserPosts: async (_, { userid }) => {
      const uri = '你的MongoDB连接字符串'; // 替换为实际连接字符串，建议用Lambda环境变量存储
      const databasename = "trans_db";
      const uid = ObjectId(userid);

      // 建立MongoDB连接
      const client = await MongoClient.connect(uri, { useNewUrlParser: true });
      const db = client.db(databasename);

      // 执行聚合查询
      const posts = await db.collection("user_posts").aggregate([
        {
          $lookup: {
            from: "user_profile",
            localField: "userid",
            foreignField: "_id",
            as: "userInfo"
          }
        },
        { $unwind: "$userInfo" },
        {
          $addFields: {
            username: "$userInfo.username",
            photo: "$userInfo.photo"
          }
        },
        { $project: { userInfo: 0 } },
        {
          $lookup: {
            from: "user_profile",
            localField: "like.userid",
            foreignField: "_id",
            as: "likeUserInfo"
          }
        },
        {
          $addFields: {
            like: {
              $map: {
                input: "$like",
                as: "likeItem",
                in: {
                  $mergeObjects: [
                    "$$likeItem",
                    {
                      $arrayElemAt: [
                        "$likeUserInfo",
                        { $indexOfArray: ["$likeUserInfo._id", "$$likeItem.userid"] }
                      ]
                    }
                  ]
                }
              }
            }
          }
        },
        { $project: { likeUserInfo: 0 } },
        {
          $lookup: {
            from: "user_profile",
            localField: "comment.userid",
            foreignField: "_id",
            as: "commentUserInfo"
          }
        },
        {
          $addFields: {
            comment: {
              $map: {
                input: "$comment",
                as: "commentItem",
                in: {
                  $mergeObjects: [
                    "$$commentItem",
                    {
                      $arrayElemAt: [
                        "$commentUserInfo",
                        { $indexOfArray: ["$commentUserInfo._id", "$$commentItem.userid"] }
                      ]
                    }
                  ]
                }
              }
            }
          }
        },
        { $project: { commentUserInfo: 0 } },
        {
          $lookup: {
            from: "user_profile",
            localField: "share.userid",
            foreignField: "_id",
            as: "shareUserInfo"
          }
        },
        {
          $addFields: {
            share: {
              $map: {
                input: "$share",
                as: "shareItem",
                in: {
                  $mergeObjects: [
                    "$$shareItem",
                    {
                      $arrayElemAt: [
                        "$shareUserInfo",
                        { $indexOfArray: ["$shareUserInfo._id", "$$shareItem.userid"] }
                      ]
                    }
                  ]
                }
              }
            }
          }
        },
        { $project: { shareUserInfo: 0 } },
        { $match: { userid: uid } }
      ]).toArray();

      client.close();
      return posts;
    }
  }
};

// 3. 创建Apollo Server实例
const server = new ApolloServer({
  typeDefs,
  resolvers,
  context: ({ event, context }) => ({
    headers: event.headers,
    functionName: context.functionName,
    event,
    context,
  }),
});

// 导出Lambda处理函数
exports.handler = server.createHandler();

3. 部署与测试 #

• 把代码打包（包含node_modules）上传到AWS Lambda，或者用SAM/Serverless Framework自动化部署。
• 部署完成后，访问Lambda提供的API网关地址，就能打开Apollo的GraphQL Playground，发送以下查询测试：

query {
  getUserPosts(userid: "5d518caed55bc00001d235c1") {
    _id
    userid
    username
    photo
    media
    type
    like {
      userid
      status
      username
      photo
    }
    comment {
      userid
      comment
      username
      photo
    }
    share {
      userid
      status
      username
      photo
    }
  }
}

关键注意事项 #

• 连接复用：在Lambda中建议把MongoDB client实例放在全局变量里，避免每次请求都重新创建连接，提升性能。
• 权限配置：确保Lambda有访问MongoDB的网络权限（比如MongoDB Atlas要把Lambda的IP加入白名单，或使用VPC对等连接）。
• 错误处理：在代码中添加try/catch逻辑，捕获并处理查询错误，避免Lambda直接抛出异常导致服务中断。

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