mongodb-guide-skill

MongoDB Module Skill

面向已有后端项目的开发者，快速集成 MongoDB 能力。

能力清单

能力	说明
连接配置	MongoClient/连接池/环境变量配置
文档模型	Document/Entity 设计
CRUD 操作	增删改查/批量操作
聚合查询	Match/Group/Lookup/Pipeline
索引优化	索引设计/慢查询优化
事务处理	多文档事务

触发场景

用户说"帮我加 MongoDB"或"集成 MongoDB"时触发。

核心配置

Java (Spring Boot)

# application.yml
spring:
  data:
    mongodb:
      uri: mongodb://${MONGO_HOST:localhost}:${MONGO_PORT:27017}/${MONGO_DATABASE:myapp}
      uuid-representation: standard

Python (FastAPI)

# config.py
from pydantic_settings import BaseSettings

class Settings(BaseSettings):
    mongo_host: str = "localhost"
    mongo_port: int = 27017
    mongo_database: str = "myapp"

# database.py
from motor.motor_asyncio import AsyncIOMotorClient

client = AsyncIOMotorClient(f"mongodb://{host}:{port}")
db = client[database]

Go (Gin + mongo-driver)

// config/config.go
package config

import (
    "fmt"
    "os"
)

type Config struct {
    MongoHost     string
    MongoPort     string
    MongoDatabase string
}

func Load() *Config {
    return &Config{
        MongoHost:     getEnv("MONGO_HOST", "localhost"),
        MongoPort:     getEnv("MONGO_PORT", "27017"),
        MongoDatabase: getEnv("MONGO_DATABASE", "myapp"),
    }
}

func (c *Config) URI() string {
    return fmt.Sprintf("mongodb://%s:%s", c.MongoHost, c.MongoPort)
}

func getEnv(key, defaultVal string) string {
    if val := os.Getenv(key); val != "" {
        return val
    }
    return defaultVal
}

// database/database.go
package database

import (
    "context"
    "fmt"
    "log"
    "time"

    "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
    "go.mongodb.org/mongo-driver/mongo/readpref"
)

var (
    Client *mongo.Client
    DB     *mongo.Database
)

func Init(uri, dbName string) {
    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel()

    var err error
    Client, err = mongo.Connect(ctx, options.Client().ApplyURI(uri))
    if err != nil {
        log.Fatalf("连接 MongoDB 失败: %v", err)
    }

    if err = Client.Ping(ctx, readpref.Primary()); err != nil {
        log.Fatalf("Ping MongoDB 失败: %v", err)
    }

    DB = Client.Database(dbName)
    fmt.Println("MongoDB 连接成功")
}

func Close() {
    if Client != nil {
        ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
        defer cancel()
        Client.Disconnect(ctx)
    }
}