mirrors/mongodb
mirrors/mongodb
0
0
Fork 0
mirror of https://github.com/mongodb/mongo.git synced 2024-12-01 09:32:32 +01:00
Code Issues Releases Activity

Cross platform RNG

Browse Source
This commit is contained in:
greg 2011-03-01 16:04:57 -05:00
parent ea825cd6c7
commit df9b6cef28
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
client/distlock_test.cpp
View File

@ -22,6 +22,12 @@
#include "../db/commands.h"
#include "../util/bson_util.h"
// Cross-platform RNG
#include <boost/random/mersenne_twister.hpp>
#include <boost/random/uniform_int.hpp>
#include <boost/random/variate_generator.hpp>
// TODO: Make a method in BSONObj if useful, don't modify for now
#define string_field(obj, name, def) ( obj.hasField(name) ? obj[name].String() : def )
#define number_field(obj, name, def) ( obj.hasField(name) ? obj[name].Number() : def )
@ -168,6 +174,14 @@ namespace mongo {
// Whether or not we should hang some threads
int hangThreads = (int) number_field(cmdObj, "hangThreads", 0);
boost::mt19937 gen(seed);
boost::variate_generator<boost::mt19937&, boost::uniform_int<> > randomSkew(gen, boost::uniform_int<>(0, skewRange));
boost::variate_generator<boost::mt19937&, boost::uniform_int<> > randomWait(gen, boost::uniform_int<>(1, threadWait));
boost::variate_generator<boost::mt19937&, boost::uniform_int<> > randomSleep(gen, boost::uniform_int<>(1, threadSleep));
int skew = 0;
bool legacy = (takeoverMins > 0);
if (!lock.get()) {
@ -177,7 +191,7 @@ namespace mongo {
skew = -skewRange / 2;
else if(threadId == 1)
skew = skewRange / 2;
else skew = (rand_r(&seed) % skewRange) - (skewRange / 2);
else skew = randomSkew() - (skewRange / 2);
// Skew this thread
jsTimeVirtualThreadSkew( skew );
@ -204,7 +218,7 @@ namespace mongo {
count++;
int before = count;
int sleep = (rand_r(&seed) % threadWait);
int sleep = randomWait();
sleepmillis(sleep);
int after = count;
@ -233,7 +247,7 @@ namespace mongo {
break;
}
sleepmillis(rand_r(&seed) % threadSleep);
sleepmillis(randomSleep());
}
result << "errors" << errors