如何从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的{​​{1}}中的等效内容是什么？是否可以只获取符号数组而不是mgo对的集合？

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()方法获得。您必须将切片传递给它，每个元素对应一个聚合阶段。

聚合的结果始终是文档列表，因此如果您希望将结果作为字符串切片，则必须执行＆＃34;转换＆＃34;手动

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()?