janestreet / async_udp Goto Github PK

open Core
open Async

module Udp = Async_udp

let () = Shutdown.set_default_force (fun () -> after (sec 30.))

let n_packets = 100000
let socket_fd = Socket.fd (Socket.create (Socket.Type.udp))
let inet_addr = Caml.Unix.inet_addr_of_string "127.0.0.1"
let socket_address = Socket.Address.Inet.create inet_addr ~port:8125
let buf = "memleak" |> Iobuf.of_string |> Iobuf.read_only

let run_sendto () =
  let send = Udp.sendto () |> Or_error.ok_exn in
  Deferred.for_ 1 ~to_:n_packets ~do_:
    (fun i ->
       send socket_fd buf socket_address >>| fun () -> 
       if (i % 10000) = 0 then Gc.compact ();
       if (i = n_packets) then Shutdown.shutdown 0;
       Iobuf.reset buf)
  
let () =
  run_sendto () |> Deferred.don't_wait_for;
  never_returns (Scheduler.go ())