Question

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

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

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

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

一些解决方案：我之前尝试过一些解决方案。比如，将Game struct编码/解码为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
})

Answer 1

回答＆＃34; 我的真实问题：如何在upsert之前自定义mgo的行为？＆＃34; - 您可以通过为模型定义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。 E.g。

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操作

1 个答案: