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

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

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

我的真正问题:我将使用哪个 Mongo 命令并不重要。我会想办法的。我真正想知道的是:如何在 upsert 之前自定义 mgo 的行为？

一些解决方案:我之前自己尝试过一些解决方案。例如，将 Game 结构编码/解码为 bson.M 以对其进行自定义。但是，我发现它既麻烦又凌乱。如果没有其他办法，我会使用它。

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

示例:

// 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
})

最佳答案

要回答“我的真正问题:如何在更新插入之前自定义 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});

为了清楚起见 - 自定义编码(marshal)拆收器将用于所有 mgo 查询，因此您可能不想将其直接定义到模型，而是定义到它的派生对象以仅在更新插入操作中使用:

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})

关于mongodb - 自定义 mgo upsert 操作，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/45718580/