summaryrefslogtreecommitdiff
path: root/db/dbhelpers.cpp
blob: 21ceb981bcc6ae8fb1c4b19be46d7ceec8a82ed4 (plain)
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
// dbhelpers.cpp

/**
*    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/>.
*/

#include "stdafx.h"
#include "db.h"
#include "dbhelpers.h"
#include "query.h"

/* Get the first object from a collection.  Generally only useful if the collection
   only ever has a single object -- which is a "singleton collection.

   Returns: true if object exists.
*/
bool getSingleton(const char *ns, BSONObj& result) {
    DBContext context(ns);

    auto_ptr<Cursor> c = DataFileMgr::findAll(ns);
    if ( !c->ok() )
        return false;

    result = c->current();
    return true;
}

void putSingleton(const char *ns, BSONObj obj) {
    DBContext context(ns);
    stringstream ss;
    updateObjects(ns, obj, /*pattern=*/emptyObj, /*upsert=*/true, ss);
}

void emptyCollection(const char *ns) {
    DBContext context(ns);
    deleteObjects(ns, emptyObj, false);
}