opendal/services/swift/
error.rs

1// Licensed to the Apache Software Foundation (ASF) under one
2// or more contributor license agreements.  See the NOTICE file
3// distributed with this work for additional information
4// regarding copyright ownership.  The ASF licenses this file
5// to you under the Apache License, Version 2.0 (the
6// "License"); you may not use this file except in compliance
7// with the License.  You may obtain a copy of the License at
8//
9//   http://www.apache.org/licenses/LICENSE-2.0
10//
11// Unless required by applicable law or agreed to in writing,
12// software distributed under the License is distributed on an
13// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14// KIND, either express or implied.  See the License for the
15// specific language governing permissions and limitations
16// under the License.
17
18use bytes::Buf;
19use bytes::Bytes;
20use http::Response;
21use http::StatusCode;
22use quick_xml::de;
23use serde::Deserialize;
24
25use crate::raw::*;
26use crate::*;
27
28#[allow(dead_code)]
29#[derive(Debug, Deserialize)]
30struct ErrorResponse {
31    h1: String,
32    p: String,
33}
34
35pub(super) fn parse_error(resp: Response<Buffer>) -> Error {
36    let (parts, body) = resp.into_parts();
37    let bs = body.to_bytes();
38
39    let (kind, retryable) = match parts.status {
40        StatusCode::NOT_FOUND => (ErrorKind::NotFound, false),
41        StatusCode::UNAUTHORIZED | StatusCode::FORBIDDEN => (ErrorKind::PermissionDenied, false),
42        StatusCode::PRECONDITION_FAILED => (ErrorKind::ConditionNotMatch, false),
43        StatusCode::INTERNAL_SERVER_ERROR
44        | StatusCode::BAD_GATEWAY
45        | StatusCode::SERVICE_UNAVAILABLE
46        | StatusCode::GATEWAY_TIMEOUT => (ErrorKind::Unexpected, true),
47        _ => (ErrorKind::Unexpected, false),
48    };
49
50    let message = parse_error_response(&bs);
51
52    let mut err = Error::new(kind, message);
53
54    err = with_error_response_context(err, parts);
55
56    if retryable {
57        err = err.set_temporary();
58    }
59
60    err
61}
62
63fn parse_error_response(resp: &Bytes) -> String {
64    match de::from_reader::<_, ErrorResponse>(resp.clone().reader()) {
65        Ok(swift_err) => swift_err.p,
66        Err(_) => String::from_utf8_lossy(resp).into_owned(),
67    }
68}
69
70#[cfg(test)]
71mod tests {
72    use super::*;
73
74    #[test]
75    fn parse_error_response_test() -> Result<()> {
76        let resp = Bytes::from(
77            r#"
78<html>
79<h1>Not Found</h1>
80<p>The resource could not be found.</p>
81
82</html>
83            "#,
84        );
85
86        let msg = parse_error_response(&resp);
87        assert_eq!(msg, "The resource could not be found.".to_string(),);
88        Ok(())
89    }
90}
opendal/services/swift/error.rs

opendal/services/swift/
error.rs
