如何从 MongoDB 集合中获取聚合

Question

以下集合包含各种交换符号的刻度（在本例中为 BTCUSD 和 BTCEUR）：

{ "_id" : ObjectId("5a08d2b956df9b2302759d1a"), "symbol" : "BTCUSD", "time" : ISODate("2013-04-13T00:00:00Z"), "open" : 112, "close" : 91.1, "high" : 130, "low" : 81.12, "volume" : 23866.6770456 }
{ "_id" : ObjectId("5a08d2b956df9b2302759d1c"), "symbol" : "BTCUSD", "time" : ISODate("2013-04-14T00:00:00Z"), "open" : 91.1, "close" : 90.171, "high" : 109, "low" : 20, "volume" : 16437.2196645 }
{ "_id" : ObjectId("5a08d2b956df9b2302759d1e"), "symbol" : "BTCEUR", "time" : ISODate("2013-04-15T00:00:00Z"), "open" : 89.86, "close" : 83.302, "high" : 104, "low" : 71.497, "volume" : 16393.12856398 }
{ "_id" : ObjectId("5a08d2b956df9b2302759d20"), "symbol" : "BTCEUR", "time" : ISODate("2013-04-16T00:00:00Z"), "open" : 84.27, "close" : 67.588, "high" : 84.48, "low" : 0.01, "volume" : 26092.5432296 }

如何获取Go 集合中实际存在的符号列表？例如，我正在寻找的结果是：

{ "BTCUSD", "BTCEUR" }

这可以让我从交换服务中检索最新的分时值，而不必将我感兴趣的符号存储在某个地方。

从mongo shell 我试过这个......

db.candles.aggregate({$group: {_id:"$symbol"}})

...结果如下：

{ "_id" : "BTCUSD" }
{ "_id" : "BTCEUR" }

在Go 中使用mgo 的等价物是什么？是否可以只获取符号数组而不是 "_id" : "value" 对的集合？

Answer 1

使用Distinct()

使用collection.distinct() 可以轻松完成您想要的操作。在 MongoDB 控制台中，它看起来像这样：

db.candles.distinct("symbol")

在 Go 中，使用 mgo、Query.Distinct()：

var sess *mgo.Session = ... // Acquire session
c := sess.DB("dbname").C("candles")

var symbols []string
err := c.Find(nil).Distinct("symbol", &symbols)
if err != nil {
    log.Printf("Error: %v", err)
    return
}
fmt.Println(symbols)

使用聚合框架

您也可以使用 MongoDB 聚合来完成此操作，该聚合可通过 Collection.Pipe() 方法获得。你必须向它传递一个切片，每个元素对应一个聚合阶段。

聚合的结果始终是文档列表，因此如果您希望结果为字符串切片，则必须手动进行“转换”。

var sess *mgo.Session = ... // Acquire session
c := sess.DB("dbname").C("candles")

pipe := c.Pipe([]bson.M{
    {
        "$group": bson.M{
            "_id": "$symbol",
        },
    },
})
var results = []bson.M{}
if err := pipe.All(&results); err != nil {
    log.Printf("Error: %v", err)
    return
}

// results holds {"_id": "symbol"} documents
// To get a slice of symbols:
symbols := make([]string, len(results))
for i, doc := range results {
    symbols[i] = doc["_id"].(string)
}
fmt.Println(symbols)

有关如何获取切片值而不是文档的更多技术，请参阅Is there a way to get slice as result of Find()?