test(cohttp): client bin

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 ()))

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))