mgo 实现mongodb中的sortByCount方法

最新推荐文章于 2024-04-27 07:00:00 发布

谁是人生一场梦

最新推荐文章于 2024-04-27 07:00:00 发布

阅读量2.1k

点赞数

CC 4.0 BY-SA版权

分类专栏： Go 文章标签： golang go mgo mongodb sortByCount

本文链接：https://blog.youkuaiyun.com/u014239709/article/details/80599198

Go 专栏收录该内容

3 篇文章

订阅专栏

本文介绍如何使用Go语言实现MongoDB的sortByCount方法，通过管道操作完成数据聚合与排序，适用于需要对特定字段进行计数并排序的应用场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

mgo使用pipe来实现mongodb中的aggregation,所以在实现mongodb中的一些方法的时候就需要我们自己去拼接，现在使用go给大家分享一下我实现sortByCount的方法，大家可以自己自己实现一些其它的方法，如果有需要，我会继续实现一些其中的方法，希望大家多多关注！现在直接上代码

type FieldCount struct {
    Field string `json:"field"`
    Count int    `json:"count"`
}
type fieldGroup struct {
    ID    map[string]string `bson:"_id,omitempty"`
    Count int
}

/*
 *
 * db 数据库名字 
 * coll collections
 * field 按照哪个字段分类
 * sort 是否排序
 */
func SortByCount(db string, coll string, field string, sort bool) (fieldCount []FieldCount, err error) {
    session, err := mgo.DialWithTimeout(MONGODB_URL, time.Second)
    if err != nil {
        panic(err)
    }
    defer session.Close()
    session.SetMode(mgo.Monotonic, true)
    c := session.DB(db).C(coll)
    var pipeLine []bson.M
    if sort {
        pipeLine = []bson.M {
            bson.M{"$group" : bson.M{"_id" : bson.M{field : "$" + field}, "count" : bson.M{"$sum" : 1}}},
            bson.M{"$sort" : bson.M{"count" : -1}},
        }
    } else {
        pipeLine = []bson.M {
            bson.M{"$group" : bson.M{"_id" : bson.M{field : "$" + field}, "count" : bson.M{"$sum" : 1}}},
        }
    }
    pipe := c.Pipe(pipeLine)
    var group []fieldGroup
    err = pipe.All(&group)

    for _, g := range group {
        fieldCount = append(fieldCount, FieldCount {
            Field: g.ID[field],
            Count: g.Count,
        })
    }
    return
}

下面是我的测试方法：

func TestSortByCount(t *testing.T) {
    fieldcounttest, _ := SortByCount(constantutils.IP_URL, "logs20180601", "host", true)
    for _, item := range fieldcounttest {
        fmt.Print(item.Field + " : ")
        fmt.Println(item.Count)
    }
}