Source file `eio_inotify.ml`

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
type t = {
  queue   : Inotify.event Queue.t;
  unix_fd : Unix.file_descr;
  poll : Iomux.Poll.t
}

let create () =
  try
    let unix_fd = Inotify.create () in
    let poll = Iomux.Poll.create () in
    Iomux.Poll.set_index poll 0 unix_fd (Iomux.Poll.Flags.pollin);
    {
      queue   = Queue.create ();
      unix_fd; 
      poll
      }
  with exn ->
    raise exn

let add_watch inotify path selector =
  Inotify.add_watch inotify.unix_fd path selector

let rm_watch inotify wd =
  Inotify.rm_watch inotify.unix_fd wd

let rec read inotify =
  try
    Queue.take inotify.queue
  with Queue.Empty ->
    Eio_unix.await_readable inotify.unix_fd;
    let events = Inotify.read inotify.unix_fd in
    List.iter (fun event -> Queue.push event inotify.queue) events;
    read inotify

let with_timeout d = Eio.Fiber.first (fun () -> Eio_unix.sleep d; Error `Timeout)

let rec try_read inotify =
  try
    Some (Queue.take inotify.queue)
  with Queue.Empty ->
  match Iomux.Poll.ppoll_or_poll inotify.poll (Iomux.Util.fd_of_unix inotify.unix_fd) Nowait with
    | r when r > 0 -> Some (read inotify)
    | _ -> None

let close inotify =
  Unix.close inotify.unix_fd