使用Golang和mgo从集合MongoDB获取元素

时间:2016-02-27 15:29:56

标签: mongodb go chat mgo

我的任务是写历史聊天。因此,为了创建历史记录,我需要将每条消息发送给Mongodb,当我有下一个连接时,我需要获取所有消息,并使用循环发送给所有连接到聊天的客户端

这是我的ChatServer的代码

func ChatServer(ws *websocket.Conn) {

// Connecting to MongoDB, collection History
session, err := mgo.Dial("mongodb://******:*******@ds045795.mongolab.com:45795/catalog")
if err != nil {
    panic(err)
}
defer session.Close()
session.SetMode(mgo.Monotonic, true)
c := session.DB("catalog").C("History")

// fmt.Println(c.Find())
// Adding clients to the map
clientId := ws.RemoteAddr().String()
defer ws.Close()
clients[ws] = true

// Loop for receiving msg
for {
    var msg string
    // If can not read msg - delete client from map
    if err := websocket.Message.Receive(ws, &msg); err != nil {
        delete(clients, ws)
        return
    }
    sendAll(msg)
    err = c.Insert(&Connect{clientId, msg})
    if err != nil {
        log.Fatal(err)
    }
}
}

所以我的问题正在从集合中按顺序获取所有元素。 我不知道该怎么做,因为在文档中找不到合适的功能。 也许你还有其他优惠?

1 个答案:

答案 0 :(得分:0)

首先,我同意上述评论者的意见 - 您应该为Connect结构添加时间戳。但即使没有它,您也可以按ObjectID对条目进行排序,因为它的一部分是时间戳。是的,这是相当肮脏的方式,可能cause issues if you use sharding or transfer database to another server,但在你的情况下(单个Mongolab实例)它可能与类似的东西(我不知道你的结构所以“ip”和“消息”只是假设)

var connects []Connect
c.Find(bson.M{"ip": "127.0.0.1"}).Sort("-_id").Limit(50).All(&connects) // 50 entries in desc order

for _, connect := range connects {
    log.Println(connect.Message)
}

但是真的为你的Connect添加时间

package main

import (
    "fmt"
    "gopkg.in/mgo.v2"
    "gopkg.in/mgo.v2/bson"
    "log"
    "time"
)

type Connect struct {
    Ip      string
    Message string
    Ts      time.Time
}

func main() {
    session, err := mgo.Dial("mongodb://souser:123456@ds055855.mlab.com:55855/catalog")

    if err != nil {
        panic(err)
    }
    defer session.Close()
    session.SetMode(mgo.Monotonic, true)

    c := session.DB("catalog").C("History")

    for i := 0; i < 100; i++ {
        c.Insert(&Connect{"127.0.0.2", fmt.Sprintf("Test message #%d", i), time.Now()})
        if err != nil {
            log.Fatal(err)
        }
    }

    var connects []Connect
    c.Find(bson.M{"ip": "127.0.0.2"}).Sort("-ts").Limit(50).All(&connects)

    for _, connect := range connects {
        log.Println(connect.Message)
    }
}