/**
* Copyright (C) 2013 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 .
*
* 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/s/write_ops/batched_upsert_detail.h"
#include "mongo/db/field_parser.h"
#include "mongo/util/mongoutils/str.h"
namespace mongo {
using std::string;
using mongoutils::str::stream;
const BSONField BatchedUpsertDetail::index("index");
const BSONField BatchedUpsertDetail::upsertedID("_id");
BatchedUpsertDetail::BatchedUpsertDetail() {
clear();
}
BatchedUpsertDetail::~BatchedUpsertDetail() {
}
bool BatchedUpsertDetail::isValid(std::string* errMsg) const {
std::string dummy;
if (errMsg == NULL) {
errMsg = &dummy;
}
// All the mandatory fields must be present.
if (!_isIndexSet) {
*errMsg = stream() << "missing " << index.name() << " field";
return false;
}
if (!_isUpsertedIDSet) {
*errMsg = stream() << "missing " << upsertedID.name() << " field";
return false;
}
return true;
}
BSONObj BatchedUpsertDetail::toBSON() const {
BSONObjBuilder builder;
if (_isIndexSet) builder.append(index(), _index);
// We're using the BSONObj to store the _id value.
if (_isUpsertedIDSet) {
builder.appendAs(_upsertedID.firstElement(), upsertedID());
}
return builder.obj();
}
bool BatchedUpsertDetail::parseBSON(const BSONObj& source, string* errMsg) {
clear();
std::string dummy;
if (!errMsg) errMsg = &dummy;
FieldParser::FieldState fieldState;
fieldState = FieldParser::extract(source, index, &_index, errMsg);
if (fieldState == FieldParser::FIELD_INVALID) return false;
_isIndexSet = fieldState == FieldParser::FIELD_SET;
fieldState = FieldParser::extractID(source, upsertedID, &_upsertedID, errMsg);
if (fieldState == FieldParser::FIELD_INVALID) return false;
_isUpsertedIDSet = fieldState == FieldParser::FIELD_SET;
return true;
}
void BatchedUpsertDetail::clear() {
_index = 0;
_isIndexSet = false;
_upsertedID = BSONObj();
_isUpsertedIDSet = false;
}
void BatchedUpsertDetail::cloneTo(BatchedUpsertDetail* other) const {
other->clear();
other->_index = _index;
other->_isIndexSet = _isIndexSet;
other->_upsertedID = _upsertedID;
other->_isUpsertedIDSet = _isUpsertedIDSet;
}
std::string BatchedUpsertDetail::toString() const {
return "implement me";
}
void BatchedUpsertDetail::setIndex(int index) {
_index = index;
_isIndexSet = true;
}
void BatchedUpsertDetail::unsetIndex() {
_isIndexSet = false;
}
bool BatchedUpsertDetail::isIndexSet() const {
return _isIndexSet;
}
int BatchedUpsertDetail::getIndex() const {
dassert(_isIndexSet);
return _index;
}
void BatchedUpsertDetail::setUpsertedID(const BSONObj& upsertedID) {
_upsertedID = upsertedID.firstElement().wrap( "" ).getOwned();
_isUpsertedIDSet = true;
}
void BatchedUpsertDetail::unsetUpsertedID() {
_isUpsertedIDSet = false;
}
bool BatchedUpsertDetail::isUpsertedIDSet() const {
return _isUpsertedIDSet;
}
const BSONObj& BatchedUpsertDetail::getUpsertedID() const {
dassert(_isUpsertedIDSet);
return _upsertedID;
}
} // namespace mongo