mirror of
https://github.com/ocaml-tracing/ocaml-opentelemetry.git
synced 2026-03-08 03:47:59 -04:00
59 lines
2.2 KiB
OCaml
59 lines
2.2 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 = 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 ()))
|