Source file trace_service.ml

[@@@ocaml.warning "-23-27-30-39-44"]

type export_trace_service_request = {
  mutable resource_spans : Trace.resource_spans list;
}

type export_trace_partial_success = {
  mutable _presence: Pbrt.Bitfield.t; (** presence for 2 fields *)
  mutable rejected_spans : int64;
  mutable error_message : string;
}

type export_trace_service_response = {
  mutable partial_success : export_trace_partial_success option;
}

let default_export_trace_service_request (): export_trace_service_request =
{
  resource_spans=[];
}

let default_export_trace_partial_success (): export_trace_partial_success =
{
  _presence=Pbrt.Bitfield.empty;
  rejected_spans=0L;
  error_message="";
}

let default_export_trace_service_response (): export_trace_service_response =
{
  partial_success=None;
}


(** {2 Make functions} *)


let[@inline] export_trace_service_request_set_resource_spans (self:export_trace_service_request) (x:Trace.resource_spans list) : unit =
  self.resource_spans <- x

let copy_export_trace_service_request (self:export_trace_service_request) : export_trace_service_request =
  { self with resource_spans = self.resource_spans }

let make_export_trace_service_request 
  ?(resource_spans=[])
  () : export_trace_service_request  =
  let _res = default_export_trace_service_request () in
  export_trace_service_request_set_resource_spans _res resource_spans;
  _res

let[@inline] export_trace_partial_success_has_rejected_spans (self:export_trace_partial_success) : bool = (Pbrt.Bitfield.get self._presence 0)
let[@inline] export_trace_partial_success_has_error_message (self:export_trace_partial_success) : bool = (Pbrt.Bitfield.get self._presence 1)

let[@inline] export_trace_partial_success_set_rejected_spans (self:export_trace_partial_success) (x:int64) : unit =
  self._presence <- (Pbrt.Bitfield.set self._presence 0); self.rejected_spans <- x
let[@inline] export_trace_partial_success_set_error_message (self:export_trace_partial_success) (x:string) : unit =
  self._presence <- (Pbrt.Bitfield.set self._presence 1); self.error_message <- x

let copy_export_trace_partial_success (self:export_trace_partial_success) : export_trace_partial_success =
  { self with rejected_spans = self.rejected_spans }

let make_export_trace_partial_success 
  ?(rejected_spans:int64 option)
  ?(error_message:string option)
  () : export_trace_partial_success  =
  let _res = default_export_trace_partial_success () in
  (match rejected_spans with
  | None -> ()
  | Some v -> export_trace_partial_success_set_rejected_spans _res v);
  (match error_message with
  | None -> ()
  | Some v -> export_trace_partial_success_set_error_message _res v);
  _res


let[@inline] export_trace_service_response_set_partial_success (self:export_trace_service_response) (x:export_trace_partial_success) : unit =
  self.partial_success <- Some x

let copy_export_trace_service_response (self:export_trace_service_response) : export_trace_service_response =
  { self with partial_success = self.partial_success }

let make_export_trace_service_response 
  ?(partial_success:export_trace_partial_success option)
  () : export_trace_service_response  =
  let _res = default_export_trace_service_response () in
  (match partial_success with
  | None -> ()
  | Some v -> export_trace_service_response_set_partial_success _res v);
  _res

[@@@ocaml.warning "-23-27-30-39"]

(** {2 Formatters} *)

let rec pp_export_trace_service_request fmt (v:export_trace_service_request) = 
  let pp_i fmt () =
    Pbrt.Pp.pp_record_field ~first:true "resource_spans" (Pbrt.Pp.pp_list Trace.pp_resource_spans) fmt v.resource_spans;
  in
  Pbrt.Pp.pp_brk pp_i fmt ()

let rec pp_export_trace_partial_success fmt (v:export_trace_partial_success) = 
  let pp_i fmt () =
    Pbrt.Pp.pp_record_field ~absent:(not (export_trace_partial_success_has_rejected_spans v)) ~first:true "rejected_spans" Pbrt.Pp.pp_int64 fmt v.rejected_spans;
    Pbrt.Pp.pp_record_field ~absent:(not (export_trace_partial_success_has_error_message v)) ~first:false "error_message" Pbrt.Pp.pp_string fmt v.error_message;
  in
  Pbrt.Pp.pp_brk pp_i fmt ()

let rec pp_export_trace_service_response fmt (v:export_trace_service_response) = 
  let pp_i fmt () =
    Pbrt.Pp.pp_record_field ~first:true "partial_success" (Pbrt.Pp.pp_option pp_export_trace_partial_success) fmt v.partial_success;
  in
  Pbrt.Pp.pp_brk pp_i fmt ()

[@@@ocaml.warning "-23-27-30-39"]

(** {2 Protobuf Encoding} *)

let rec encode_pb_export_trace_service_request (v:export_trace_service_request) encoder = 
  Pbrt.List_util.rev_iter_with (fun x encoder ->
    Pbrt.Encoder.nested Trace.encode_pb_resource_spans x encoder;
    Pbrt.Encoder.key 1 Pbrt.Bytes encoder; 
  ) v.resource_spans encoder;
  ()

let rec encode_pb_export_trace_partial_success (v:export_trace_partial_success) encoder = 
  if export_trace_partial_success_has_rejected_spans v then (
    Pbrt.Encoder.int64_as_varint v.rejected_spans encoder;
    Pbrt.Encoder.key 1 Pbrt.Varint encoder; 
  );
  if export_trace_partial_success_has_error_message v then (
    Pbrt.Encoder.string v.error_message encoder;
    Pbrt.Encoder.key 2 Pbrt.Bytes encoder; 
  );
  ()

let rec encode_pb_export_trace_service_response (v:export_trace_service_response) encoder = 
  begin match v.partial_success with
  | Some x -> 
    Pbrt.Encoder.nested encode_pb_export_trace_partial_success x encoder;
    Pbrt.Encoder.key 1 Pbrt.Bytes encoder; 
  | None -> ();
  end;
  ()

[@@@ocaml.warning "-23-27-30-39"]

(** {2 Protobuf Decoding} *)

let rec decode_pb_export_trace_service_request d =
  let v = default_export_trace_service_request () in
  let continue__= ref true in
  while !continue__ do
    match Pbrt.Decoder.key d with
    | None -> (
      (* put lists in the correct order *)
      export_trace_service_request_set_resource_spans v (List.rev v.resource_spans);
    ); continue__ := false
    | Some (1, Pbrt.Bytes) -> begin
      export_trace_service_request_set_resource_spans v ((Trace.decode_pb_resource_spans (Pbrt.Decoder.nested d)) :: v.resource_spans);
    end
    | Some (1, pk) -> 
      Pbrt.Decoder.unexpected_payload_message "export_trace_service_request" 1 pk
    | Some (_, payload_kind) -> Pbrt.Decoder.skip d payload_kind
  done;
  (v : export_trace_service_request)

let rec decode_pb_export_trace_partial_success d =
  let v = default_export_trace_partial_success () in
  let continue__= ref true in
  while !continue__ do
    match Pbrt.Decoder.key d with
    | None -> (
    ); continue__ := false
    | Some (1, Pbrt.Varint) -> begin
      export_trace_partial_success_set_rejected_spans v (Pbrt.Decoder.int64_as_varint d);
    end
    | Some (1, pk) -> 
      Pbrt.Decoder.unexpected_payload_message "export_trace_partial_success" 1 pk
    | Some (2, Pbrt.Bytes) -> begin
      export_trace_partial_success_set_error_message v (Pbrt.Decoder.string d);
    end
    | Some (2, pk) -> 
      Pbrt.Decoder.unexpected_payload_message "export_trace_partial_success" 2 pk
    | Some (_, payload_kind) -> Pbrt.Decoder.skip d payload_kind
  done;
  (v : export_trace_partial_success)

let rec decode_pb_export_trace_service_response d =
  let v = default_export_trace_service_response () in
  let continue__= ref true in
  while !continue__ do
    match Pbrt.Decoder.key d with
    | None -> (
    ); continue__ := false
    | Some (1, Pbrt.Bytes) -> begin
      export_trace_service_response_set_partial_success v (decode_pb_export_trace_partial_success (Pbrt.Decoder.nested d));
    end
    | Some (1, pk) -> 
      Pbrt.Decoder.unexpected_payload_message "export_trace_service_response" 1 pk
    | Some (_, payload_kind) -> Pbrt.Decoder.skip d payload_kind
  done;
  (v : export_trace_service_response)

[@@@ocaml.warning "-23-27-30-39"]

(** {2 Protobuf YoJson Encoding} *)

let rec encode_json_export_trace_service_request (v:export_trace_service_request) = 
  let assoc = ref [] in
  assoc := (
    let l = v.resource_spans |> List.map Trace.encode_json_resource_spans in
    ("resourceSpans", `List l) :: !assoc 
  );
  `Assoc !assoc

let rec encode_json_export_trace_partial_success (v:export_trace_partial_success) = 
  let assoc = ref [] in
  if export_trace_partial_success_has_rejected_spans v then (
    assoc := ("rejectedSpans", Pbrt_yojson.make_string (Int64.to_string v.rejected_spans)) :: !assoc;
  );
  if export_trace_partial_success_has_error_message v then (
    assoc := ("errorMessage", Pbrt_yojson.make_string v.error_message) :: !assoc;
  );
  `Assoc !assoc

let rec encode_json_export_trace_service_response (v:export_trace_service_response) = 
  let assoc = ref [] in
  assoc := (match v.partial_success with
    | None -> !assoc
    | Some v -> ("partialSuccess", encode_json_export_trace_partial_success v) :: !assoc);
  `Assoc !assoc

[@@@ocaml.warning "-23-27-30-39"]

(** {2 JSON Decoding} *)

let rec decode_json_export_trace_service_request d =
  let v = default_export_trace_service_request () in
  let assoc = match d with
    | `Assoc assoc -> assoc
    | _ -> assert(false)
  in
  List.iter (function 
    | ("resourceSpans", `List l) -> begin
      export_trace_service_request_set_resource_spans v @@ List.map (function
        | json_value -> (Trace.decode_json_resource_spans json_value)
      ) l;
    end
    
    | (_, _) -> () (*Unknown fields are ignored*)
  ) assoc;
  ({
    resource_spans = v.resource_spans;
  } : export_trace_service_request)

let rec decode_json_export_trace_partial_success d =
  let v = default_export_trace_partial_success () in
  let assoc = match d with
    | `Assoc assoc -> assoc
    | _ -> assert(false)
  in
  List.iter (function 
    | ("rejectedSpans", json_value) -> 
      export_trace_partial_success_set_rejected_spans v (Pbrt_yojson.int64 json_value "export_trace_partial_success" "rejected_spans")
    | ("errorMessage", json_value) -> 
      export_trace_partial_success_set_error_message v (Pbrt_yojson.string json_value "export_trace_partial_success" "error_message")
    
    | (_, _) -> () (*Unknown fields are ignored*)
  ) assoc;
  ({
    _presence = v._presence;
    rejected_spans = v.rejected_spans;
    error_message = v.error_message;
  } : export_trace_partial_success)

let rec decode_json_export_trace_service_response d =
  let v = default_export_trace_service_response () in
  let assoc = match d with
    | `Assoc assoc -> assoc
    | _ -> assert(false)
  in
  List.iter (function 
    | ("partialSuccess", json_value) -> 
      export_trace_service_response_set_partial_success v (decode_json_export_trace_partial_success json_value)
    
    | (_, _) -> () (*Unknown fields are ignored*)
  ) assoc;
  ({
    partial_success = v.partial_success;
  } : export_trace_service_response)