X-Git-Url: http://git.annexia.org/?p=virt-top.git;a=blobdiff_plain;f=virt-top%2Fvirt_top_utils.ml;h=53c9bf1084d4d4b235019ee0000ad60f139978a7;hp=e92d150895fa40c1bb5d4b17b4846bea2828fb10;hb=40cca545e1e010e8ee1a4ed4e9636b7c0119d5a2;hpb=9bb5f67c0d8a6e8dad7de72e2488ee73c90f7d43

diff --git a/virt-top/virt_top_utils.ml b/virt-top/virt_top_utils.ml
old mode 100644
new mode 100755
index e92d150..53c9bf1
--- a/virt-top/virt_top_utils.ml
+++ b/virt-top/virt_top_utils.ml
@@ -15,10 +15,24 @@
    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+
+   This file contains utility functions.
 *)
 
+open Printf
+
+module C = Libvirt.Connect
+module D = Libvirt.Domain
+module N = Libvirt.Network
+
 let (//) = Filename.concat
 
+(* Int64 operators for convenience. *)
+let (+^) = Int64.add
+let (-^) = Int64.sub
+let ( *^ ) = Int64.mul
+let (/^) = Int64.div
+
 (* Input a whole file as a list of lines. *)
 let input_all_lines chan =
   let lines = ref [] in
@@ -87,3 +101,119 @@ let read_config_file filename =
       let key, value = ExtString.String.split line " " in
       lineno, trim key, trim value
   ) lines
+
+(* Pad a string to the full width with spaces.  If too long, truncate. *)
+let pad width str =
+  let n = String.length str in
+  if n = width then str
+  else if n > width then String.sub str 0 width
+  else (* if n < width then *) str ^ String.make (width-n) ' '
+
+module Show = struct
+  (* Show a percentage in 4 chars. *)
+  let percent percent =
+    if percent <= 0. then " 0.0"
+    else if percent <= 9.9 then sprintf " %1.1f" percent
+    else if percent <= 99.9 then sprintf "%2.1f" percent
+    else "100 "
+
+  (* Show an int64 option in 4 chars. *)
+  let rec int64_option = function
+    | None -> "    "
+    | Some n -> int64 n
+  (* Show an int64 in 4 chars. *)
+  and int64 = function
+    | n when n < 0L -> "-!!!"
+    | n when n <= 9999L ->
+	sprintf "%4Ld" n
+    | n when n /^ 1024L <= 999L ->
+	sprintf "%3LdK" (n /^ 1024L)
+    | n when n /^ 1_048_576L <= 999L ->
+	sprintf "%3LdM" (n /^ 1_048_576L)
+    | n when n /^ 1_073_741_824L <= 999L ->
+	sprintf "%3LdG" (n /^ 1_073_741_824L)
+    | _ -> ">!!!"
+
+  (* Format the total time (may be large!) in 9 chars. *)
+  let time ns =
+    let secs_in_ns = 1_000_000_000L in
+    let mins_in_ns = 60_000_000_000L in
+    let hours_in_ns = 3_600_000_000_000L in
+
+    let hours = ns /^ hours_in_ns in
+    let ns = ns -^ (hours *^ hours_in_ns) in
+    let mins = ns /^ mins_in_ns in
+    let ns = ns -^ (mins *^ mins_in_ns) in
+    let secs = ns /^ secs_in_ns in
+    let ns = ns -^ (secs *^ secs_in_ns) in
+    let pennies = ns /^ 10_000_000L in
+
+    if hours < 12L then
+      sprintf "%3Ld:%02Ld.%02Ld" (hours *^ 60L +^ mins) secs pennies
+    else if hours <= 999L then
+      sprintf "%3Ld:%02Ld:%02Ld" hours mins secs
+    else (
+      let days = hours /^ 24L in
+      let hours = hours -^ (days *^ 24L) in
+      sprintf "%3Ldd%02Ld:%02Ld" days hours mins
+    )
+end
+
+(* Sum Domain.block_stats structures together.  Missing fields
+ * get forced to 0.  Empty list returns all 0.
+ *)
+let zero_block_stats =
+  { D.rd_req = 0L; rd_bytes = 0L; wr_req = 0L; wr_bytes = 0L; errs = 0L }
+let add_block_stats bs1 bs2 =
+  let add f1 f2 = if f1 >= 0L && f2 >= 0L then f1 +^ f2 else 0L in
+  { D.rd_req = add bs1.D.rd_req   bs2.D.rd_req;
+    rd_bytes = add bs1.D.rd_bytes bs2.D.rd_bytes;
+    wr_req   = add bs1.D.wr_req   bs2.D.wr_req;
+    wr_bytes = add bs1.D.wr_bytes bs2.D.wr_bytes;
+    errs     = add bs1.D.errs     bs2.D.errs }
+let sum_block_stats =
+  List.fold_left add_block_stats zero_block_stats
+
+(* Get the difference between two block_stats structures.  Missing data
+ * forces the difference to -1.
+ *)
+let diff_block_stats curr prev =
+  let sub f1 f2 = if f1 >= 0L && f2 >= 0L then f1 -^ f2 else -1L in
+  { D.rd_req = sub curr.D.rd_req   prev.D.rd_req;
+    rd_bytes = sub curr.D.rd_bytes prev.D.rd_bytes;
+    wr_req   = sub curr.D.wr_req   prev.D.wr_req;
+    wr_bytes = sub curr.D.wr_bytes prev.D.wr_bytes;
+    errs     = sub curr.D.errs     prev.D.errs }
+
+(* Sum Domain.interface_stats structures together.  Missing fields
+ * get forced to 0.  Empty list returns all 0.
+ *)
+let zero_interface_stats =
+  { D.rx_bytes = 0L; rx_packets = 0L; rx_errs = 0L; rx_drop = 0L;
+    tx_bytes = 0L; tx_packets = 0L; tx_errs = 0L; tx_drop = 0L }
+let add_interface_stats is1 is2 =
+  let add f1 f2 = if f1 >= 0L && f2 >= 0L then f1 +^ f2 else 0L in
+  { D.rx_bytes = add is1.D.rx_bytes   is2.D.rx_bytes;
+    rx_packets = add is1.D.rx_packets is2.D.rx_packets;
+    rx_errs    = add is1.D.rx_errs    is2.D.rx_errs;
+    rx_drop    = add is1.D.rx_drop    is2.D.rx_drop;
+    tx_bytes   = add is1.D.tx_bytes   is2.D.tx_bytes;
+    tx_packets = add is1.D.tx_packets is2.D.tx_packets;
+    tx_errs    = add is1.D.tx_errs    is2.D.tx_errs;
+    tx_drop    = add is1.D.tx_drop    is2.D.tx_drop }
+let sum_interface_stats =
+  List.fold_left add_interface_stats zero_interface_stats
+
+(* Get the difference between two interface_stats structures.
+ * Missing data forces the difference to -1.
+ *)
+let diff_interface_stats curr prev =
+  let sub f1 f2 = if f1 >= 0L && f2 >= 0L then f1 -^ f2 else -1L in
+  { D.rx_bytes = sub curr.D.rx_bytes   prev.D.rx_bytes;
+    rx_packets = sub curr.D.rx_packets prev.D.rx_packets;
+    rx_errs    = sub curr.D.rx_errs    prev.D.rx_errs;
+    rx_drop    = sub curr.D.rx_drop    prev.D.rx_drop;
+    tx_bytes   = sub curr.D.tx_bytes   prev.D.tx_bytes;
+    tx_packets = sub curr.D.tx_packets prev.D.tx_packets;
+    tx_errs    = sub curr.D.tx_errs    prev.D.tx_errs;
+    tx_drop    = sub curr.D.tx_drop    prev.D.tx_drop }