func main() {
// needed for making/accessing collections:
host := flag.String("host", "localhost", "host:port string of database to connect to")
dbname := "partitionStress"
collname := "partitionStress"
flag.Parse()
session, err := mgo.Dial(*host)
if err != nil {
log.Fatal("Error connecting to ", *host, ": ", err)
}
// so we are not in fire and forget
session.SetSafe(&mgo.Safe{})
defer session.Close()
indexes := make([]mgo.Index, 3)
indexes[0] = mgo.Index{Key: []string{"pr", "cid"}}
indexes[1] = mgo.Index{Key: []string{"crid", "pr", "cid"}}
indexes[2] = mgo.Index{Key: []string{"pr", "ts", "cid"}}
mongotools.MakeCollections(collname, dbname, 1, session, indexes)
// at this point we have created the collection, now run the benchmark
res := new(iibench.Result)
numWriters := 8
numQueryThreads := 16
workers := make([]benchmark.WorkInfo, 0, numWriters+numQueryThreads)
currCollectionString := mongotools.GetCollectionString(collname, 0)
for i := 0; i < numWriters; i++ {
copiedSession := session.Copy()
defer copiedSession.Close()
var gen = iibench.NewDocGenerator()
gen.CharFieldLength = 100
gen.NumCharFields = 0
workers = append(workers, mongotools.NewInsertWork(gen, copiedSession.DB(dbname).C(currCollectionString), 0))
}
for i := 0; i < numQueryThreads; i++ {
copiedSession := session.Copy()
defer copiedSession.Close()
workers = append(workers, iibench.NewQueryWork(copiedSession, dbname, currCollectionString))
}
{
copiedSession := session.Copy()
defer copiedSession.Close()
var addPartitionItem = partition_stress.AddPartitionWork{copiedSession.DB(dbname), currCollectionString, time.Hour}
workers = append(workers, benchmark.WorkInfo{addPartitionItem, 1, 1, 0})
}
{
copiedSession := session.Copy()
defer copiedSession.Close()
var dropPartitionItem = partition_stress.DropPartitionWork{copiedSession.DB(dbname), currCollectionString, 7 * time.Hour}
workers = append(workers, benchmark.WorkInfo{dropPartitionItem, 1, 1, 0})
}
// have this go for a looooooong time
benchmark.Run(res, workers, time.Duration(1<<32)*time.Second)
}
func main() {
flag.Parse()
if *numInsertsPerThread > 0 && (*numQueryThreads > 0 || *numSeconds > 0) {
log.Fatal("Invalid values for numInsertsPerThread: ", *numInsertsPerThread, ", numQueryThreads: ", *numQueryThreads, ", numSeconds: ", *numSeconds)
}
session, err := mgo.Dial(*host)
if err != nil {
log.Fatal("Error connecting to ", *host, ": ", err)
}
// so we are not in fire and forget
session.SetSafe(&mgo.Safe{})
defer session.Close()
indexes := make([]mgo.Index, 3)
indexes[0] = mgo.Index{Key: []string{"pr", "cid"}}
indexes[1] = mgo.Index{Key: []string{"crid", "pr", "cid"}}
indexes[2] = mgo.Index{Key: []string{"pr", "ts", "cid"}}
mongotools.MakeCollections(*collname, *dbname, *numCollections, session, indexes)
// at this point we have created the collection, now run the benchmark
res := new(iibench.Result)
workers := make([]benchmark.WorkInfo, 0, *numWriters+*numQueryThreads)
for i := 0; i < *numWriters; i++ {
copiedSession := session.Copy()
defer copiedSession.Close()
var gen = iibench.NewDocGenerator()
currCollectionString := mongotools.GetCollectionString(*collname, i%*numCollections)
workers = append(workers, mongotools.NewInsertWork(gen, copiedSession.DB(*dbname).C(currCollectionString), *numInsertsPerThread))
}
for i := 0; i < *numQueryThreads; i++ {
currCollectionString := mongotools.GetCollectionString(*collname, i%*numCollections)
copiedSession := session.Copy()
defer copiedSession.Close()
workers = append(workers, iibench.NewQueryWork(copiedSession, *dbname, currCollectionString))
}
benchmark.Run(res, workers, time.Duration(*numSeconds)*time.Second)
}