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
|
// rs.cpp
/* Copyright 2009 10gen Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
/**
* example of using replica sets from c++
*/
#include "client/dbclient.h"
#include <iostream>
#include <vector>
using namespace mongo;
using namespace std;
void workerThread( string collName , bool print , DBClientReplicaSet * conn ) {
while ( true ) {
try {
conn->update( collName , BSONObj() , BSON( "$inc" << BSON( "x" << 1 ) ) , true );
BSONObj x = conn->findOne( collName , BSONObj() );
if ( print ) {
cout << x << endl;
}
BSONObj a = conn->slaveConn().findOne( collName , BSONObj() , 0 , QueryOption_SlaveOk );
BSONObj b = conn->findOne( collName , BSONObj() , 0 , QueryOption_SlaveOk );
if ( print ) {
cout << "\t A " << a << endl;
cout << "\t B " << b << endl;
}
}
catch ( std::exception& e ) {
cout << "ERROR: " << e.what() << endl;
}
sleepmillis( 10 );
}
}
int main( int argc , const char ** argv ) {
unsigned nThreads = 1;
bool print = false;
bool testTimeout = false;
for ( int i=1; i<argc; i++ ) {
if ( mongoutils::str::equals( "--threads" , argv[i] ) ) {
nThreads = atoi( argv[++i] );
}
else if ( mongoutils::str::equals( "--print" , argv[i] ) ) {
print = true;
}
// Run a special mode to demonstrate the DBClientReplicaSet so_timeout option.
else if ( mongoutils::str::equals( "--testTimeout" , argv[i] ) ) {
testTimeout = true;
}
else {
cerr << "unknown option: " << argv[i] << endl;
return 1;
}
}
string errmsg;
ConnectionString cs = ConnectionString::parse( "foo/127.0.0.1" , errmsg );
if ( ! cs.isValid() ) {
cout << "error parsing url: " << errmsg << endl;
return 1;
}
DBClientReplicaSet * conn = dynamic_cast<DBClientReplicaSet*>(cs.connect( errmsg, testTimeout ? 10 : 0 ));
if ( ! conn ) {
cout << "error connecting: " << errmsg << endl;
return 2;
}
string collName = "test.rs1";
conn->dropCollection( collName );
if ( testTimeout ) {
conn->insert( collName, BSONObj() );
try {
conn->count( collName, BSON( "$where" << "sleep(40000)" ) );
} catch( DBException& ) {
return 0;
}
cout << "expected socket exception" << endl;
return 1;
}
vector<boost::shared_ptr<boost::thread> > threads;
for ( unsigned i=0; i<nThreads; i++ ) {
string errmsg;
threads.push_back( boost::shared_ptr<boost::thread>( new boost::thread( boost::bind( workerThread , collName , print , (DBClientReplicaSet*)cs.connect(errmsg) ) ) ) );
}
for ( unsigned i=0; i<threads.size(); i++ ) {
threads[i]->join();
}
}
|
