2016-06-21 26 views
1

首先,对于长标题感到抱歉。Rust AWS API网关服务代理使用原始https请求上传S3文件

我已经设置了一个api网关作为s3代理,所以我可以通过向api url发送一个PUT请求来上传文件。该API工作正常(或者至少我认为它),但似乎只能正确上传文本文件。

对于上传的文本文件(例如Content-Type = text/plain),本地文件的大小与上传到s3存储区的文件大小相同。但是,二进制文件并非如此(例如,Content-Type = application/pdf)。 s3桶中的文件具有更大的尺寸。当我从s3下载二进制文件时,我无法打开它们,它们已损坏。

这里是铁锈代码发送请求,我使用hyper's http client

match File::open(file.as_path()) { 
     Err(_) => Err("Failed to open file".to_owned()), 
     Ok(mut openned_file) => { 
      let file_mime = mime_guess::guess_mime_type(file.as_path()); 
      let connector = HttpsConnector::new(OpensslClient::default()); 
      let url_str = format!("https://my.api.com/upload/{}", 
           file.file_name().unwrap().to_str().unwrap()); 
      let mut client = 
       Request::with_connector(Method::Put, Url::parse(&url_str).unwrap(), &connector) 
        .unwrap(); 
      client.headers_mut().set(ContentType(file_mime.clone())); 
      // client.headers_mut().set(ContentLength(openned_file.metadata().unwrap().len())); 
      let file_mime_str = file_mime.to_string(); 
      let mut buffer: [u8; 4096] = [0; 4096]; 
      let mut uploaded: usize = 0; 
      let request = match file_mime { 
       Mime(TopLevel::Text, _, _) | 
       Mime(TopLevel::Application, SubLevel::Javascript, _) => { 
        let mut request = client.start().unwrap(); 
        println!("Uploading text ...",); 
        while let Ok(read_count) = openned_file.read(&mut buffer) { 
         if read_count > 0 { 
          println!("Uploading {} bytes", read_count); 
          request.write_all(&buffer[0..read_count]); 
          uploaded += read_count; 
         } else { 
          request.flush(); 
          println!("File mime: {}", file_mime_str); 
          println!("File size: {}, Total uploaded: {}", 
            openned_file.metadata().unwrap().len(), 
            uploaded); 
          break; 
         } 
        } 
        request 
       } 
       _ => { 
        // client.headers_mut() 
        //  .set_raw("Content-Encoding", vec![b"base64".to_vec()]); 
        let mut request = client.start().unwrap(); 

        let mut config = MIME; 
        config.line_length = None; 
        println!("Uploading binary ...",); 
        while let Ok(read_count) = openned_file.read(&mut buffer) { 
         if read_count > 0 { 
          println!("Uploading {} bytes", read_count); 
          request.write_all(&buffer[0..read_count]); 
          // let base64_str = buffer[0..read_count].to_base64(STANDARD); 
          // request.write_all(base64_str.into_bytes().as_slice()); 
          uploaded += read_count; 
         } else { 
          request.flush(); 
          println!("File mime: {}", file_mime_str); 
          println!("File size: {}, Total uploaded: {}", 
            openned_file.metadata().unwrap().len(), 
            uploaded); 
          break; 
         } 
        } 
        request 
       } 
      }; 

      match request.send() { 
       Err(err) => Err(format!("{}", err)), 
       Ok(mut response) => { 
        let mut rep_str = String::new(); 
        response.read_to_string(&mut rep_str); 
        Err(format!("{}", rep_str)) 
       } 
      } 

你可以从注释掉的代码中看到的,我已经使用内容编码=的base64编码和尝试从文件中读取的字节上传。但Content-Encoding = base64似乎不是s3接受的有效编码类型。无论何时我设置Content-Encoding,我都无法完全上传(500:内部服务器错误,我甚至无法在s3存储区中看到大小错误的文件)。但文本文件正在完美工作。

参考:

回答

1

不幸的是,此时API网关不支持二进制有效载荷。这是一个常见的功能请求,因此它在我们的待办事项中,尽管我们无法评论发布日期。当功能发布后,我们一定会更新这篇文章。

相关问题