xapi-cli-server: Add rate limit CLI operations

Signed-off-by: Christian Pardillo Laursen <[email protected]>

Author: cplaursen

ocaml/xapi-cli-server/cli_frontend.ml

@@ -3869,6 +3869,18 @@ let rec cmdtable_data : (string * cmd_spec) list =
       ; flags= []
       }
     )
+  ; ( "rate-limit-create"
+    , {
+        reqd= ["client-id"; "burst-size"; "fill-rate"]
+      ; optn= []
+      ; help=
+          "Add a rate limit to a XAPI client, by specifying fill rate \
+           (requests per second) and burst size (maximum number of requests at \
+           once)"
+      ; implementation= No_fd Cli_operations.Rate_limit.create
+      ; flags= []
+      }
+    )
   ]
 
 let cmdtable : (string, cmd_spec) Hashtbl.t = Hashtbl.create 50

ocaml/xapi-cli-server/cli_operations.ml

@@ -8179,3 +8179,23 @@ module VM_group = struct
     in
     Client.VM_group.destroy ~rpc ~session_id ~self:ref
 end
+
+module Rate_limit = struct
+  let create printer rpc session_id params =
+    let client_id = List.assoc "client-id" params in
+    let burst_size = float_of_string (List.assoc "burst-size" params) in
+    let fill_rate = float_of_string (List.assoc "fill-rate" params) in
+    let ref =
+      Client.Rate_limit.create ~rpc ~session_id ~client_id ~burst_size
+        ~fill_rate
+    in
+    let uuid = Client.Rate_limit.get_uuid ~rpc ~session_id ~self:ref in
+    printer (Cli_printer.PMsg uuid)
+
+  let destroy _printer rpc session_id params =
+    let ref =
+      Client.Rate_limit.get_by_uuid ~rpc ~session_id
+        ~uuid:(List.assoc "uuid" params)
+    in
+    Client.Rate_limit.destroy ~rpc ~session_id ~self:ref
+end

ocaml/xapi-cli-server/records.ml

@@ -5933,3 +5933,38 @@ let pci_record rpc session_id pci =
           ()
       ]
   }
+
+let rate_limit_record rpc session_id rate_limit =
+  let _ref = ref rate_limit in
+  let empty_record =
+    ToGet (fun () -> Client.Rate_limit.get_record ~rpc ~session_id ~self:!_ref)
+  in
+  let record = ref empty_record in
+  let x () = lzy_get record in
+  {
+    setref=
+      (fun r ->
+        _ref := r ;
+        record := empty_record
+      )
+  ; setrefrec=
+      (fun (a, b) ->
+        _ref := a ;
+        record := Got b
+      )
+  ; record= x
+  ; getref= (fun () -> !_ref)
+  ; fields=
+      [
+        make_field ~name:"uuid" ~get:(fun () -> (x ()).API.rate_limit_uuid) ()
+      ; make_field ~name:"client_id"
+          ~get:(fun () -> (x ()).API.rate_limit_client_id)
+          ()
+      ; make_field ~name:"burst_size"
+          ~get:(fun () -> string_of_float (x ()).API.rate_limit_burst_size)
+          ()
+      ; make_field ~name:"fill_rate"
+          ~get:(fun () -> string_of_float (x ()).API.rate_limit_fill_rate)
+          ()
+      ]
+  }