From 2673276cc3fcf040df533f579491bad4a5c913b0 Mon Sep 17 00:00:00 2001
From: Vilsol <me@vil.so>
Date: Fri, 26 Apr 2024 15:45:13 +0300
Subject: [PATCH] fix: override encoding and decoding sizes

---
 cfg/config.json |  2 +-
 src/main.rs     | 18 ++++++++++++++----
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/cfg/config.json b/cfg/config.json
index 5e5bdb4..9699618 100644
--- a/cfg/config.json
+++ b/cfg/config.json
@@ -8,7 +8,7 @@
   "server": {
     "host": "0.0.0.0",
     "port": 50062,
-    "message_size_limit": 16777215
+    "message_size_limit": 64000000
   },
 
   "s3": {
diff --git a/src/main.rs b/src/main.rs
index c7d93c0..cfdd772 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -5,6 +5,7 @@ use std::str::FromStr;
 use std::{env, error::Error, net::ToSocketAddrs};
 use tokio::sync::mpsc;
 
+use tonic::codegen::InterceptedService;
 use tonic::{transport::Server, Request, Status};
 use tonic_health::ServingStatus;
 
@@ -60,13 +61,22 @@ async fn main() -> Result<(), Box<dyn Error>> {
 
     start_renderer(rx).await?;
 
+    let pdf_service =
+        InterceptedService::new(
+            PdfRenderingServiceServer::new(pdf_server)
+                .max_decoding_message_size(
+                    config.get_int("server.message_size_limit").unwrap() as usize
+                )
+                .max_encoding_message_size(
+                    config.get_int("server.message_size_limit").unwrap() as usize
+                ),
+            logging,
+        );
+
     let server = Server::builder()
-        .max_frame_size(config.get_int("server.message_size_limit").map(|t| { t as u32 }).ok())
         .add_service(health_service)
         .add_service(reflection_service)
-        .add_service(PdfRenderingServiceServer::with_interceptor(
-            pdf_server, logging,
-        ))
+        .add_service(pdf_service)
         .serve(
             (format!(
                 "{}:{}",