国展网站建设,鄂州网站建设企业推广,免费空间 网站,网站规划和网站建设本博客是在杨旭老师的 rust web 全栈教程项目基础上进行修改#xff0c;支持了图片资源返回#xff0c;杨旭老师的rust web链接如下#xff1a;
https://www.bilibili.com/video/BV1RP4y1G7KFp1vd_source8595fbbf160cc11a0cc07cadacf22951
本人默认读者已经学习了相关…本博客是在杨旭老师的 rust web 全栈教程项目基础上进行修改支持了图片资源返回杨旭老师的rust web链接如下
https://www.bilibili.com/video/BV1RP4y1G7KFp1vd_source8595fbbf160cc11a0cc07cadacf22951
本人默认读者已经学习了相关基础教程和杨旭老师的相关课程直接开始针对项目里面对应文件进行修改说明。
一、新增加载图片资源方法
在原先项目加载文件方法同一层新增支持加载图片资源如下。
pub trait Handler{fn handle(res:HttpRequest) -HttpResponse;fn load_file(file_name:str) -OptionVecu8{println!(load_file file_name{},file_name);let default_path format!({}/public,env!(CARGO_MANIFEST_DIR));println!(load_file default_path{},default_path);let public_path env::var(PUBLIC_PATH).unwrap_or(default_path);let full_path format!({}/{},public_path,file_name);println!(load_file full_path{},full_path);let contents fs::read(full_path);contents.ok()}fn load_image(image_name:str) -OptionVecu8{let default_path format!({}/public,env!(CARGO_MANIFEST_DIR));let public_path env::var(PUBLIC_PATH).unwrap_or(default_path);let full_path format!({}/{}/{},public_path,image,image_name);println!(load_image full_path{},full_path);let contents: ResultVecu8, std::io::Error fs::read(full_path);contents.ok()}
}
load_image 方法就是本博客新增方法根据入参图片名称去指定public目录下image文件夹下读取图片文件返回Vecu8 字节流。
我们在旧的新增了图片处理分支如下
impl Handler for StaticPageHandler{fn handle(req:HttpRequest) -HttpResponse{let http::httprequest::Resource::Path(s) req.resourece;let route:Vecstr s.split(/).collect();println!(route{},route[1]);if route[1]image {match Self::load_image(route[2]){Some(content) {let mut map:HashMapstr,str HashMap::new();map.insert(Content-Type,image/webp);println!(读取到文件长度{},content.len());return HttpResponse::new(200,Some(map),Some(content));},None {return HttpResponse::new(404,None,Self::load_file(404.html))}}}match route[1] { HttpResponse::new(200,None,Self::load_file(index.html)),health HttpResponse::new(200,None,Self::load_file(health.html)),path match Self::load_file(path) {Some(contents) {let mut map:HashMapstr,str HashMap::new();if path.ends_with(.css) {map.insert(Content-Type,text/css);}else if path.ends_with(.js) {map.insert(Content-Type,text/javascript);}else {map.insert(Content-Type,text/html);}HttpResponse::new(200,Some(map),Some(contents))},None HttpResponse::new(404,None,Self::load_file(404.html))}}}
}
特别说明点根据加载图片类型不同map.insert(Content-Type,image/webp);
这里的Content-Type也要对应修改我这里返回的文件是webp格式。
二、修改HttpResponse 结构体body数据类型
旧项目中由于返回的都是文本内容body数据类型定义成字符串切片由于这次改动读取图片是字节流所以这里的body也需要变更成OptionVecu8类型
#[derive(Debug,PartialEq,Clone)]
pub struct HttpResponsea{version:a str,status_code:a str,status_text:a str,header:OptionHashMapa str,a str,body:OptionVecu8,
}
三、修改HttpResponse 响应方法
上面已经提到了旧项目返回的是文本内容故在发送HttpResponse数据是直接将数据转化成字符串发送。这里我们为了兼容图片资源的响应所以body为Vecu8类型时就不适用先前的方法。我们对除了body字段其他字段依然按照旧的逻辑转化成字符串。
impl a FromHttpResponsea for String{fn from(res:HttpResponse) - String{let res1 res.clone();format!({} {} {}\r\n{}Content-Length:{}\r\n\r\n,res1.version(),res1.status_code(),res1.status_text(),res1.header(),res.body.unwrap().len(),)}
} 相比较旧的方法我们去掉了body参数。
四、返回body数据
上面讲过了我们在响应转String时将body数据拆分出来了那么通过什么方法返回body数据。 pub fn send_response(self, write_stream:mut impl Write) - Result(),std::io::Error{let res self.clone();let response_string: String String::from(res); // from traitwrite_stream.write(response_string.as_bytes()).unwrap();write_stream.write(self.body()).unwrap();write_stream.flush().unwrap();Ok(())} 我们就是通过write_stream.write(self.body()).unwrap();将body数据再独立发送一下。
整理上来讲改动很小就是将body数据类型从str改成了Vecu8,在发送响应时将body拆分出来发送。
