fix tailcall in recursive loop (#77)

anmonteiro · Jul 28, 2024 · 5af38bf · 5af38bf
1 parent a2439ef
commit 5af38bf
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lwt/gluten_lwt.ml b/lwt/gluten_lwt.ml
@@ -66,14 +66,15 @@ module IO_loop = struct
             (fun () ->
                read (module Io) socket read_buffer >>= fun (_ : int) ->
                let (_ : int) = Buffer.get read_buffer ~f:(Runtime.read t) in
-               read_loop_step ())
+               Lwt.return_unit)
             (function
                | End_of_file ->
                  let (_ : int) =
                    Buffer.get read_buffer ~f:(Runtime.read_eof t)
                  in
-                 read_loop_step ()
+                 Lwt.return_unit
                | exn -> Lwt.fail exn)
+          >>= read_loop_step
         | `Yield ->
           Runtime.yield_reader t read_loop;
           Lwt.return_unit