Skip to content

Commit 9225dab

Browse files
saghmsaghm
committed
RUST-679 Remove inner Arc from Error
1 parent eff526c commit 9225dab

File tree

29 files changed

+99
-108
lines changed

29 files changed

+99
-108
lines changed

‎src/client/executor.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ impl Client {
108108

109109
// Retryable writes are only supported by storage engines with document-level
110110
// locking, so users need to disable retryable writes if using mmapv1.
111-
if let ErrorKind::CommandError(ref err) = err.kind.as_ref() {
111+
if let ErrorKind::CommandError(ref err) = err.kind {
112112
if err.code == 20 && err.message.starts_with("Transaction numbers") {
113113
let mut err = err.clone();
114114
err.message = "This MongoDB deployment does not support retryable writes. \

‎src/client/options/test.rs

+2-2
Original file line numberDiff line numberDiff line change
@@ -317,7 +317,7 @@ async fn run_test(test_file: TestFile) {
317317
match ClientOptions::parse(&test_case.uri)
318318
.await
319319
.as_ref()
320-
.map_err(|e| e.as_ref())
320+
.map_err(|e| &e.kind)
321321
{
322322
Ok(_) => panic!("expected {}", expected_type),
323323
Err(ErrorKind::ArgumentError { .. }) => {}
@@ -342,7 +342,7 @@ async fn run_connection_string_spec_tests() {
342342
async fn parse_uri(option: &str, suggestion: Option<&str>) {
343343
match ClientOptionsParser::parse(&format!("mongodb://host:27017/?{}=test", option))
344344
.as_ref()
345-
.map_err(|e| e.as_ref())
345+
.map_err(|e| &e.kind)
346346
{
347347
Ok(_) => panic!("expected error for option {}", option),
348348
Err(ErrorKind::ArgumentError { message, .. }) => {

‎src/cmap/mod.rs

+5-6
Original file line numberDiff line numberDiff line change
@@ -156,12 +156,11 @@ impl ConnectionPool {
156156
});
157157
}
158158
Err(ref e) => {
159-
let failure_reason =
160-
if let ErrorKind::WaitQueueTimeoutError { .. } = e.kind.as_ref() {
161-
ConnectionCheckoutFailedReason::Timeout
162-
} else {
163-
ConnectionCheckoutFailedReason::ConnectionError
164-
};
159+
let failure_reason = if let ErrorKind::WaitQueueTimeoutError { .. } = e.kind {
160+
ConnectionCheckoutFailedReason::Timeout
161+
} else {
162+
ConnectionCheckoutFailedReason::ConnectionError
163+
};
165164

166165
self.emit_event(|handler| {
167166
handler.handle_connection_checkout_failed_event(ConnectionCheckoutFailedEvent {

‎src/cmap/test/file.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ pub struct Error {
9797

9898
impl Error {
9999
pub fn assert_matches(&self, error: &crate::error::Error, description: &str) {
100-
match error.kind.as_ref() {
100+
match error.kind {
101101
ErrorKind::WaitQueueTimeoutError { .. } => {
102102
assert_eq!(self.type_, "WaitQueueTimeoutError", "{}", description);
103103
}

‎src/coll/mod.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -437,7 +437,7 @@ where
437437
}
438438
}
439439
}
440-
Err(e) => match e.kind.as_ref() {
440+
Err(e) => match e.kind {
441441
ErrorKind::BulkWriteError(failure) => {
442442
let failure_ref =
443443
cumulative_failure.get_or_insert_with(BulkWriteFailure::new);

‎src/concern/test.rs

+3-12
Original file line numberDiff line numberDiff line change
@@ -107,10 +107,7 @@ async fn inconsistent_write_concern_rejected() {
107107
)
108108
.await
109109
.expect_err("insert should fail");
110-
assert!(matches!(
111-
error.kind.as_ref(),
112-
ErrorKind::ArgumentError { .. }
113-
));
110+
assert!(matches!(error.kind, ErrorKind::ArgumentError { .. }));
114111

115112
let coll = db.collection(function_name!());
116113
let wc = WriteConcern {
@@ -123,10 +120,7 @@ async fn inconsistent_write_concern_rejected() {
123120
.insert_one(doc! {}, options)
124121
.await
125122
.expect_err("insert should fail");
126-
assert!(matches!(
127-
error.kind.as_ref(),
128-
ErrorKind::ArgumentError { .. }
129-
));
123+
assert!(matches!(error.kind, ErrorKind::ArgumentError { .. }));
130124
}
131125

132126
#[cfg_attr(feature = "tokio-runtime", tokio::test)]
@@ -148,8 +142,5 @@ async fn unacknowledged_write_concern_rejected() {
148142
)
149143
.await
150144
.expect_err("insert should fail");
151-
assert!(matches!(
152-
error.kind.as_ref(),
153-
ErrorKind::ArgumentError { .. }
154-
));
145+
assert!(matches!(error.kind, ErrorKind::ArgumentError { .. }));
155146
}

‎src/cursor/common.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ pub(super) trait GetMoreProviderResult {
137137
match self.as_ref() {
138138
Ok(res) => res.exhausted,
139139
Err(e) => {
140-
matches!(e.kind.as_ref(), ErrorKind::CommandError(e) if e.code == 43 || e.code == 237)
140+
matches!(e.kind, ErrorKind::CommandError(ref e) if e.code == 43 || e.code == 237)
141141
}
142142
}
143143
}

‎src/error.rs

+44-29
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ pub type Result<T> = std::result::Result<T, Error>;
2727
#[non_exhaustive]
2828
pub struct Error {
2929
/// The type of error that occurred.
30-
pub kind: Arc<ErrorKind>,
30+
pub kind: ErrorKind,
3131
labels: Vec<String>,
3232
}
3333

@@ -63,7 +63,7 @@ impl Error {
6363

6464
/// Whether this error is an "ns not found" error or not.
6565
pub(crate) fn is_ns_not_found(&self) -> bool {
66-
matches!(self.kind.as_ref(), ErrorKind::CommandError(err) if err.code == 26)
66+
matches!(self.kind, ErrorKind::CommandError(ref err) if err.code == 26)
6767
}
6868

6969
/// Whether a read operation should be retried if this error occurs.
@@ -109,7 +109,7 @@ impl Error {
109109
/// Whether an error originated from the server.
110110
pub(crate) fn is_server_error(&self) -> bool {
111111
matches!(
112-
self.kind.as_ref(),
112+
self.kind,
113113
ErrorKind::AuthenticationError { .. }
114114
| ErrorKind::BulkWriteError(_)
115115
| ErrorKind::CommandError(_)
@@ -119,13 +119,13 @@ impl Error {
119119

120120
/// Returns the labels for this error.
121121
pub fn labels(&self) -> &[String] {
122-
match self.kind.as_ref() {
123-
ErrorKind::CommandError(err) => &err.labels,
124-
ErrorKind::WriteError(err) => match err {
122+
match self.kind {
123+
ErrorKind::CommandError(ref err) => &err.labels,
124+
ErrorKind::WriteError(ref err) => match err {
125125
WriteFailure::WriteError(_) => &self.labels,
126-
WriteFailure::WriteConcernError(err) => &err.labels,
126+
WriteFailure::WriteConcernError(ref err) => &err.labels,
127127
},
128-
ErrorKind::BulkWriteError(err) => match err.write_concern_error {
128+
ErrorKind::BulkWriteError(ref err) => match err.write_concern_error {
129129
Some(ref err) => &err.labels,
130130
None => &self.labels,
131131
},
@@ -143,24 +143,24 @@ impl Error {
143143
/// Returns a copy of this Error with the specified label added.
144144
pub(crate) fn with_label<T: AsRef<str>>(mut self, label: T) -> Self {
145145
let label = label.as_ref().to_string();
146-
match self.kind.as_ref() {
147-
ErrorKind::CommandError(err) => {
146+
match self.kind {
147+
ErrorKind::CommandError(ref err) => {
148148
let mut err = err.clone();
149149
err.labels.push(label);
150150
ErrorKind::CommandError(err).into()
151151
}
152-
ErrorKind::WriteError(err) => match err {
152+
ErrorKind::WriteError(ref err) => match err {
153153
WriteFailure::WriteError(_) => {
154154
self.labels.push(label);
155155
self
156156
}
157-
WriteFailure::WriteConcernError(err) => {
157+
WriteFailure::WriteConcernError(ref err) => {
158158
let mut err = err.clone();
159159
err.labels.push(label);
160160
ErrorKind::WriteError(WriteFailure::WriteConcernError(err)).into()
161161
}
162162
},
163-
ErrorKind::BulkWriteError(err) => match err.write_concern_error {
163+
ErrorKind::BulkWriteError(ref err) => match err.write_concern_error {
164164
Some(ref write_concern_error) => {
165165
let mut err = err.clone();
166166
let mut write_concern_error = write_concern_error.clone();
@@ -187,14 +187,38 @@ where
187187
{
188188
fn from(err: E) -> Self {
189189
Self {
190-
kind: Arc::new(err.into()),
190+
kind: err.into(),
191191
labels: Vec::new(),
192192
}
193193
}
194194
}
195195

196+
impl From<bson::de::Error> for ErrorKind {
197+
fn from(err: bson::de::Error) -> Self {
198+
Self::BsonDecode(Arc::new(err))
199+
}
200+
}
201+
202+
impl From<bson::ser::Error> for ErrorKind {
203+
fn from(err: bson::ser::Error) -> Self {
204+
Self::BsonEncode(Arc::new(err))
205+
}
206+
}
207+
208+
impl From<std::io::Error> for ErrorKind {
209+
fn from(err: std::io::Error) -> Self {
210+
Self::Io(Arc::new(err))
211+
}
212+
}
213+
214+
impl From<std::io::ErrorKind> for ErrorKind {
215+
fn from(err: std::io::ErrorKind) -> Self {
216+
Self::Io(Arc::new(err.into()))
217+
}
218+
}
219+
196220
impl std::ops::Deref for Error {
197-
type Target = Arc<ErrorKind>;
221+
type Target = ErrorKind;
198222

199223
fn deref(&self) -> &Self::Target {
200224
&self.kind
@@ -203,7 +227,7 @@ impl std::ops::Deref for Error {
203227

204228
/// The types of errors that can occur.
205229
#[allow(missing_docs)]
206-
#[derive(Debug, Error)]
230+
#[derive(Clone, Debug, Error)]
207231
#[non_exhaustive]
208232
pub enum ErrorKind {
209233
/// Wrapper around [`std::net::AddrParseError`](https://doc.rust-lang.org/std/net/struct.AddrParseError.html).
@@ -215,10 +239,6 @@ pub enum ErrorKind {
215239
#[non_exhaustive]
216240
ArgumentError { message: String },
217241

218-
#[cfg(feature = "async-std-runtime")]
219-
#[error("{0}")]
220-
AsyncStdTimeout(#[from] async_std::future::TimeoutError),
221-
222242
/// An error occurred while the [`Client`](../struct.Client.html) attempted to authenticate a
223243
/// connection.
224244
#[error("{message}")]
@@ -227,11 +247,11 @@ pub enum ErrorKind {
227247

228248
/// Wrapper around `bson::de::Error`.
229249
#[error("{0}")]
230-
BsonDecode(#[from] crate::bson::de::Error),
250+
BsonDecode(Arc<crate::bson::de::Error>),
231251

232252
/// Wrapper around `bson::ser::Error`.
233253
#[error("{0}")]
234-
BsonEncode(#[from] crate::bson::ser::Error),
254+
BsonEncode(Arc<crate::bson::ser::Error>),
235255

236256
/// An error occurred when trying to execute a write operation consisting of multiple writes.
237257
#[error("An error occurred when trying to execute a write operation: {0:?}")]
@@ -262,7 +282,7 @@ pub enum ErrorKind {
262282

263283
/// Wrapper around [`std::io::Error`](https://doc.rust-lang.org/std/io/struct.Error.html).
264284
#[error("{0}")]
265-
Io(#[from] std::io::Error),
285+
Io(Arc<std::io::Error>),
266286

267287
#[error("No DNS results for domain {0}")]
268288
NoDnsResults(StreamAddress),
@@ -301,11 +321,6 @@ pub enum ErrorKind {
301321
#[non_exhaustive]
302322
SrvLookupError { message: String },
303323

304-
/// A timeout occurred before a Tokio task could be completed.
305-
#[cfg(feature = "tokio-runtime")]
306-
#[error("{0}")]
307-
TokioTimeoutElapsed(#[from] tokio::time::error::Elapsed),
308-
309324
#[error("{0}")]
310325
RustlsConfig(#[from] rustls::TLSError),
311326

@@ -561,7 +576,7 @@ impl WriteFailure {
561576
/// Translates ErrorKind::BulkWriteError cases to ErrorKind::WriteErrors, leaving all other errors
562577
/// untouched.
563578
pub(crate) fn convert_bulk_errors(error: Error) -> Error {
564-
match *error.kind {
579+
match error.kind {
565580
ErrorKind::BulkWriteError(ref bulk_failure) => {
566581
match WriteFailure::from_bulk_failure(bulk_failure.clone()) {
567582
Ok(failure) => ErrorKind::WriteError(failure).into(),

‎src/operation/aggregate/test.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -310,7 +310,7 @@ async fn handle_write_concern_error() {
310310
let error = aggregate
311311
.handle_response(response)
312312
.expect_err("should get wc error");
313-
match *error.kind {
313+
match error.kind {
314314
ErrorKind::WriteError(WriteFailure::WriteConcernError(_)) => {}
315315
ref e => panic!("should have gotten WriteConcernError, got {:?} instead", e),
316316
}

‎src/operation/count/test.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ async fn handle_response_no_n() {
9696
let response = CommandResponse::with_document(doc! { "ok" : 1 });
9797

9898
let result = count_op.handle_response(response);
99-
match result.as_ref().map_err(|e| e.as_ref()) {
99+
match result.as_ref().map_err(|e| &e.kind) {
100100
Err(ErrorKind::ResponseError { .. }) => {}
101101
other => panic!("expected response error, but got {:?}", other),
102102
}

‎src/operation/create/test.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ async fn handle_write_concern_error() {
102102
let result = op.handle_response(response);
103103
assert!(result.is_err());
104104

105-
match *result.unwrap_err().kind {
105+
match result.unwrap_err().kind {
106106
ErrorKind::WriteError(WriteFailure::WriteConcernError(ref wc_err)) => {
107107
assert_eq!(wc_err.code, 100);
108108
assert_eq!(wc_err.code_name, "hello world");

‎src/operation/delete/test.rs

+2-2
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ async fn handle_write_failure() {
143143
});
144144
let write_error_result = op.handle_response(write_error_response);
145145
assert!(write_error_result.is_err());
146-
match *write_error_result.unwrap_err().kind {
146+
match write_error_result.unwrap_err().kind {
147147
ErrorKind::WriteError(WriteFailure::WriteError(ref error)) => {
148148
let expected_err = WriteError {
149149
code: 1234,
@@ -181,7 +181,7 @@ async fn handle_write_concern_failure() {
181181
let wc_error_result = op.handle_response(wc_error_response);
182182
assert!(wc_error_result.is_err());
183183

184-
match *wc_error_result.unwrap_err().kind {
184+
match wc_error_result.unwrap_err().kind {
185185
ErrorKind::WriteError(WriteFailure::WriteConcernError(ref wc_error)) => {
186186
let expected_wc_err = WriteConcernError {
187187
code: 456,

‎src/operation/distinct/test.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ async fn handle_response_no_values() {
144144
});
145145

146146
let result = distinct_op.handle_response(response);
147-
match result.as_ref().map_err(|e| e.as_ref()) {
147+
match result.as_ref().map_err(|e| &e.kind) {
148148
Err(ErrorKind::ResponseError { .. }) => {}
149149
other => panic!("expected response error, but got {:?}", other),
150150
}

‎src/operation/drop_collection/test.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ async fn handle_write_concern_error() {
8080
let result = op.handle_response(response);
8181
assert!(result.is_err());
8282

83-
match *result.unwrap_err().kind {
83+
match result.unwrap_err().kind {
8484
ErrorKind::WriteError(WriteFailure::WriteConcernError(ref wc_err)) => {
8585
assert_eq!(wc_err.code, 100);
8686
assert_eq!(wc_err.code_name, "hello world");

‎src/operation/drop_database/test.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ async fn handle_write_concern_error() {
7878
let result = op.handle_response(response);
7979
assert!(result.is_err());
8080

81-
match *result.unwrap_err().kind {
81+
match result.unwrap_err().kind {
8282
ErrorKind::WriteError(WriteFailure::WriteConcernError(ref wc_err)) => {
8383
assert_eq!(wc_err.code, 100);
8484
assert_eq!(wc_err.code_name, "hello world");

‎src/operation/insert/test.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ async fn handle_write_failure() {
193193
let write_error_result = fixtures.op.handle_response(write_error_response);
194194
assert!(write_error_result.is_err());
195195

196-
match *write_error_result.unwrap_err().kind {
196+
match write_error_result.unwrap_err().kind {
197197
ErrorKind::BulkWriteError(ref error) => {
198198
let write_errors = error.write_errors.clone().unwrap();
199199
assert_eq!(write_errors.len(), 1);

‎src/operation/list_databases/test.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ async fn handle_response_no_databases() {
140140
});
141141

142142
let result = list_databases_op.handle_response(response);
143-
match result.as_ref().map_err(|e| e.as_ref()) {
143+
match result.as_ref().map_err(|e| &e.kind) {
144144
Err(ErrorKind::ResponseError { .. }) => {}
145145
other => panic!("expected response error, but got {:?}", other),
146146
}

‎src/operation/update/test.rs

+2-2
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,7 @@ async fn handle_write_failure() {
227227
});
228228
let write_error_result = op.handle_response(write_error_response);
229229
assert!(write_error_result.is_err());
230-
match *write_error_result.unwrap_err().kind {
230+
match write_error_result.unwrap_err().kind {
231231
ErrorKind::WriteError(WriteFailure::WriteError(ref error)) => {
232232
let expected_err = WriteError {
233233
code: 1234,
@@ -266,7 +266,7 @@ async fn handle_write_concern_failure() {
266266
let wc_error_result = op.handle_response(wc_error_response);
267267
assert!(wc_error_result.is_err());
268268

269-
match *wc_error_result.unwrap_err().kind {
269+
match wc_error_result.unwrap_err().kind {
270270
ErrorKind::WriteError(WriteFailure::WriteConcernError(ref wc_error)) => {
271271
let expected_wc_err = WriteConcernError {
272272
code: 456,

0 commit comments

Comments
 (0)