diff --git a/tests/bin/cohttp_client.ml b/tests/bin/cohttp_client.ml new file mode 100644 index 00000000..9070b30a --- /dev/null +++ b/tests/bin/cohttp_client.ml @@ -0,0 +1,59 @@ +module T = Opentelemetry +module Otel_lwt = Opentelemetry_lwt +let spf = Printf.sprintf +let (let@) f x = f x + +let sleep_inner = ref 0.1 +let sleep_outer = ref 2.0 + +let mk_client ~scope = Opentelemetry_cohttp_lwt.client ~scope (module Cohttp_lwt_unix.Client) + +let run () = + Printf.printf "collector is on %S\n%!" (Opentelemetry_client_ocurl.get_url()); + let open Lwt.Syntax in + let rec go () = + let@ scope = + Otel_lwt.Trace.with_ + ~kind:T.Span.Span_kind_producer + "loop.outer" + in + let* () = Lwt_unix.sleep !sleep_outer in + let module C = (val mk_client ~scope) in + let* (res, body) = C.get (Uri.of_string "https://enec1hql02hz.x.pipedream.net") in + let* () = Cohttp_lwt.Body.drain_body body in + go () + in + go () + +let () = + Sys.catch_break true; + T.Globals.service_name := "ocaml-otel-cohttp-client"; + T.Globals.service_namespace := Some "ocaml-otel.test"; + + let debug = ref false in + let thread = ref true in + let batch_traces = ref 400 in + let batch_metrics = ref 3 in + let opts = [ + "--debug", Arg.Bool ((:=) debug), " enable debug output"; + "--thread", Arg.Bool ((:=) thread), " use a background thread"; + "--batch-traces", Arg.Int ((:=) batch_traces), " size of traces batch"; + "--batch-metrics", Arg.Int ((:=) batch_metrics), " size of metrics batch"; + "--sleep-inner", Arg.Set_float sleep_inner, " sleep (in s) in inner loop"; + "--sleep-outer", Arg.Set_float sleep_outer, " sleep (in s) in outer loop"; + ] |> Arg.align in + + Arg.parse opts (fun _ -> ()) "emit1 [opt]*"; + + let some_if_nzero r = if !r > 0 then Some !r else None in + let config = Opentelemetry_client_ocurl.Config.make + ~debug:!debug + ~batch_traces:(some_if_nzero batch_traces) + ~batch_metrics:(some_if_nzero batch_metrics) + ~thread:!thread () in + Format.printf "@[<2>sleep outer: %.3fs,@ sleep inner: %.3fs,@ config: %a@]@." + !sleep_outer !sleep_inner Opentelemetry_client_ocurl.Config.pp config; + + Format.printf "Check HTTP requests at https://requestbin.com/r/enec1hql02hz/26qShWryt5vJc1JfrOwalhr5vQt@."; + + Opentelemetry_client_ocurl.with_setup ~config () (fun () -> Lwt_main.run (run ())) diff --git a/tests/bin/dune b/tests/bin/dune index d41bf219..639b3ec4 100644 --- a/tests/bin/dune +++ b/tests/bin/dune @@ -1,3 +1,9 @@ (executable (name emit1) + (modules emit1) (libraries unix opentelemetry opentelemetry-client-ocurl)) + +(executable + (name cohttp_client) + (modules cohttp_client) + (libraries cohttp-lwt-unix opentelemetry opentelemetry-client-ocurl opentelemetry-cohttp-lwt))