自定义 mgo upsert 操作

对于一个Golang开发者来说，牢固扎实的基础是十分重要的，golang学习网就来带大家一点点的掌握基础知识点。今天本篇文章带大家了解《自定义 mgo upsert 操作》，主要介绍了golang，希望对大家的知识积累有所帮助，快点收藏起来吧，否则需要时就找不到了！

问题内容

我有一个游戏分析休息 API，它存储玩家的平均表现统计数据。当新的统计数据到来时，我想通过将新的 delta 合并到现有文档中来更新 Mongodb 中现有的游戏记录。我也在存储过去的分析数据。因此，我可以返回自游戏上次更新以来玩家的统计数据正在减少或增加的数据。

问题是： 当我想用 mgo 将我的新游戏数据插入到 Mongodb 中时，它会覆盖所有玩家的统计数据数组。实际上，这是意料之中的。如果我可以修改 mgo 尝试将其插入到 Mongodb 中的文档，我知道如何修复它。

问题： 如何自定义 mgo upsert 行为？这样我就可以$push在前面添加一个运算符，Player.Stats以防止 Mongodb 擦除stats文档内的数组。

我真正的问题： 我将使用哪个 Mongo 命令并不重要。我会以某种方式解决的。我真正想知道的是：如何在 upsert 之前自定义 mgo 的行为？

一些解决方案： 我之前自己尝试过一些解决方案。就像，编码/解码Game结构到bson.M自定义它。但是，我发现它既麻烦又凌乱。如果没有其他方法，我会使用它。

块： 我不想用 手写我的所有结构字段，只是为了在一个字段上bson.M使用运算符。$push因为有几十个字段，这很容易出错并且会增加我的代码复杂性。

例子：

// Assume that, this is an existing game in Mongodb:
existingGame := Game{
    ID: 1,
    Name: "Existing game",
    // The game has just one player
    Players: []Player{
        // The player has some stats. The newest one is 2.0.
        {1, "foo", []{3.5, 2.0}},
    }
}

// This is a new request coming to my API
// I want to upsert this into the existing Game
newGame := Game{
    ID: 1,
    Players: []Player{
        // As expectedly, this will reset player foo's stats to 5.0
        //
        // After upserting, I want it to be as: 
        //
        // []{3.5, 2.0, 5.0}
        //
        // in Mongodb
        {1, "foo", []{5.0}},
    }
}

// Example 2:
// If new Game request like this:
newGame := Game{ID: 1, Players: []Player{{1, "foo", []{5.0},{1, "bar", []{6.7}}}}
// I'm expecting this result:
Game{ID: 1, Players: []Player{{1, "foo", []{3.5, 2.0, 5.0},{1, "bar", []{6.7}}}}

func (db *Store) Merge(newGame *Game) error {
    sess := db.session.Copy()
    defer sess.Close()

    col := sess.DB("foo").C("games")
    // I want to modify newGame here to add a $push operator
    // into a new `bson.M` or `bson.D` to make mgo to upsert
    // my new delta without resetting the player stats
    _, err := col.UpsertId(newGame.ID, newGame)

    return err
}

type Game struct {
    ID int `bson:"_id"`
    Name string
    Players []Player `bson:",omitempty"`
    // ...I omitted other details for simplicity here...
}

type Player struct {
    // This connects the player to the game
    GameID int `bson:"game_id"`
    Name string
    // I want to keep the previous values of stats
    // So, that's why I'm using an array here
    Stats []float64
    // ...
}

我在控制台中尝试了这个 Mongodb 命令来更新特定游戏的播放器：

db.competitions.update({
   _id: 1,
   "players.game_id": 1
}, {
   $push: { 
       "players.$.stats": 3
   }
}, {
   upsert: true
})

正确答案

要回答“ 我的真正问题：如何在 upsert 之前自定义 mgo 的行为？ ” - 您可以通过为模型定义 bson Getter来自定义 bson 编组。

为了说明它是如何工作的，让我们简化模型以避免嵌套文档：

type Game struct {
    ID int `bson:"_id"`
    Name string
    Stats [] float64
}

使用 newGame 如下：

newGame := Game{
    ID: 1,
    Name: "foo",
    Stats: []{5.0}
}

默认情况下，更新col.UpsertId(newGame.ID, newGame)编组newGame为 JSON，生成 mongo 查询，如：

update({_id:1}, {name: "foo", stats: [5]}, {upsert: true});

要使用$set等$push，您可以定义一个自定义的 bson getter。例如

func (g Game) GetBSON() (interface{}, error) {
    return bson.M{
        "$set": bson.M{"name": g.Name}, 
        "$push": bson.M{"stats": bson.M{"$each": g.Stats}},
    }, nil
}

所以更新col.UpsertId(newGame.ID, newGame)会产生一个mongodb查询

update({_id:1}, {$set: {name: "foo"}, $push: {stats: {$each: [5]}}}, {upsert: true});

为了清楚起见 - 自定义封送拆收器将用于所有 mgo 查询，因此您可能不想将其直接定义为模型，而是定义为仅在 upsert 操作中使用的派生类：

type UpdatedGame struct {
    Game
}

func (g UpdatedGame) GetBSON() (interface{}, error) {
    return bson.M{....}
}

.....

newGame := Game{
    ID: 1,
    Name: "foo",
    Stats: []{5.0}
}

col.UpsertId(newGame.ID, UpdatedGame{newGame})

今天关于《自定义 mgo upsert 操作》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！