Golang中如何使用Echo框架结合Mongo Go Models（mgm）

在阅读了你的文章后，我注意到你展示了如何使用Echo框架结合Mongo Go Models（mgm）进行基本的CRUD操作。这是一个很好的起点，但我想补充一些更深入的示例和最佳实践，以帮助其他开发者更好地理解这种集成。

首先，你正确地设置了mgm的配置，并展示了如何定义模型。这里是一个更完整的模型定义示例，包括了一些常用的mgm特性：

package models

import (
    "github.com/Kamva/mgm/v3"
    "go.mongodb.org/mongo-driver/bson/primitive"
)

type User struct {
    mgm.DefaultModel `bson:",inline"`
    Name             string `json:"name" bson:"name"`
    Email            string `json:"email" bson:"email"`
    Age              int    `json:"age" bson:"age"`
}

func NewUser(name, email string, age int) *User {
    return &User{
        Name:  name,
        Email: email,
        Age:   age,
    }
}

对于Echo路由处理程序，这里是一个更完整的CRUD操作示例：

package handlers

import (
    "net/http"
    "github.com/Kamva/mgm/v3"
    "github.com/Kamva/mgm/v3/operator"
    "github.com/labstack/echo/v4"
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/bson/primitive"
    "your-project/models"
)

// CreateUser 创建用户
func CreateUser(c echo.Context) error {
    user := new(models.User)
    if err := c.Bind(user); err != nil {
        return c.JSON(http.StatusBadRequest, map[string]string{"error": err.Error()})
    }
    
    if err := mgm.Coll(user).Create(user); err != nil {
        return c.JSON(http.StatusInternalServerError, map[string]string{"error": err.Error()})
    }
    
    return c.JSON(http.StatusCreated, user)
}

// GetUser 获取单个用户
func GetUser(c echo.Context) error {
    id := c.Param("id")
    objID, err := primitive.ObjectIDFromHex(id)
    if err != nil {
        return c.JSON(http.StatusBadRequest, map[string]string{"error": "Invalid ID"})
    }
    
    user := &models.User{}
    if err := mgm.Coll(user).FindByID(objID, user); err != nil {
        return c.JSON(http.StatusNotFound, map[string]string{"error": "User not found"})
    }
    
    return c.JSON(http.StatusOK, user)
}

// UpdateUser 更新用户
func UpdateUser(c echo.Context) error {
    id := c.Param("id")
    objID, err := primitive.ObjectIDFromHex(id)
    if err != nil {
        return c.JSON(http.StatusBadRequest, map[string]string{"error": "Invalid ID"})
    }
    
    updateData := make(map[string]interface{})
    if err := c.Bind(&updateData); err != nil {
        return c.JSON(http.StatusBadRequest, map[string]string{"error": err.Error()})
    }
    
    user := &models.User{}
    if err := mgm.Coll(user).FindByID(objID, user); err != nil {
        return c.JSON(http.StatusNotFound, map[string]string{"error": "User not found"})
    }
    
    // 使用mgm的Update方法
    if err := mgm.Coll(user).Update(user); err != nil {
        return c.JSON(http.StatusInternalServerError, map[string]string{"error": err.Error()})
    }
    
    return c.JSON(http.StatusOK, user)
}

// DeleteUser 删除用户
func DeleteUser(c echo.Context) error {
    id := c.Param("id")
    objID, err := primitive.ObjectIDFromHex(id)
    if err != nil {
        return c.JSON(http.StatusBadRequest, map[string]string{"error": "Invalid ID"})
    }
    
    user := &models.User{}
    if err := mgm.Coll(user).FindByID(objID, user); err != nil {
        return c.JSON(http.StatusNotFound, map[string]string{"error": "User not found"})
    }
    
    if err := mgm.Coll(user).Delete(user); err != nil {
        return c.JSON(http.StatusInternalServerError, map[string]string{"error": err.Error()})
    }
    
    return c.JSON(http.StatusOK, map[string]string{"message": "User deleted successfully"})
}

// ListUsers 列出所有用户（带分页）
func ListUsers(c echo.Context) error {
    page := c.QueryParam("page")
    limit := c.QueryParam("limit")
    
    // 这里可以添加分页逻辑
    users := []models.User{}
    
    if err := mgm.Coll(&models.User{}).SimpleFind(&users, bson.M{}); err != nil {
        return c.JSON(http.StatusInternalServerError, map[string]string{"error": err.Error()})
    }
    
    return c.JSON(http.StatusOK, users)
}

// SearchUsers 搜索用户
func SearchUsers(c echo.Context) error {
    name := c.QueryParam("name")
    
    users := []models.User{}
    filter := bson.M{}
    
    if name != "" {
        filter["name"] = bson.M{operator.Regex: primitive.Regex{Pattern: name, Options: "i"}}
    }
    
    if err := mgm.Coll(&models.User{}).SimpleFind(&users, filter); err != nil {
        return c.JSON(http.StatusInternalServerError, map[string]string{"error": err.Error()})
    }
    
    return c.JSON(http.StatusOK, users)
}

对于Echo路由的设置，这里是一个更结构化的示例：

package main

import (
    "context"
    "log"
    "time"
    
    "github.com/Kamva/mgm/v3"
    "github.com/labstack/echo/v4"
    "github.com/labstack/echo/v4/middleware"
    "go.mongodb.org/mongo-driver/mongo/options"
    "your-project/handlers"
)

func main() {
    // 初始化mgm
    err := mgm.SetDefaultConfig(nil, "your-db-name", options.Client().ApplyURI("mongodb://localhost:27017"))
    if err != nil {
        log.Fatal(err)
    }
    
    // 创建Echo实例
    e := echo.New()
    
    // 中间件
    e.Use(middleware.Logger())
    e.Use(middleware.Recover())
    e.Use(middleware.CORS())
    
    // 路由
    api := e.Group("/api/v1")
    {
        users := api.Group("/users")
        {
            users.POST("", handlers.CreateUser)
            users.GET("", handlers.ListUsers)
            users.GET("/search", handlers.SearchUsers)
            users.GET("/:id", handlers.GetUser)
            users.PUT("/:id", handlers.UpdateUser)
            users.DELETE("/:id", handlers.DeleteUser)
        }
    }
    
    // 启动服务器
    e.Logger.Fatal(e.Start(":8080"))
}

对于更复杂的查询操作，这里是一个使用mgm高级特性的示例：

// FindUsersWithAggregation 使用聚合管道查询用户
func FindUsersWithAggregation(c echo.Context) error {
    pipeline := []bson.M{
        {
            "$match": bson.M{
                "age": bson.M{"$gte": 18},
            },
        },
        {
            "$group": bson.M{
                "_id": "$age",
                "count": bson.M{"$sum": 1},
                "users": bson.M{"$push": "$$ROOT"},
            },
        },
        {
            "$sort": bson.M{"_id": 1},
        },
    }
    
    var results []bson.M
    if err := mgm.Coll(&models.User{}).SimpleAggregate(&results, pipeline...); err != nil {
        return c.JSON(http.StatusInternalServerError, map[string]string{"error": err.Error()})
    }
    
    return c.JSON(http.StatusOK, results)
}

// TransactionExample 事务处理示例
func TransactionExample(c echo.Context) error {
    ctx := c.Request().Context()
    
    // 开始事务
    err := mgm.Transaction(ctx, func(sessionContext context.Context) error {
        // 创建用户1
        user1 := models.NewUser("John", "john@example.com", 25)
        if err := mgm.Coll(user1).CreateWithCtx(sessionContext, user1); err != nil {
            return err
        }
        
        // 创建用户2
        user2 := models.NewUser("Jane", "jane@example.com", 30)
        if err := mgm.Coll(user2).CreateWithCtx(sessionContext, user2); err != nil {
            return err
        }
        
        return nil
    })
    
    if err != nil {
        return c.JSON(http.StatusInternalServerError, map[string]string{"error": err.Error()})
    }
    
    return c.JSON(http.StatusOK, map[string]string{"message": "Transaction completed"})
}

这些示例展示了如何更充分地利用mgm的特性与Echo框架结合，包括基本的CRUD操作、高级查询、聚合管道和事务处理。