qubes-mirage-firewall/frameQ.ml

(* Copyright (C) 2016, Thomas Leonard <thomas.leonard@unikernel.com>
   See the README file for details. *)

let src = Logs.Src.create "frameQ" ~doc:"Interface output queue"
module Log = (val Logs.src_log src : Logs.LOG)

type t = {
  name : string;
  mutable items : int;
}

let create name = { name; items = 0 }
  
let send q fn =
  (* TODO: drop if queue too long *)
  let sent = fn () in
  if Lwt.state sent = Lwt.Sleep then (
    q.items <- q.items + 1;
    Log.info (fun f -> f "Queue length for %s: incr to %d" q.name q.items);
    Lwt.on_termination sent (fun () ->
      q.items <- q.items - 1;
      Log.info (fun f -> f "Queue length for %s: decr to %d" q.name q.items);
    )
  );
  sent