Use roxmltree for parsing

This commit is contained in:
Aditya Manthramurthy 2019-05-30 16:41:03 -07:00
parent 21f4de0540
commit 035fced940
4 changed files with 18 additions and 16 deletions

View File

@ -11,5 +11,5 @@ http = "0.1.17"
hyper = "0.12.28"
hyper-tls = "0.3.2"
ring = "0.14.6"
roxmltree = "0.6.0"
time = "0.1.42"
xml-rs = "0.8.0"

View File

@ -159,7 +159,8 @@ impl Client {
resp.into_body()
.concat2()
.map_err(|err| Err::HyperErr(err))
.and_then(move |chunk| xml::parse_bucket_location(chunk.into_bytes()))
.and_then(move |chunk| b2s(chunk.into_bytes()))
.and_then(|s| xml::parse_bucket_location(s))
})
}
}

View File

@ -1,7 +1,7 @@
use bytes::Bytes;
use hyper::{body::Body, Response};
use roxmltree;
use std::string;
use xml;
pub struct Region(String);
@ -27,7 +27,7 @@ pub enum Err {
HyperErr(hyper::Error),
FailStatusCodeErr(hyper::StatusCode, Bytes),
Utf8DecodingErr(string::FromUtf8Error),
XmlParseErr(xml::reader::Error),
XmlParseErr(roxmltree::Error),
UnexpectedEOF(String),
RawSvcErr(hyper::StatusCode, Response<Body>),
}

View File

@ -1,16 +1,17 @@
use crate::minio::types::{Err, Region};
use bytes::Bytes;
use xml::reader::{EventReader, XmlEvent};
use roxmltree;
pub fn parse_bucket_location(b: Bytes) -> Result<Region, Err> {
let mut reader = EventReader::new(b.as_ref());
loop {
let event = reader.next();
match event {
Err(err) => return Err(Err::XmlParseErr(err)),
Ok(XmlEvent::EndDocument) => return Err(Err::UnexpectedEOF("xml parsing".to_string())),
Ok(XmlEvent::Characters(s)) => return Ok(Region::new(&s)),
_ => continue,
};
pub fn parse_bucket_location(s: String) -> Result<Region, Err> {
let res = roxmltree::Document::parse(&s);
match res {
Ok(doc) => {
let region_res = doc.root_element().text();
if let Some(region) = region_res {
Ok(Region::new(region))
} else {
Ok(Region::empty())
}
}
Err(e) => Err(Err::XmlParseErr(e)),
}
}