Golang Mgo起搏

Gar*_*ary 1 go mgo

我正在编写一个快速写入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)
  }
}
Run Code Online (Sandbox Code Playgroud)

我经常会遇到如下错误:

panic: Could not insert into database
Run Code Online (Sandbox Code Playgroud)

要么

panic: write tcp 127.0.0.1:27017: i/o timeout
Run Code Online (Sandbox Code Playgroud)

Tys*_*son 6

我怀疑如果允许Go使用多个线程并且Copy()然后Close()你的会话,你将获得更好的性能.

要回答你的问题,这可能是一个完美的通道用例.将项目输入到一个goroutine中的通道中并将它们消耗/写入另一个中的Mongo.您可以调整通道的大小以满足您的需求.生成器线程在尝试发送到它时,一旦通道已满就会阻塞.

您可能还想使用Safe()方法设置.设置W:0将使Mongo处于"即发即忘"模式,这将大大加快性能,但可能会丢失一些数据.您还可以更改超时时间.