使用 Go 编写高效的 MongoDB 事务包装器

学习Golang要努力，但是不要急！今天的这篇文章《使用 Go 编写高效的 MongoDB 事务包装器》将会介绍到等等知识点，如果你想深入学习Golang，可以关注我！我会持续更新相关文章的，希望对大家都能有所帮助！

我目前正在将事务逻辑集成到我的 go+mongodb api 中。

我已经创建了这个示例端点。它允许您检索用户文档并将其以 json 编码发送回客户端。

func GetUser(w http.ResponseWriter, r *http.Request) {
    id := mux.Vars(r)["user-id"]
    objectId, _ := primitive.ObjectIDFromHex(id)

    user, err := UserById(objectId)
    if err != nil {
        // ...
    }

    // do some stuff with the user, whatever

    // eventually send it back
    json.NewEncoder(w).Encode(user)
}

func UserById(id primitive.ObjectID) (models.StoredUser, error) {
    session, err := mongoClient.StartSession()
    if err != nil {
        return models.StoredUser{}, fmt.Errorf("failed starting session for transaction | %s", err.Error())
    }
    defer session.EndSession(context.TODO())

    callback := func(ctx mongo.SessionContext) (any, error) {
        res := usersCollection.FindOne(
            ctx,
            bson.M{
                "_id": id,
            },
        )

        if res.Err() != nil {
            return models.StoredUser{}, fmt.Errorf("failed querying db | %s", res.Err())
        }

        return res, nil
    }

    result, err := session.WithTransaction(context.TODO(), callback)
    if err != nil {
        return models.StoredUser{}, fmt.Errorf("failed executing transaction | %s", err.Error())
    }

    asserted := result.(*mongo.SingleResult)
    var ret models.StoredUser
    if err := asserted.Decode(&ret); err != nil {
        return models.StoredUser{}, fmt.Errorf("failed parsing user data in struct | %s", err.Error())
    }

    return ret, nil
}

以下是主要步骤：

1. 解析请求内容获取用户id
2. 创建会话来执行事务
3. 使用 id 参数声明回调函数
4. 从事务中调用回调函数
5. 获取 *mongo.singleresult 作为 interface{} 并将其解析回其原始类型
6. 解码 *mongo.singleresult 中包含的 bson 文档，将其放入返回结构体中

这个函数可以工作，但是非常冗长。代码非常重复。

我想知道是否有一种方法可以不为我想要创建的每个函数重复相同的代码。我之前的包装函数尝试没有产生任何结果，因为我实际上需要变量现在在每次调用时所在的位置。

尽管如此，第 2 步和第 5 步似乎显得非常多余且效率低下。

有人知道吗？

正确答案

好的，我发现了以下内容：

func transaction(callback func(ctx mongo.sessioncontext) (any, error)) (any, error) {
    session, err := mongoclient.startsession()
    if err != nil {
        return nil, fmt.errorf("failed creating session | %s", err.error())
    }

    defer session.endsession(context.todo())

    res, err := session.withtransaction(ctx, callback)
    if err != nil {
        return nil, fmt.errorf("failed executing transaction | %s", err.error())
    }

    return res, nil
}

假设我想获取用户对象：

func GetUsers() ([]models.User, error) {
    callback := func(ctx mongo.SessionContext) (any, error) {
        res, err := usersCollection.Find(ctx, bson.M{})
        if err != nil {
            return nil, fmt.Errorf("failed querying users collection | %s", err.Error())
        }

        var ret []models.User
        if err := res.All(context.TODO(), &ret); err != nil {
            return nil, fmt.Errorf("failed parsing results in struct | %s", err.Error())
        }

        return ret, nil
    }

    result, err := Transaction(callback)
    if err != nil {
        return []models.User{}, fmt.Errorf("failed executing transaction | %s", err.Error())
    }

    classes, _ := result.([]models.StoredClass)
    return classes, nil
}

以上就是《使用 Go 编写高效的 MongoDB 事务包装器》的详细内容，更多关于的资料请关注golang学习网公众号！