如何组合Lwt过滤器?

时间:2015-03-14 14:09:10

标签: asynchronous ocaml lwt

我目前正在学习Lwt。我感兴趣的是使用异步进程来替换OCaml例程的一些shell例程。

让我们看一下简化的第一次尝试,其中通过组合运行cat的两个线程来创建过滤器:

let filter_cat ()=
  Lwt_process.pmap_lines ("cat", [| "cat" |])

let filter_t () =
  Lwt_io.stdin
  |> Lwt_io.read_lines
  |> filter_cat ()
  |> filter_cat ()
  |> Lwt_io.write_lines Lwt_io.stdout

let () =
  filter_t ()
  |> Lwt_main.run

此过滤器以某种方式工作,但在其标准输入关闭而不是退出时挂起。如果我删除其中一个filter_cat,它会按预期工作。

我猜我没有适当地编写这些过滤器,因此无法加入我正在启动的两个线程。编写这些过滤器的正确方法是什么,以便程序在EOF上读取stdin之后终止?


您可以在BSD Owl中找到此程序以及Github gist Makefile。

1 个答案:

答案 0 :(得分:1)

答案是,Lwt中存在一个小错误。有一个内部函数,monitor执行管道:

(* Monitor the thread [sender] in the stream [st] so write errors are
   reported. *)
let monitor sender st =
  let sender = sender >|= fun () -> None in
  let state = ref Init in
  Lwt_stream.from
    (fun () ->
       match !state with
         | Init ->
             let getter = Lwt.apply Lwt_stream.get st in
             let result _ =
               match Lwt.state sender with
                 | Lwt.Sleep ->
                     (* The sender is still sleeping, behave as the
                        getter. *)
                     getter
                 | Lwt.Return _ ->
                     (* The sender terminated successfully, we are
                        done monitoring it. *)
                     state := Done;
                     getter
                 | Lwt.Fail _ ->
                     (* The sender failed, behave as the sender for
                        this element and save current getter. *)
                     state := Save getter;
                     sender
             in
             Lwt.try_bind (fun () -> Lwt.choose [sender; getter]) result result
         | Save t ->
             state := Done;
             t
         | Done ->
             Lwt_stream.get st)

问题在于定义

let getter = Lwt.apply Lwt_stream.get st

getter进程遇到流的末尾时,它会被保存,但sender会丢失,这似乎会阻止完成。这可以通过改进getter的定义来解决,方法是告诉它在到达流末尾时表现为sender