opendal/services/alluxio/
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 http::Response;
20use serde::Deserialize;
21
22use crate::raw::*;
23use crate::*;
24
25/// the error response of alluxio
26#[derive(Default, Debug, Deserialize)]
27#[serde(rename_all = "camelCase")]
28#[allow(dead_code)]
29struct AlluxioError {
30    status_code: String,
31    message: String,
32}
33
34pub(super) fn parse_error(resp: Response<Buffer>) -> Error {
35    let (parts, body) = resp.into_parts();
36    let bs = body.to_bytes();
37
38    let mut kind = match parts.status.as_u16() {
39        500 => ErrorKind::Unexpected,
40        _ => ErrorKind::Unexpected,
41    };
42
43    let (message, alluxio_err) = serde_json::from_reader::<_, AlluxioError>(bs.clone().reader())
44        .map(|alluxio_err| (format!("{alluxio_err:?}"), Some(alluxio_err)))
45        .unwrap_or_else(|_| (String::from_utf8_lossy(&bs).into_owned(), None));
46
47    if let Some(alluxio_err) = alluxio_err {
48        kind = match alluxio_err.status_code.as_str() {
49            "ALREADY_EXISTS" => ErrorKind::AlreadyExists,
50            "NOT_FOUND" => ErrorKind::NotFound,
51            _ => ErrorKind::Unexpected,
52        }
53    }
54
55    let mut err = Error::new(kind, message);
56
57    err = with_error_response_context(err, parts);
58
59    err
60}
61
62#[cfg(test)]
63mod tests {
64    use http::StatusCode;
65
66    use super::*;
67
68    /// Error response example is from https://docs.aws.amazon.com/AmazonS3/latest/API/ErrorResponses.html
69    #[test]
70    fn test_parse_error() {
71        let err_res = vec![
72            (
73                r#"{"statusCode":"ALREADY_EXISTS","message":"The resource you requested already exist"}"#,
74                ErrorKind::AlreadyExists,
75            ),
76            (
77                r#"{"statusCode":"NOT_FOUND","message":"The resource you requested does not exist"}"#,
78                ErrorKind::NotFound,
79            ),
80            (
81                r#"{"statusCode":"INTERNAL_SERVER_ERROR","message":"Internal server error"}"#,
82                ErrorKind::Unexpected,
83            ),
84        ];
85
86        for res in err_res {
87            let bs = bytes::Bytes::from(res.0);
88            let body = Buffer::from(bs);
89            let resp = Response::builder()
90                .status(StatusCode::INTERNAL_SERVER_ERROR)
91                .body(body)
92                .unwrap();
93
94            let err = parse_error(resp);
95
96            assert_eq!(err.kind(), res.1);
97        }
98    }
99}
opendal/services/alluxio/error.rs

opendal/services/alluxio/
error.rs
