// Basic CRUD client to provide load for Powercycle testing.

function randString(maxLength) {
    maxLength = maxLength || 1024;
    const randChars = "ABCDEFGHIKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
    let rString = "";
    for (let i = 0; i < Random.randInt(maxLength); i++) {
        rString += randChars.charAt(Random.randInt(randChars.length));
    }
    return rString;
}

function weightedChoice(choices) {
    let total = 0;
    for (var choice in choices) {
        total += choices[choice];
    }
    let ran = Random.randInt(total);
    let upto = 0;
    for (choice in choices) {
        let weight = choices[choice];
        if (upto + weight >= ran) {
            return choice;
        }
        upto += weight;
    }
}

let operations = {
    "bulk insert": 15,
    "count": 20,
    "find": 15,
    "remove multi": 15,
    "remove one": 15,
    "upsert one": 15,
    "upsert multi": 5,
};

Random.setRandomSeed();

if (typeof TestData === "undefined") {
    TestData = {};
}
let dbName = TestData.dbName || "test";
let collectionName = TestData.collectionName || "crud";
let bulkNum = TestData.bulkNum || 1000;
let baseNum = TestData.baseNum || 100000;
// Set numLoops <= 0 to have an infinite loop.
let numLoops = TestData.numLoops || 0;

print("****Starting CRUD client, namespace", dbName, collectionName, "numLoops", numLoops, "****");

let coll = db.getSiblingDB(dbName)[collectionName];
coll.createIndex({x: 1});

let shouldLoopForever = numLoops <= 0;
while (shouldLoopForever || numLoops > 0) {
    if (!shouldLoopForever) {
        numLoops -= 1;
    }

    let info = db.hostInfo();
    let serverStatus = db.serverStatus();
    print("(" + collectionName + ") dbHostInfo status:", info.ok, serverStatus.version, "uptime:", serverStatus.uptime);
    let match = Random.randInt(baseNum);
    let matchQuery = {$gte: match, $lt: match + baseNum * 0.01};

    let operation = weightedChoice(operations);

    if (operation == "upsert multi") {
        var updateOpts = {upsert: true, multi: true};
        print(
            "(" + collectionName + ") Upsert multi docs",
            tojsononeline(matchQuery),
            tojsononeline(updateOpts),
            tojsononeline(coll.update({x: matchQuery}, {$inc: {x: baseNum}, $set: {n: "hello"}}, updateOpts)),
        );
    } else if (operation == "upsert one") {
        var updateOpts = {upsert: true, multi: false};
        print(
            "(" + collectionName + ") Upsert single doc",
            match,
            tojsononeline(updateOpts),
            tojsononeline(coll.update({x: match}, {$inc: {x: baseNum}, $set: {n: "hello"}}, updateOpts)),
        );
    } else if (operation == "bulk insert") {
        let bulk = coll.initializeUnorderedBulkOp();
        for (let i = 0; i < bulkNum; i++) {
            bulk.insert({x: (match + i) % baseNum, doc: randString()});
        }
        print("(" + collectionName + ") Bulk insert", bulkNum, "docs", tojsononeline(bulk.execute()));
    } else if (operation == "count") {
        let countOpts = {count: collectionName, query: {x: matchQuery}};
        print(
            "(" + collectionName + ") Count docs",
            tojsononeline(matchQuery),
            tojsononeline(db.runCommand(countOpts)),
        );
    } else if (operation == "find") {
        let findOpts = {find: collectionName, singleBatch: true, filter: {x: matchQuery}};
        print(
            "(" + collectionName + ") Find docs",
            tojsononeline(matchQuery),
            "find status",
            db.runCommand(findOpts).ok,
        );
    } else if (operation == "remove multi") {
        var removeOpts = {};
        var removeQuery = {x: matchQuery, justOne: false};
        print(
            "(" + collectionName + ") Remove docs",
            tojsononeline(removeQuery),
            tojsononeline(removeOpts),
            tojsononeline(coll.remove(removeQuery, removeOpts)),
        );
    } else if (operation == "remove one") {
        var removeOpts = {};
        var removeQuery = {x: match, justOne: true};
        print(
            "(" + collectionName + ") Remove docs",
            tojsononeline(removeQuery),
            tojsononeline(removeOpts),
            tojsononeline(coll.remove(removeQuery, removeOpts)),
        );
    }
}