1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
/*
* Copyright (C) 2010 10gen Inc.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License, version 3,
* as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
// Tool.h
#pragma once
#include <string>
#include <boost/program_options.hpp>
#if defined(_WIN32)
#include <io.h>
#endif
#include "client/dbclient.h"
#include "db/instance.h"
#include "db/matcher.h"
using std::string;
namespace mongo {
class Tool {
public:
enum DBAccess {
NONE = 0 ,
REMOTE_SERVER = 1 << 1 ,
LOCAL_SERVER = 1 << 2 ,
SPECIFY_DBCOL = 1 << 3 ,
ALL = REMOTE_SERVER | LOCAL_SERVER | SPECIFY_DBCOL
};
Tool( string name , DBAccess access=ALL, string defaultDB="test" ,
string defaultCollection="", bool usesstdout=true);
virtual ~Tool();
int main( int argc , char ** argv );
boost::program_options::options_description_easy_init add_options() {
return _options->add_options();
}
boost::program_options::options_description_easy_init add_hidden_options() {
return _hidden_options->add_options();
}
void addPositionArg( const char * name , int pos ) {
_positonalOptions.add( name , pos );
}
string getParam( string name , string def="" ) {
if ( _params.count( name ) )
return _params[name.c_str()].as<string>();
return def;
}
int getParam( string name , int def ) {
if ( _params.count( name ) )
return _params[name.c_str()].as<int>();
return def;
}
bool hasParam( string name ) {
return _params.count( name );
}
string getNS() {
if ( _coll.size() == 0 ) {
cerr << "no collection specified!" << endl;
throw -1;
}
return _db + "." + _coll;
}
void useStandardOutput( bool mode ) {
_usesstdout = mode;
}
bool isMaster();
bool isMongos();
virtual void preSetup() {}
virtual int run() = 0;
virtual void printHelp(ostream &out);
virtual void printExtraHelp( ostream & out ) {}
virtual void printExtraHelpAfter( ostream & out ) {}
virtual void printVersion(ostream &out);
protected:
mongo::DBClientBase &conn( bool slaveIfPaired = false );
void auth( string db = "" );
string _name;
string _db;
string _coll;
string _fileName;
string _username;
string _password;
bool _usesstdout;
bool _noconnection;
bool _autoreconnect;
void addFieldOptions();
void needFields();
vector<string> _fields;
BSONObj _fieldsObj;
string _host;
protected:
mongo::DBClientBase * _conn;
mongo::DBClientBase * _slaveConn;
bool _paired;
boost::program_options::options_description * _options;
boost::program_options::options_description * _hidden_options;
boost::program_options::positional_options_description _positonalOptions;
boost::program_options::variables_map _params;
};
class BSONTool : public Tool {
bool _objcheck;
auto_ptr<Matcher> _matcher;
public:
BSONTool( const char * name , DBAccess access=ALL, bool objcheck = false );
virtual int doRun() = 0;
virtual void gotObject( const BSONObj& obj ) = 0;
virtual int run();
long long processFile( const path& file );
};
}
|
