Splice

client/rpc-servers/src/lib.rs

@@ -188,16 +188,30 @@ mod inner {
 		cors: Option<&Vec<String>>,
 		io: RpcHandler<M>,
 		maybe_max_payload_mb: Option<usize>,
+		maybe_max_out_buffer_capacity_mb: Option<usize>,
 		server_metrics: ServerMetrics,
 	) -> io::Result<ws::Server> {
-		let rpc_max_payload = maybe_max_payload_mb
+		let max_payload = maybe_max_payload_mb
 			.map(|mb| mb.saturating_mul(MEGABYTE))
 			.unwrap_or(RPC_MAX_PAYLOAD_DEFAULT);
+		let max_out_buffer_capacity = maybe_max_out_buffer_capacity_mb
+			.map(|mb| mb.saturating_mul(MEGABYTE))
+			.unwrap_or(WS_MAX_BUFFER_CAPACITY_DEFAULT);
+
+		if max_payload > max_out_buffer_capacity {
+			log::warn!(
+				"maximum payload ({}) is more than maximum output buffer ({}) size in ws server, the payload will actually be limited by the buffer size",
+				max_payload,
+				max_out_buffer_capacity,
+			)
+		}
+
 		ws::ServerBuilder::with_meta_extractor(io, |context: &ws::RequestContext| {
 			context.sender().into()
 		})
-		.max_payload(rpc_max_payload)
+		.max_payload(max_payload)
 		.max_connections(max_connections.unwrap_or(WS_MAX_CONNECTIONS))
+		.max_out_buffer_capacity(max_out_buffer_capacity)
 		.allowed_origins(map_cors(cors))
 		.allowed_hosts(hosts_filtering(cors.is_some()))
 		.session_stats(server_metrics)