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
|
/**
* Copyright (C) 2012 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/>.
*
* As a special exception, the copyright holders give permission to link the
* code of portions of this program with the OpenSSL library under certain
* conditions as described in each individual source file and distribute
* linked combinations including the program with the OpenSSL library. You
* must comply with the GNU Affero General Public License in all respects for
* all of the code used other than as permitted herein. If you modify file(s)
* with this exception, you may extend this exception to your version of the
* file(s), but you are not obligated to do so. If you do not wish to do so,
* delete this exception statement from your version. If you delete this
* exception statement from all source files in the program, then also delete
* it in the license file.
*/
#include "mongo/db/index_rebuilder.h"
#include "mongo/db/auth/authorization_session.h"
#include "mongo/db/auth/user_name.h"
#include "mongo/db/client.h"
#include "mongo/db/instance.h"
#include "mongo/db/pdfile.h"
#include "mongo/db/repl/rs.h"
#include "mongo/util/scopeguard.h"
namespace mongo {
IndexRebuilder indexRebuilder;
IndexRebuilder::IndexRebuilder() {}
std::string IndexRebuilder::name() const {
return "IndexRebuilder";
}
void IndexRebuilder::run() {
Client::initThread(name().c_str());
ON_BLOCK_EXIT_OBJ(cc(), &Client::shutdown);
cc().getAuthorizationSession()->grantInternalAuthorization();
std::vector<std::string> dbNames;
getDatabaseNames(dbNames);
std::vector<std::string> nsToCheck;
try {
for (std::vector<std::string>::const_iterator it = dbNames.begin();
it < dbNames.end();
it++) {
const std::string systemNS = *it + ".system.namespaces";
DBDirectClient cli;
scoped_ptr<DBClientCursor> cursor(cli.query(systemNS, Query()));
// This depends on system.namespaces not changing while we iterate
while (cursor->more()) {
BSONObj nsDoc = cursor->nextSafe();
nsToCheck.push_back(nsDoc["name"].valuestrsafe());
}
}
{
boost::unique_lock<boost::mutex> lk(ReplSet::rss.mtx);
ReplSet::rss.indexRebuildDone = true;
ReplSet::rss.cond.notify_all();
}
checkNS(nsToCheck);
}
catch (const DBException&) {
warning() << "index rebuilding did not complete" << endl;
{
boost::unique_lock<boost::mutex> lk(ReplSet::rss.mtx);
ReplSet::rss.indexRebuildDone = true;
ReplSet::rss.cond.notify_all();
}
}
LOG(1) << "checking complete" << endl;
}
void IndexRebuilder::checkNS(const std::vector<std::string>& nsToCheck) {
bool firstTime = true;
for (std::vector<std::string>::const_iterator it = nsToCheck.begin();
it != nsToCheck.end();
++it) {
// This write lock is held throughout the index building process
// for this namespace.
Client::WriteContext ctx(*it);
NamespaceDetails* nsd = nsdetails(*it);
if ( nsd == NULL || nsd->getIndexBuildsInProgress() == 0 ) {
continue;
}
log() << "found interrupted index build(s) on " << *it << endl;
if (firstTime) {
log() << "note: restart the server with --noIndexBuildRetry to skip index rebuilds"
<< endl;
firstTime = false;
}
// If the indexBuildRetry flag isn't set, just clear the inProg flag
if (!cmdLine.indexBuildRetry) {
// If we crash between unsetting the inProg flag and cleaning up the index, the
// index space will be lost.
nsd->blowAwayInProgressIndexEntries();
continue;
}
// We go from right to left building these indexes, so that indexBuildInProgress-- has
// the correct effect of "popping" an index off the list.
std::string dbName = it->substr(0, it->find('.'));
while ( nsd->getTotalIndexCount() > nsd->getCompletedIndexCount() ) {
retryIndexBuild(dbName, nsd);
}
}
}
void IndexRebuilder::retryIndexBuild(const std::string& dbName,
NamespaceDetails* nsd ) {
// First, clean up the in progress index build. Save the system.indexes entry so that we
// can add it again afterwards.
BSONObj indexObj = nsd->prepOneUnfinishedIndex();
// The index has now been removed from system.indexes, so the only record of it is in-
// memory. If there is a journal commit between now and when insert() rewrites the entry and
// the db crashes before the new system.indexes entry is journalled, the index will be lost
// forever. Thus, we're assuming no journaling will happen between now and the entry being
// re-written.
try {
const std::string ns = dbName + ".system.indexes";
theDataFileMgr.insert(ns.c_str(), indexObj.objdata(), indexObj.objsize(), false, true);
}
catch (const DBException& e) {
log() << "building index failed: " << e.what() << " (" << e.getCode() << ")"
<< endl;
}
}
}
|
