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
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
|
/**
* Copyright (C) 2018-present MongoDB, Inc.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the Server Side Public License, version 1,
* as published by MongoDB, Inc.
*
* 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
* Server Side Public License for more details.
*
* You should have received a copy of the Server Side Public License
* along with this program. If not, see
* <http://www.mongodb.com/licensing/server-side-public-license>.
*
* 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 Server Side 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.
*/
#pragma once
#include "mongo/base/string_data.h"
#include "mongo/db/jsobj.h"
#include "mongo/db/logical_session_id.h"
#include "mongo/db/ops/write_ops.h"
#include "mongo/db/repl/optime.h"
#include "mongo/rpc/write_concern_error_detail.h"
#include "mongo/s/write_ops/batched_upsert_detail.h"
namespace mongo {
/**
* This class represents the layout and content of a insert/update/delete runCommand,
* the response side.
*/
class BatchedCommandResponse {
BatchedCommandResponse(const BatchedCommandResponse&) = delete;
BatchedCommandResponse& operator=(const BatchedCommandResponse&) = delete;
public:
static const BSONField<long long> n;
static const BSONField<long long> nModified;
static const BSONField<std::vector<BatchedUpsertDetail*>> upsertDetails;
static const BSONField<OID> electionId;
static const BSONField<WriteConcernErrorDetail*> writeConcernError;
static const BSONField<std::vector<std::string>> errorLabels;
static const BSONField<std::vector<StmtId>> retriedStmtIds;
BatchedCommandResponse();
~BatchedCommandResponse();
BatchedCommandResponse(BatchedCommandResponse&&) = default;
BatchedCommandResponse& operator=(BatchedCommandResponse&&) = default;
BSONObj toBSON() const;
bool parseBSON(const BSONObj& source, std::string* errMsg);
void clear();
//
// individual field accessors
//
/**
* This group of getters/setters is only for the top-level command status. If you want to know
* if all writes succeeded, use toStatus() below which considers all of the ways that writes can
* fail.
*/
void setStatus(Status status);
Status getTopLevelStatus() const {
dassert(_isStatusSet);
return _status;
}
bool getOk() const {
dassert(_isStatusSet);
return _status.isOK();
}
/**
* Converts the specified command response into a status, based on all of its contents.
*/
Status toStatus() const;
void setNModified(long long n);
long long getNModified() const;
void setN(long long n);
long long getN() const;
void setUpsertDetails(const std::vector<BatchedUpsertDetail*>& upsertDetails);
void addToUpsertDetails(BatchedUpsertDetail* upsertDetails);
void unsetUpsertDetails();
bool isUpsertDetailsSet() const;
std::size_t sizeUpsertDetails() const;
const std::vector<BatchedUpsertDetail*>& getUpsertDetails() const;
const BatchedUpsertDetail* getUpsertDetailsAt(std::size_t pos) const;
void setLastOp(repl::OpTime lastOp);
bool isLastOpSet() const;
repl::OpTime getLastOp() const;
void setElectionId(const OID& electionId);
bool isElectionIdSet() const;
OID getElectionId() const;
// errDetails ownership is transferred to here.
void addToErrDetails(write_ops::WriteError error);
void unsetErrDetails();
bool isErrDetailsSet() const;
std::size_t sizeErrDetails() const;
std::vector<write_ops::WriteError>& getErrDetails();
const std::vector<write_ops::WriteError>& getErrDetails() const;
const write_ops::WriteError& getErrDetailsAt(std::size_t pos) const;
void setWriteConcernError(WriteConcernErrorDetail* error);
bool isWriteConcernErrorSet() const;
const WriteConcernErrorDetail* getWriteConcernError() const;
bool isErrorLabelsSet() const;
const std::vector<std::string>& getErrorLabels() const;
bool areRetriedStmtIdsSet() const;
const std::vector<StmtId>& getRetriedStmtIds() const;
void setRetriedStmtIds(std::vector<StmtId> retriedStmtIds);
private:
// Convention: (M)andatory, (O)ptional
// (M) The top-level command status.
Status _status = Status::OK();
bool _isStatusSet;
// (M) number of documents affected
long long _n;
bool _isNSet;
// (O) number of documents updated
long long _nModified;
bool _isNModifiedSet;
// (O) Array of upserted items' _id's
// Should only be present if _singleUpserted is not.
std::unique_ptr<std::vector<BatchedUpsertDetail*>> _upsertDetails;
// (O) repl::OpTime assigned to the write op when it was written to the oplog.
// Normally, getLastError can use Client::_lastOp, but this is not valid for
// mongos which loses track of the session due to RCAR. Therefore, we must
// keep track of the lastOp manually ourselves.
repl::OpTime _lastOp;
bool _isLastOpSet;
// (O) In addition to keeping track of the above lastOp repl::OpTime, we must also keep
// track of the primary we talked to. This is because if the primary moves,
// subsequent calls to getLastError are invalid. The only way we know if an
// election has occurred is to use the unique electionId.
OID _electionId;
bool _isElectionIdSet;
// (O) Array of item-level error information
boost::optional<std::vector<write_ops::WriteError>> _writeErrors;
// (O) errors that occurred while trying to satisfy the write concern.
std::unique_ptr<WriteConcernErrorDetail> _wcErrDetails;
// (O) array containing the error labels in string format.
std::vector<std::string> _errorLabels;
// (O) Array containing the retried statement ids from the response.
std::vector<StmtId> _retriedStmtIds;
};
/**
* Error, which is very specific to the batch write commands execution and should never be used
* internally between the cluster nodes. Indicates that more than one type of error occurred while
* executing a batch write command and contains the details for each type.
*/
class MultipleErrorsOccurredInfo final : public ErrorExtraInfo {
public:
static constexpr auto code = ErrorCodes::MultipleErrorsOccurred;
MultipleErrorsOccurredInfo(BSONArray arr) : _arr(std::move(arr)) {}
static std::shared_ptr<const ErrorExtraInfo> parse(const BSONObj& obj);
void serialize(BSONObjBuilder* bob) const;
private:
BSONArray _arr;
};
} // namespace mongo
|
