-
Notifications
You must be signed in to change notification settings - Fork 5.6k
/
dbhelpers.h
89 lines (69 loc) · 2.85 KB
/
dbhelpers.h
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
// dbhelpers.h
/**
* 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/>.
*/
/* db helpers are helper functions and classes that let us easily manipulate the local
database instance.
*/
#pragma once
namespace mongo {
struct Helpers {
/* ensure the specified index exists.
@param keyPattern key pattern, e.g., { ts : 1 }
@param name index name, e.g., "name_1"
This method can be a little (not much) cpu-slow, so you may wish to use
OCCASIONALLY ensureIndex(...);
Note: use ensureHaveIdIndex() for the _id index: it is faster.
Note: does nothing if collection does not yet exist.
*/
static void ensureIndex(const char *ns, BSONObj keyPattern, const char *name);
/* fetch a single object from collection ns that matches query.
set your db context first.
@param requireIndex if true, complain if no index for the query. a way to guard against
writing a slow query.
@return true if object found
*/
static bool findOne(const char *ns, BSONObj query, BSONObj& result, bool requireIndex = false);
/* Get/put the first object from a collection. Generally only useful if the collection
only ever has a single object -- which is a "singleton collection".
You do not need to set the database before calling.
Returns: true if object exists.
*/
static bool getSingleton(const char *ns, BSONObj& result);
static void putSingleton(const char *ns, BSONObj obj);
/* Remove all objects from a collection.
You do not need to set the database before calling.
*/
static void emptyCollection(const char *ns);
};
/* Set database we want to use, then, restores when we finish (are out of scope)
Note this is also helpful if an exception happens as the state if fixed up.
*/
class DBContext {
Database *old;
public:
DBContext(const char *ns) {
old = database;
setClientTempNs(ns);
}
DBContext(string ns) {
old = database;
setClientTempNs(ns.c_str());
}
~DBContext() {
database = old;
}
};
} // namespace mongo