rust-embed

rust macro which loads files into the rust binary at compile time during release and loads the file from the fs during dev.

49357864 — Bernardo

avoid vector allocation

Files changed (3) hide show

examples/actix.rs CHANGED Viewed

@@ -4,7 +4,7 @@ extern crate rust_embed;
 extern crate mime_guess;
 use actix_web::http::Method;
-use actix_web::{server, App, HttpRequest, HttpResponse};
+use actix_web::{server, App, Body, HttpRequest, HttpResponse};
 use mime_guess::guess_mime_type;
 #[derive(RustEmbed)]
@@ -13,7 +13,9 @@ struct Asset;
 fn handle_embedded_file(path: &str) -> HttpResponse {
   match Asset::get(path) {
+    Some(content) => HttpResponse::Ok()
-    Some(content) => HttpResponse::Ok().content_type(guess_mime_type(path).as_ref()).body(content),
+      .content_type(guess_mime_type(path).as_ref())
+      .body(Body::from_slice(content.as_ref())),
     None => HttpResponse::NotFound().body("404 Not Found"),
   }
 }

examples/basic.rs CHANGED Viewed

@@ -7,5 +7,5 @@ struct Asset;
 fn main() {
   let index_html = Asset::get("index.html").unwrap();
-  println!("{:?}", std::str::from_utf8(&index_html));
+  println!("{:?}", std::str::from_utf8(index_html.as_ref()));
 }

src/lib.rs CHANGED Viewed

@@ -15,7 +15,7 @@ use syn::*;
 fn generate_assets(ident: &syn::Ident, folder_path: String) -> quote::Tokens {
   quote!{
       impl #ident {
-          pub fn get(file_path: &str) -> Option<Vec<u8>> {
+          pub fn get(file_path: &str) -> Option<impl AsRef<[u8]>> {
               use std::fs::File;
               use std::io::Read;
               use std::path::Path;
@@ -63,13 +63,13 @@ fn generate_assets(ident: &syn::Ident, folder_path: String) -> quote::Tokens {
     let key = if std::path::MAIN_SEPARATOR == '\\' { key.replace('\\', "/") } else { key };
     let canonical_path_str = canonical_path.to_str();
     let value = quote!{
-      #key => Some(include_bytes!(#canonical_path_str).to_vec()),
+      #key => Some(&include_bytes!(#canonical_path_str)[..]),
     };
     values.push(value);
   }
   quote!{
       impl #ident {
-          pub fn get(file_path: &str) -> Option<Vec<u8>> {
+          pub fn get(file_path: &str) -> Option<impl AsRef<[u8]>> {
               match file_path {
                   #(#values)*
                   _ => None,