mongo-update-document

Instructions

Implementation Reference

• src/index.ts:178-198 (handler)

  Handler function that executes the MongoDB update operation, either updateOne or updateMany based on the updateMany flag.

  async ({ database: dbName, collection: collectionName, filter, update, updateMany = false }) => {
    try {
      const db = await ensureConnection(dbName);
      const collection: Collection = db.collection(collectionName);
      
      const result = updateMany 
        ? await collection.updateMany(filter, update)
        : await collection.updateOne(filter, update);
      
      return {
        content: [
          {
            type: "text",
            text: `Update operation completed. Matched: ${result.matchedCount}, Modified: ${result.modifiedCount}`,
          },
        ],
      };
    } catch (error) {
      throw new Error(`Failed to update document(s): ${error instanceof Error ? error.message : 'Unknown error'}`);
    }
  }

• src/index.ts:171-177 (schema)

  Zod input schema defining parameters: database, collection, filter, update, and optional updateMany.

  {
    database: z.string().describe("Database name"),
    collection: z.string().describe("Collection name"),
    filter: z.record(z.any()).describe("Query filter to match documents to update"),
    update: z.record(z.any()).describe("Update operations as JSON object"),
    updateMany: z.boolean().optional().describe("Whether to update multiple documents (default: false)"),
  },

• src/index.ts:168-199 (registration)

  Registration of the mongo-update-document tool with McpServer.tool(), including schema and inline handler implementation.

  server.tool(
    "mongo-update-document",
    "Update documents in a MongoDB collection",
    {
      database: z.string().describe("Database name"),
      collection: z.string().describe("Collection name"),
      filter: z.record(z.any()).describe("Query filter to match documents to update"),
      update: z.record(z.any()).describe("Update operations as JSON object"),
      updateMany: z.boolean().optional().describe("Whether to update multiple documents (default: false)"),
    },
    async ({ database: dbName, collection: collectionName, filter, update, updateMany = false }) => {
      try {
        const db = await ensureConnection(dbName);
        const collection: Collection = db.collection(collectionName);
        
        const result = updateMany 
          ? await collection.updateMany(filter, update)
          : await collection.updateOne(filter, update);
        
        return {
          content: [
            {
              type: "text",
              text: `Update operation completed. Matched: ${result.matchedCount}, Modified: ${result.modifiedCount}`,
            },
          ],
        };
      } catch (error) {
        throw new Error(`Failed to update document(s): ${error instanceof Error ? error.message : 'Unknown error'}`);
      }
    }
  );

• src/index.ts:88-100 (helper)

  Helper function to establish and cache MongoDB connection and database instance, used by the handler.

  async function ensureConnection(dbName: string): Promise<Db> {
    if (!mongoClient) {
      const uri = getMongoUri();
      mongoClient = new MongoClient(uri);
      await mongoClient.connect();
    }
    
    if (!databases.has(dbName)) {
      databases.set(dbName, mongoClient.db(dbName));
    }
    
    return databases.get(dbName)!;
  }