simon/ocaml-opentelemetry
1
0
Fork 0
mirror of https://github.com/ocaml-tracing/ocaml-opentelemetry.git synced 2026-03-09 20:33:36 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

chore(cohttp): trace more methods

Browse source
This commit is contained in:
Matt Bray 2022-03-24 20:40:55 +00:00
parent c8391fc522
commit 32fec9b2f5
1 changed files with 21 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/integrations/cohttp/opentelemetry_cohttp_lwt.ml
View file

@ -174,8 +174,8 @@ let client ?(scope : Otel.Trace.scope option) (module C : Cohttp_lwt.S.Client)
in in
Some headers Some headers
let get ?ctx ?headers (uri : Uri.t) : (Response.t * Cohttp_lwt.Body.t) Lwt.t = let call ?ctx ?headers ?body ?chunked meth (uri : Uri.t) : (Response.t * Cohttp_lwt.Body.t) Lwt.t =
let (trace_id, parent, attrs) = context_for ~uri ~meth:`GET in let (trace_id, parent, attrs) = context_for ~uri ~meth in
Otel_lwt.Trace.with_ "request" Otel_lwt.Trace.with_ "request"
~kind:Span_kind_client ~kind:Span_kind_client
?trace_id ?trace_id
@ -183,12 +183,29 @@ let client ?(scope : Otel.Trace.scope option) (module C : Cohttp_lwt.S.Client)
~attrs ~attrs
(fun scope -> (fun scope ->
let headers = add_traceparent headers in let headers = add_traceparent headers in
let* (res, body) = C.get ?ctx ?headers uri in let* (res, body) = C.call ?ctx ?headers ?body ?chunked meth uri in
Otel.Trace.add_attrs scope (fun () -> Otel.Trace.add_attrs scope (fun () ->
let code = Response.status res in let code = Response.status res in
let code = Code.code_of_status code in let code = Code.code_of_status code in
[ ("http.status_code", `Int code) ]) ; [ ("http.status_code", `Int code) ]) ;
Lwt.return (res, body)) Lwt.return (res, body))
end
let head ?ctx ?headers uri =
let open Lwt.Infix in
call ?ctx ?headers `HEAD uri >|= fst
let get ?ctx ?headers uri = call ?ctx ?headers `GET uri
let delete ?ctx ?body ?chunked ?headers uri =
call ?ctx ?headers ?body ?chunked `DELETE uri
let post ?ctx ?body ?chunked ?headers uri =
call ?ctx ?headers ?body ?chunked `POST uri
let put ?ctx ?body ?chunked ?headers uri =
call ?ctx ?headers ?body ?chunked `PUT uri
let patch ?ctx ?body ?chunked ?headers uri =
call ?ctx ?headers ?body ?chunked `PATCH uri end
in in
(module Traced : Cohttp_lwt.S.Client) (module Traced : Cohttp_lwt.S.Client)