CP-308863: Count vGPU migrations

The total count can be calculated as AVERAGE * number of samples. For
example, for one-day data granularity, the total count of one day is
AVERAGE * (86400 / 5), in which 86400 is the seconds of one day and 5 is
the interval (in seconds) of data update. Another example is for one
day's data (86400 / 5 = 17280) samples, only one sample is valid as 10,
other 17279 samples are invalid or missing. Then the AVERAGE will be
10 * ( 1 / 17280) and the total count is 10 as expected.

Signed-off-by: Ming Lu <[email protected]>

Author: minglumlu

ocaml/xapi/message_forwarding.ml

@@ -2501,6 +2501,10 @@ functor
             let snapshot = Db.VM.get_record ~__context ~self:vm in
             reserve_memory_for_vm ~__context ~vm ~host ~snapshot
               ~host_op:`vm_migrate (fun () ->
+                Db.VM.get_VGPUs ~__context ~self:vm
+                |> List.iter (fun _ ->
+                       Xapi_stats.incr_pool_vgpu_migration_count ()
+                   ) ;
                 forward_vm_op ~local_fn ~__context ~vm ~remote_fn
             )
         ) ;
@@ -2622,6 +2626,10 @@ functor
                   assert_can_migrate ~__context ~vm ~dest ~live ~vdi_map
                     ~vif_map ~vgpu_map ~options
               ) ;
+              vgpu_map
+              |> List.iter (fun _ ->
+                     Xapi_stats.incr_pool_vgpu_migration_count ()
+                 ) ;
               forward_migrate_send ()
           )
         in

ocaml/xapi/xapi_stats.ml

@@ -16,6 +16,10 @@ module D = Debug.Make (struct let name = "xapi_stats" end)
 
 let with_lock = Xapi_stdext_threads.Threadext.Mutex.execute
 
+let pool_vgpu_migration_count : int Atomic.t = Atomic.make 0
+
+let incr_pool_vgpu_migration_count () = Atomic.incr pool_vgpu_migration_count
+
 let generate_master_stats ~__context =
   let session_count =
     Db.Session.get_all ~__context |> List.length |> Int64.of_int
@@ -44,7 +48,23 @@ let generate_master_stats ~__context =
         ~min:0.0 ~units:"sessions/s" ()
     )
   in
-  [session_count_ds; task_count_ds; session_count_change_ds]
+  let vgpu_migration_count =
+    Atomic.exchange pool_vgpu_migration_count 0 |> Int64.of_int
+  in
+  let vgpu_migration_count_ds =
+    ( Rrd.Host
+    , Ds.ds_make ~name:"pool_vgpu_migration_count"
+        ~description:"Number of vGPU migrations occurred since last update"
+        ~value:(Rrd.VT_Int64 vgpu_migration_count) ~ty:Rrd.Absolute
+        ~default:true ~min:0. ~units:"count" ()
+    )
+  in
+  [
+    session_count_ds
+  ; task_count_ds
+  ; session_count_change_ds
+  ; vgpu_migration_count_ds
+  ]
 
 let gc_debug = ref true
 

ocaml/xapi/xapi_stats.mli

@@ -18,3 +18,6 @@ val start : unit -> unit
 
 val stop : unit -> unit
 (** Stop the stats reporting thread. *)
+
+val incr_pool_vgpu_migration_count : unit -> unit
+(** Increments the pool_vgpu_migration_count by 1 . *)