axum_layer 0.1.1

axum_layer
Documentation 
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

#![feature(impl_trait_in_assoc_type)]

use axum::{http::StatusCode, Router};

pub mod log;

pub fn use_zstd(router: Router) -> Router {
  use tower_http::{
    compression::{
      predicate::{NotForContentType, Predicate, SizeAbove},
      CompressionLayer,
    },
    CompressionLevel,
  };

  router.layer(
    CompressionLayer::new()
      .zstd(true)
      .quality(CompressionLevel::Precise(16))
      .compress_when(
        SizeAbove::new(512)
        // still don't compress gRPC
        .and(NotForContentType::GRPC)
        // still don't compress images
        .and(NotForContentType::IMAGES),
      ),
  )
}

genv::def!(AXUM_TIMEOUT:u64 | 600);

pub fn use_timeout(router: Router) -> Router {
  use axum::error_handling::HandleErrorLayer;
  use tower::{BoxError, ServiceBuilder};

  let middleware = ServiceBuilder::new()
    .layer(HandleErrorLayer::new(|error: BoxError| async move {
      if error.is::<tower::timeout::error::Elapsed>() {
        Ok((StatusCode::REQUEST_TIMEOUT, "timeout"))
      } else {
        Err((
          StatusCode::INTERNAL_SERVER_ERROR,
          format!("Internal Error: {}", error),
        ))
      }
    }))
    .layer(crate::log::Log)
    .timeout(std::time::Duration::from_secs(AXUM_TIMEOUT()).into())
    .layer(ServiceBuilder::new());

  router.layer(middleware.into_inner())
}

pub fn layer(router: Router) -> Router {
  let router = use_zstd(router);
  let router = use_timeout(router);
  router
}