ocaml-opentelemetry/tests/bin/cohttp_client_eio.ml
2025-05-28 14:25:30 -04:00

75 lines
2.1 KiB
OCaml

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:_ = raise (Failure "TODO")
(* Opentelemetry_cohttp_lwt.client ~scope (module Cohttp_eio.Client) *)
let run () =
(* TODO *)
(* let open Lwt.Syntax in *)
(* let rec go () = *)
(* let@ scope = *)
(* Otel_lwt.Trace.with_ ~kind:T.Span.Span_kind_producer "loop.outer" *)
(* in *)
(* let () = Eio_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 batch_traces = ref 400 in
let batch_metrics = ref 3 in
let opts =
[
"--debug", Arg.Bool (( := ) debug), " enable debug output";
"--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_cohttp_eio.Config.make ~debug:!debug
~batch_traces:(some_if_nzero batch_traces)
~batch_metrics:(some_if_nzero batch_metrics)
()
in
Format.printf "@[<2>sleep outer: %.3fs,@ sleep inner: %.3fs,@ config: %a@]@."
!sleep_outer !sleep_inner Opentelemetry_client_cohttp_eio.Config.pp config;
Format.printf
"Check HTTP requests at \
https://requestbin.com/r/enec1hql02hz/26qShWryt5vJc1JfrOwalhr5vQt@.";
Opentelemetry_client_cohttp_eio.with_setup ~config () run