simon/tiny_httpd
1
0
Fork 0
mirror of https://github.com/c-cube/tiny_httpd.git synced 2025-12-06 03:05:29 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: use chunked encoding for large string responses, too

Browse source
This commit is contained in:
Simon Cruanes 2020-05-27 22:16:36 -04:00
parent 020dd056ac
commit fdd97744e0
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/Tiny_httpd.ml
View file

@ -619,9 +619,12 @@ module Response = struct
let output_ (oc:out_channel) (self:t) : unit =
Printf.fprintf oc "HTTP/1.1 %d %s\r\n" self.code (Response_code.descr self.code);
let is_chunked = match self.body with
| `String _ -> false
| `Stream _ -> true
let body, is_chunked = match self.body with
| `String s when String.length s > 1024 * 300 ->
(* chunk-encode large bodies *)
`Stream (Byte_stream.of_string s), true
| `String _ as b -> b, false
| `Stream _ as b -> b, true
in
let headers =
if is_chunked
@ -630,7 +633,7 @@ module Response = struct
in
List.iter (fun (k,v) -> Printf.fprintf oc "%s: %s\r\n" k v) headers;
output_string oc "\r\n";
begin match self.body with
begin match body with
| `String "" -> ()
| `String s -> output_string oc s;
| `Stream str -> output_stream_chunked_ oc str;