Port support teams getMongoData.js -> bin/mongodb-sharding.js

[imlucas]

https://jira.mongodb.org/secure/attachment/62189/62189_getMongoData.js

NOTE: most of this is already implemented, but just commented out and needs cleanup: https://github.com/mongodb-js/sharding/blob/master/lib/index.js#L237-L321

// Copied from Mongo Shell
function printShardInfo(){
    var configDB = db.getSisterDB("config");

    printInfo("Shard version",
              'db.getSisterDB("config").getCollection("version").findOne()');

    print("\n** Shards:");
    configDB.shards.find().sort({ _id : 1 }).forEach(
        function(z) { print(tojsononeline(z)); }
    );

    print("\n** Shard databases:");
    configDB.databases.find().sort( { name : 1 } ).forEach(
        function(db) {
            print(tojsononeline(db, "", true));
            if (db.partitioned) {
                configDB.collections.find( { _id : new RegExp( "^" +
                    RegExp.escape(db._id) + "\\." ) } ).
                    sort( { _id : 1 } ).forEach( function( coll ) {
                        if ( coll.dropped === false ){
                            print("    " + coll._id);
                            print("      shard key: " + tojson(coll.key, 0, true));
                            print("      chunks:");

                            var res = configDB.chunks.aggregate(
                                { "$match": { ns: coll._id } },
                                { "$group": { _id: "$shard", nChunks: { "$sum": 1 } } }
                            );
                            // MongoDB 2.6 and above returns a cursor instead of a document
                            res = (res.result ? res.result : res.toArray());

                            var totalChunks = 0;
                            res.forEach( function(z) {
                                totalChunks += z.nChunks;
                                print("        " + z._id + ": " + z.nChunks);
                            } );

                            configDB.chunks.find( { "ns" : coll._id } ).sort( { min : 1 } ).forEach(
                                function(chunk) {
                                    print("        " +
                                        tojson( chunk.min, 0, true) + " -> " +
                                        tojson( chunk.max, 0, true ) +
                                        " on: " + chunk.shard + " " +
                                        ( chunk.jumbo ? "jumbo " : "" )
                                    );
                                }
                            );

                            configDB.tags.find( { ns : coll._id } ).sort( { min : 1 } ).forEach(
                                function(tag) {
                                    print("        tag: " + tag.tag + "  " + tojson( tag.min ) + " -> " + tojson( tag.max ));
                                }
                            );
                        }
                    }
                );
            }
        }
    );
}

function printInfo(message, command, printResult) {
    var result = false;
    printResult = (printResult === undefined ? true : false);
    print("\n** " + message + ":");
    try {
        /* jshint evil:true */
        result = eval(command);
        /* jshint evil:false */
    } catch(err) {
        print("Error running '" + command + "':");
        print(err);
    }
    if (printResult) printjson(result);
    return result;
}

function printServerInfo() {
    printInfo('Server status info', 'db.serverStatus()');
    printInfo('Host info',          'db.hostInfo()');
    printInfo('Command line info',  'db.serverCmdLineOpts()');
}

function printReplicaSetInfo() {
    printInfo('Replica set config', 'db.getSisterDB("local").system.replset.findOne()');
    printInfo('Replica status',     'rs.status()');
    printInfo('Replica info',       'db.getReplicationInfo()');
    printInfo('Replica slave info', 'db.printSlaveReplicationInfo()', false);

}

function printDataInfo(isMongoS) {
    var dbs = printInfo('List of databases', 'db.adminCommand("listDatabases")');

    dbs.databases.forEach(function(mydb) {
        var inDB = "db.getSiblingDB('"+ mydb.name + "')";
        var collections = printInfo("List of collections for database '"+ mydb.name +"'",
                                    inDB + ".getCollectionNames()");

        printInfo('Database stats', inDB + '.stats(1024*1024)');
        collections.forEach(function(col) {
            var inCol = inDB + ".getCollection('"+ col + "')";
            printInfo('Collection stats',   inCol + '.stats(1024*1024)');
            if (isMongoS) {
                printInfo('Shard distribution', inCol + '.getShardDistribution()', false);
            }
            printInfo('Indexes',            inCol + '.getIndexes()');
            printInfo('Sample document',    inCol + '.findOne()');
        });
    });
}

function printShardOrReplicaSetInfo() {
    var isMaster = db.isMaster();
    if (isMaster.secondary) {
        print("\n** Connected to secondary");
        rs.slaveOk();
        printReplicaSetInfo();
    }
    else {
        var shardVer = db.getSisterDB("config").getCollection("version").findOne();
        if (shardVer) {
            print("\n** Connected to mongos");
            printShardInfo();
            return true;
        }
        else {
            print("\n** Connected to primary");
            printReplicaSetInfo();
        }
    }
    return false;
}

printServerInfo();
var isMongoS = printShardOrReplicaSetInfo();
printDataInfo(isMongoS);