balsoft/ligo
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Client: temporary hack

Browse Source 
reaggregate http chunks for streamed RPC... :(
This commit is contained in:
Grégoire Henry 2018-04-17 10:35:01 +02:00
parent ec25690166
commit aeaf2a240c
1 changed files with 20 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
vendors/ocplib-resto/lib_resto-cohttp/client.ml vendored
View File

@ -319,32 +319,27 @@ module Make (Encoding : Resto.ENCODING) = struct
on_close () ;
Lwt.return (`Ok None)
| Some chunk ->
let buffer = Buffer.create 2048 in
let output = Service.output_encoding service in
log.log ~media output
`OK (lazy (Lwt.return chunk)) >>= fun () ->
match media.destruct output chunk with
| Error msg ->
Lwt.return (`Unexpected_content ((chunk, media), msg))
| Ok body ->
on_chunk body ;
let rec loop () =
Lwt_stream.get stream >>= function
| None -> on_close () ; Lwt.return_unit
| Some chunk ->
log.log ~media output
`OK (lazy (Lwt.return chunk)) >>= fun () ->
match media.destruct output chunk with
| Ok body -> on_chunk body ; loop ()
| Error _msg ->
(* TODO log error. *)
ignore (Lwt_stream.junk_while (fun _ -> true) stream
: unit Lwt.t) ;
on_close () ; Lwt.return_unit in
ignore (loop () : unit Lwt.t) ;
Lwt.return (`Ok (Some (fun () ->
ignore (Lwt_stream.junk_while (fun _ -> true) stream
: unit Lwt.t) ;
())))
let rec loop = function
| None -> on_close () ; Lwt.return_unit
| Some chunk ->
Buffer.add_string buffer chunk ;
let data = Buffer.contents buffer in
log.log ~media output
`OK (lazy (Lwt.return chunk)) >>= fun () ->
match media.destruct output data with
| Ok body ->
Buffer.reset buffer ;
on_chunk body ;
Lwt_stream.get stream >>= loop
| Error _msg ->
Lwt_stream.get stream >>= loop in
ignore (loop (Some chunk) : unit Lwt.t) ;
Lwt.return (`Ok (Some (fun () ->
ignore (Lwt_stream.junk_while (fun _ -> true) stream
: unit Lwt.t) ;
())))
end
| `Conflict body ->
handle_error log service body `Conflict (fun v -> `Conflict v)