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
|
// instance.h : Global state functions.
//
/**
* Copyright (C) 2008 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/>.
*/
#pragma once
// turn on or off the oplog.* files which the db can generate.
// these files are for diagnostic purposes and are unrelated to
// local.oplog.$main used by replication.
//
#define OPLOG if( 0 )
int getOpLogging();
#define OPWRITE if( getOpLogging() & 1 ) _oplog.write((char *) m.data, m.data->len);
#define OPREAD if( getOpLogging() & 2 ) _oplog.readop((char *) m.data, m.data->len);
struct OpLog {
ofstream *f;
OpLog() : f(0) { }
void init() {
OPLOG {
stringstream ss;
ss << "oplog." << hex << time(0);
string name = ss.str();
f = new ofstream(name.c_str(), ios::out | ios::binary);
if ( ! f->good() ) {
problem() << "couldn't open log stream" << endl;
throw 1717;
}
}
}
void flush() {
OPLOG f->flush();
}
void write(char *data,int len) {
OPLOG f->write(data,len);
}
void readop(char *data, int len) {
OPLOG {
bool log = (getOpLogging() & 4) == 0;
OCCASIONALLY log = true;
if ( log )
f->write(data,len);
}
}
};
/* we defer response until we unlock. don't want a blocked socket to
keep things locked.
*/
struct DbResponse {
Message *response;
MSGID responseTo;
DbResponse(Message *r, MSGID rt) : response(r), responseTo(rt) {
}
DbResponse() {
response = 0;
}
~DbResponse() {
delete response;
}
};
bool assembleResponse( Message &m, DbResponse &dbresponse );
void receivedKillCursors(Message& m);
void receivedUpdate(Message& m, stringstream& ss);
void receivedDelete(Message& m);
void receivedInsert(Message& m, stringstream& ss);
void receivedGetMore(DbResponse& dbresponse, /*AbstractMessagingPort& dbMsgPort, */Message& m, stringstream& ss);
void receivedQuery(DbResponse& dbresponse, /*AbstractMessagingPort& dbMsgPort, */Message& m, stringstream& ss, bool logit);
void getDatabaseNames( vector< string > &names );
// --- local client ---
class DBDirectClient : public DBClientBase {
virtual string toString() { return "DBDirectClient"; }
virtual bool call( Message &toSend, Message &response, bool assertOk=true );
virtual void say( Message &toSend );
virtual void sayPiggyBack( Message &toSend ){
// don't need to piggy back when connected locally
return say( toSend );
}
};
|
