FilesExpand file tree

 master

/

main.rs

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

81 lines (66 loc) · 2.49 KB

 master

/

main.rs

Copy path

Top

File metadata and controls

• Code
• Blame

81 lines (66 loc) · 2.49 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

use std::{sync::Arc, time::Instant};

use actix_web::{get, middleware, web::Data, App, HttpResponse, HttpServer};

use awc::{http::header, Client, Connector};

use rustls::{ClientConfig, OwnedTrustAnchor, RootCertStore};

const MAP_URL: &str =

"https://upload.wikimedia.org/wikipedia/commons/f/ff/Pizigani_1367_Chart_10MB.jpg";

#[get("/")]

async fn fetch_image(client: Data<Client>) -> HttpResponse {

let start = Instant::now();

let mut res = client.get(MAP_URL).send().await.unwrap();

if !res.status().is_success() {

log::error!("Wikipedia did not return expected image");

return HttpResponse::InternalServerError().finish();

}

let payload = res

.body()

// expected image is larger than default body limit

.limit(20_000_000) // 20MB

.await

.unwrap();

log::info!(

"it took {}ms to download image to memory",

start.elapsed().as_millis()

);

HttpResponse::Ok()

.content_type(mime::IMAGE_JPEG)

.body(payload)

}

#[actix_web::main]

async fn main() -> std::io::Result<()> {

env_logger::init_from_env(env_logger::Env::new().default_filter_or("info"));

let client_tls_config = Arc::new(rustls_config());

log::info!("starting HTTP server at http://localhost:8080");

HttpServer::new(move || {

// create client _inside_ `HttpServer::new` closure to have one per worker thread

let client = Client::builder()

// Wikipedia requires a User-Agent header to make requests

.add_default_header((header::USER_AGENT, "awc-example/1.0"))

// a "connector" wraps the stream into an encrypted connection

.connector(Connector::new().rustls(Arc::clone(&client_tls_config)))

.finish();

App::new()

.wrap(middleware::Logger::default())

.app_data(Data::new(client))

.service(fetch_image)

})

.bind(("127.0.0.1", 8080))?

.workers(2)

.run()

.await

}

/// Create simple rustls client config from root certificates.

fn rustls_config() -> ClientConfig {

let mut root_store = RootCertStore::empty();

root_store.add_server_trust_anchors(webpki_roots::TLS_SERVER_ROOTS.0.iter().map(|ta| {

OwnedTrustAnchor::from_subject_spki_name_constraints(

ta.subject,

ta.spki,

ta.name_constraints,

)

}));

rustls::ClientConfig::builder()

.with_safe_defaults()

.with_root_certificates(root_store)

.with_no_client_auth()

}