Golang Mgo 起搏

Question

我正在编写一个快速写入 mongodb 的应用程序。mongodb 和 mgo 处理速度太快了。我的问题是，有没有办法让我确定 mongo 跟不上并开始阻塞？但我也不想不必要地阻止。这是模拟问题的代码示例：

package main

import (

  "labix.org/v2/mgo"

  "time"

  "fmt"

)

// in database name is a string and age is an int

type Dog struct{

  Breed string "breed"

}

type Person struct{

  Name string "name"

  Pet Dog `bson:",inline"`

  Ts        time.Time

}

func insert(session *mgo.Session, bob Person){

  err := session.DB("db_log").C("people").Insert(&bob)

  if err != nil {

    panic("Could not insert into database")

  }

}

func main() {

  session, _ := mgo.Dial("localhost:27017")

  bob := Person{Name : "Robert", Pet : Dog{}}

  i := 0

  for {

    time.Sleep(time.Duration(1) * time.Microsecond)

    i++

    go insert(session, bob)

  }

}

我经常收到以下错误：

panic: Could not insert into database

或者

panic: write tcp 127.0.0.1:27017: i/o timeout

Answer 1

我怀疑如果你允许 Go 使用多个线程和Copy() 然后 Close()你的会话，你会获得更好的性能。要回答您的问题，这可能是频道的完美用例。在一个 goroutine 中将项目送入通道，并在另一个 goroutine 中使用它们/将它们写入 Mongo。您可以调整通道的大小以满足您的需要。一旦通道已满，生产者线程将在尝试向其发送时阻塞。您可能还想使用Safe()方法设置。设置 W:0 将使 Mongo 处于“即发即忘”模式，这将极大地提高性能，但可能会丢失一些数据。您还可以更改超时时间。