Merge pull request #259 from doyoubi/FixLargePayload

Fix large payload

Author: doyoubi

src/bin/mem_broker.rs

@@ -19,8 +19,6 @@ use undermoon::broker::{
 #[global_allocator]
 static ALLOC: jemallocator::Jemalloc = jemallocator::Jemalloc;
 
-const MAX_PAYLOAD: usize = 128 * 1024 * 1024;
-
 fn gen_conf() -> MemBrokerConfig {
     let mut s = config::Config::new();
     if let Some(conf_file_path) = env::args().nth(1) {
@@ -190,8 +188,6 @@ async fn main() -> std::io::Result<()> {
     HttpServer::new(move || {
         let service = service.clone();
         App::new()
-            .data(actix_web::web::PayloadConfig::default().limit(MAX_PAYLOAD))
-            .data(actix_web::web::JsonConfig::default().limit(MAX_PAYLOAD))
             .wrap(middleware::Logger::default())
             .configure(|cfg| configure_app(cfg, service.clone()))
     })

src/broker/service.rs

@@ -23,11 +23,14 @@ use std::sync::{Arc, RwLock};
 use std::time::Duration;
 
 pub const MEM_BROKER_API_VERSION: &str = "/api/v2";
+const MAX_PAYLOAD: usize = 128 * 1024 * 1024;
 
 pub fn configure_app(cfg: &mut web::ServiceConfig, service: Arc<MemBrokerService>) {
     let service2 = service.clone();
     cfg.data(service).service(
         web::scope(MEM_BROKER_API_VERSION)
+            .app_data(actix_web::web::PayloadConfig::default().limit(MAX_PAYLOAD))
+            .app_data(actix_web::web::JsonConfig::default().limit(MAX_PAYLOAD))
             .wrap_fn(move |req, srv| {
                 let method = req.method().clone();
                 let peer_addr = match req.peer_addr() {