mongotype.cpp

Go to the documentation of this file.

//----------------------------------------------------------------------------
#include <mongotype.hpp>
#include <Parameters.hpp>
#include <BSONTypeMap.hpp>
#include <BSONObjectTypeDump.hpp>
#include <BSONDotNotationDump.hpp>
#include <JSONDump.hpp>

//----------------------------------------------------------------------------

namespace mongotype {

//----------------------------------------------------------------------------

const string VERSION("2.4.1");
const string COPYRIGHT("Copyright (c) 2013 by Mark Deazley");
const string LICENSE(
                "Free Software Foundation’s GNU AGPL v3.0.\n"
                                "This program is free software: you can redistribute it and/or modify\n"
                                "it under the terms of the GNU Affero General Public License as\n"
                                "published by the Free Software Foundation, either version 3 of the\n"
                                "License, or (at your option) any later version.\n"
                                "\n"
                                "This program is distributed in the hope that it will be useful,\n"
                                "but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
                                "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\n"
                                "GNU Affero General Public License for more details.\n"
                                "\n"
                                "You should have received a copy of the GNU Affero General Public License\n"
                                "along with this program.  If not, see http://www.gnu.org/licenses/ .\n");

//----------------------------------------------------------------------------

void dumpCollection(Parameters& params) {
        DBClientConnection c;
        string hostPort(params.getHost());
        if (hostPort.find(':') == string::npos) {
                hostPort += ":";
                hostPort += to_string(params.getPort());
        }
        c.connect(hostPort);
        time_t t;
        time(&t);

        int documentCount = c.count(params.getDbCollection());

        if (params.isDebug()) {
                cout << "{ " << params.getDbCollection() << ".count: "
                        << documentCount << " }\n";
        }

        string docPrefixString(params.getDbCollection());
//      docIndex += "{";
//      docIndex += to_string(i++);
//      docIndex += "}";

        unique_ptr<DBClientCursor> cursor = c.query(params.getDbCollection(), BSONObj());

        unique_ptr<IBSONRenderer> renderer;
        switch (params.getStyle()) {
        case STYLE_DOTTED:
                renderer = unique_ptr<IBSONRenderer>(new BSONDotNotationDump(params, docPrefixString));
                break;
        case STYLE_TREE:
                renderer = unique_ptr<IBSONRenderer>(new BSONObjectTypeDump(params, docPrefixString));
                break;
        case STYLE_JSON:
        case STYLE_JSONPACKED:
                renderer = unique_ptr<IBSONRenderer>(new JSONDump(params, "  "));
                break;
        default:
                throw std::logic_error("ISE: Undefined STYLE!");
                break;
        }
        if (renderer) {
                renderer->setOutputStream(cout);
                renderer->begin(NULL);
                int documentIndex = 0;
                while (cursor->more()) {
                        const BSONObj& o = cursor->next(); // Get the BSON Object
                        renderer->render(o, documentIndex++, documentCount);
                }
                renderer->end(NULL);
        } else {
                throw std::logic_error("ISE: Undefined renderer!");
        }
}

//----------------------------------------------------------------------------

} // End of namespace mongotype

//----------------------------------------------------------------------------

int main(int argc, char* argv[]) {
        try {
                mongotype::Parameters params;
                params.parse(argc, argv);
                mongotype::dumpCollection(params);
        } catch (const mongo::DBException &e) {
                cerr << "mongotype MongoDB Error: \"" << e.what() << "\"" << endl;
                exit(2);
        } catch (std::logic_error &e) {
                cerr << "mongotype Generic Error: \"" << e.what() << "\"" << endl;
                exit(2);
        }
        return EXIT_SUCCESS;
}