mirror of https://github.com/mongodb/mongo
81 lines
2.9 KiB
JavaScript
81 lines
2.9 KiB
JavaScript
/**
|
|
* fsm_example.js
|
|
*
|
|
* Includes documentation of each property on $config.
|
|
* Serves as a template for new workloads.
|
|
*/
|
|
export const $config = (function () {
|
|
// 'data' is passed (copied) to each of the worker threads.
|
|
let data = {};
|
|
|
|
// 'states' are the different functions callable by a worker
|
|
// thread. The 'this' argument of any exposed function is
|
|
// bound as '$config.data'.
|
|
let states = {
|
|
init: function init(db, collName) {
|
|
this.start = 10 * this.tid;
|
|
},
|
|
|
|
scanGT: function scanGT(db, collName) {
|
|
db[collName].find({_id: {$gt: this.start}}).itcount();
|
|
},
|
|
|
|
scanLTE: function scanLTE(db, collName) {
|
|
db[collName].find({_id: {$lte: this.start}}).itcount();
|
|
},
|
|
};
|
|
|
|
// 'transitions' defines how the FSM should proceed from its
|
|
// current state to the next state. The value associated with a
|
|
// particular state represents the likelihood of that transition.
|
|
//
|
|
// For example, 'init: { scanGT: 0.5, scanLTE: 0.5 }' means that
|
|
// the worker thread will transition from the 'init' state
|
|
// to the 'scanGT' state with probability 0.5, and
|
|
// to the 'scanLTE' state with probability 0.5.
|
|
//
|
|
// All state functions should appear as keys within 'transitions'.
|
|
let transitions = {
|
|
init: {scanGT: 0.5, scanLTE: 0.5},
|
|
scanGT: {scanGT: 0.8, scanLTE: 0.2},
|
|
scanLTE: {scanGT: 0.2, scanLTE: 0.8},
|
|
};
|
|
|
|
// 'setup' is run once by the parent thread after the cluster has
|
|
// been initialized, but before the worker threads have been spawned.
|
|
// The 'this' argument is bound as '$config.data'. 'cluster' is provided
|
|
// to allow execution against all mongos and mongod nodes.
|
|
function setup(db, collName, cluster) {
|
|
// Workloads should NOT drop the collection db[collName], as
|
|
// doing so is handled by runner.js before 'setup' is called.
|
|
for (let i = 0; i < 1000; ++i) {
|
|
db[collName].insert({_id: i});
|
|
}
|
|
|
|
cluster.executeOnMongodNodes(function (db) {
|
|
printjson(db.serverCmdLineOpts());
|
|
});
|
|
|
|
cluster.executeOnMongosNodes(function (db) {
|
|
printjson(db.serverCmdLineOpts());
|
|
});
|
|
}
|
|
|
|
// 'teardown' is run once by the parent thread before the cluster
|
|
// is destroyed, but after the worker threads have been reaped.
|
|
// The 'this' argument is bound as '$config.data'. 'cluster' is provided
|
|
// to allow execution against all mongos and mongod nodes.
|
|
function teardown(db, collName, cluster) {}
|
|
|
|
return {
|
|
threadCount: 5,
|
|
iterations: 10,
|
|
startState: "init", // optional, default 'init'
|
|
states: states,
|
|
transitions: transitions,
|
|
setup: setup, // optional, default empty function
|
|
teardown: teardown, // optional, default empty function
|
|
data: data, // optional, default empty object
|
|
};
|
|
})();
|