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
|
/**
* 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/>.
*/
#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;
}
}
}
|
