From ea086d30d021f7c018e4d307223162cf051de336 Mon Sep 17 00:00:00 2001
From: Guilhem Moulin <guilhem@fripost.org>
Date: Thu, 10 Sep 2015 20:39:50 +0200
Subject: Print IMAP traffic stats when receiving a SIGHUP.

---
 lib/Net/IMAP/InterIMAP.pm | 42 ++++++++++++++++++++++++++++++++----------
 1 file changed, 32 insertions(+), 10 deletions(-)

(limited to 'lib')

diff --git a/lib/Net/IMAP/InterIMAP.pm b/lib/Net/IMAP/InterIMAP.pm
index 076ec19..4222c78 100644
--- a/lib/Net/IMAP/InterIMAP.pm
+++ b/lib/Net/IMAP/InterIMAP.pm
@@ -453,6 +453,22 @@ sub new($%) {
 }
 
 
+# Print traffic statistics
+sub stats($) {
+    my $self = shift;
+    my $msg = 'IMAP traffic (bytes):';
+    $msg .= ' recv '._kibi($self->{_OUTCOUNT});
+    $msg .= ' (compr. '._kibi($self->{_OUTRAWCOUNT}).
+            ', factor '.sprintf('%.2f', $self->{_OUTRAWCOUNT}/$self->{_OUTCOUNT}).')'
+      if defined $self->{_Z_DEFLATE} and $self->{_OUTCOUNT} > 0;
+    $msg .= ' sent '._kibi($self->{_INCOUNT});
+    $msg .= ' (compr. '._kibi($self->{_INRAWCOUNT}).
+            ', factor '.sprintf('%.2f', $self->{_INRAWCOUNT}/$self->{_INCOUNT}).')'
+      if defined $self->{_Z_DEFLATE} and $self->{_INCOUNT} > 0;
+    $self->log($msg);
+}
+
+
 # Log out when the Net::IMAP::InterIMAP object is destroyed.
 sub DESTROY($) {
     my $self = shift;
@@ -462,16 +478,7 @@ sub DESTROY($) {
         $self->{$_}->close() if defined $self->{$_} and $self->{$_}->opened();
     }
 
-    unless ($self->{quiet}) {
-        my $msg = "Connection closed";
-        $msg .= " in=$self->{_INCOUNT}";
-        $msg .= " (raw=$self->{_INRAWCOUNT}, ratio ".sprintf('%.2f', $self->{_INRAWCOUNT}/$self->{_INCOUNT}).")"
-          if defined $self->{_INRAWCOUNT} and $self->{_INCOUNT} > 0 and $self->{_INCOUNT} != $self->{_INRAWCOUNT};
-        $msg .= ", out=$self->{_OUTCOUNT}";
-        $msg .= " (raw=$self->{_OUTRAWCOUNT}, ratio ".sprintf('%.2f', $self->{_OUTRAWCOUNT}/$self->{_OUTCOUNT}).")"
-          if defined $self->{_OUTRAWCOUNT} and $self->{_OUTCOUNT} > 0 and $self->{_OUTCOUNT} != $self->{_OUTRAWCOUNT};
-        $self->log($msg);
-    }
+    $self->stats() unless $self->{quiet};
 }
 
 
@@ -1542,6 +1549,21 @@ sub _select_or_examine($$$;$$) {
 }
 
 
+sub _kibi($) {
+    my $n = shift;
+    if ($n < 1024) {
+        $n;
+    } elsif ($n < 1048576) {
+        sprintf '%.2fK', $n / 1024.;
+    } elsif ($n < 1073741824) {
+        sprintf '%.2fM', $n / 1048576.;
+    } else {
+        sprintf '%.2fG', $n / 1073741824.;
+    }
+
+}
+
+
 
 #############################################################################
 # Parsing methods
-- 
cgit v1.2.3