Browse Source

Print accounting information in heartbeat messages.

Implements ticket 5526.
Peter Retzlaff 11 years ago
parent
commit
a337d4b7cb
5 changed files with 40 additions and 0 deletions
  1. 4 0
      changes/ticket5526
  2. 6 0
      src/or/hibernate.c
  3. 1 0
      src/or/hibernate.h
  4. 28 0
      src/or/status.c
  5. 1 0
      src/or/status.h

+ 4 - 0
changes/ticket5526

@@ -0,0 +1,4 @@
+  o Minor features:
+    - Log current accounting state (bytes sent and received + remaining
+      time for the current accounting period) in the server's
+      heartbeat message. 

+ 6 - 0
src/or/hibernate.c

@@ -255,6 +255,12 @@ accounting_get_interval_length(void)
   return (int)(interval_end_time - interval_start_time);
 }
 
+time_t
+accounting_get_end_time(void)
+{
+  return interval_end_time;
+}
+
 /** Called from main.c to tell us that <b>seconds</b> seconds have
  * passed, <b>n_read</b> bytes have been read, and <b>n_written</b>
  * bytes have been written. */

+ 1 - 0
src/or/hibernate.h

@@ -15,6 +15,7 @@
 int accounting_parse_options(const or_options_t *options, int validate_only);
 int accounting_is_enabled(const or_options_t *options);
 int accounting_get_interval_length(void);
+time_t accounting_get_end_time(void);
 void configure_accounting(time_t now);
 void accounting_run_housekeeping(time_t now);
 void accounting_add_bytes(size_t n_read, size_t n_written, int seconds);

+ 28 - 0
src/or/status.c

@@ -15,6 +15,7 @@
 #include "circuitlist.h"
 #include "main.h"
 #include "hibernate.h"
+#include "statefile.h"
 
 /** Return the total number of circuits. */
 static int
@@ -111,6 +112,10 @@ log_heartbeat(time_t now)
          uptime, count_circuits(),bw_sent,bw_rcvd,
          hibernating?" We are currently hibernating.":"");
 
+  if (server_mode(options) && accounting_is_enabled(options) && !hibernating) {
+    log_accounting(now, options);
+  }
+
   if (stats_n_data_cells_packaged && !hibernating)
     log_notice(LD_HEARTBEAT, "Average packaged cell fullness: %2.3f%%",
         100*(U64_TO_DBL(stats_n_data_bytes_packaged) /
@@ -128,3 +133,26 @@ log_heartbeat(time_t now)
   return 0;
 }
 
+void
+log_accounting(const time_t now, const or_options_t *options)
+{
+  or_state_t *state = get_or_state();
+  char *acc_rcvd = bytes_to_usage(state->AccountingBytesReadInInterval);
+  char *acc_sent = bytes_to_usage(state->AccountingBytesWrittenInInterval);
+  char *acc_max = bytes_to_usage(options->AccountingMax);
+  time_t interval_end = accounting_get_end_time();
+  char end_buf[ISO_TIME_LEN + 1];
+  char *remaining = NULL;
+  format_local_iso_time(end_buf, interval_end);
+  remaining = secs_to_uptime(interval_end - now);
+
+  log_notice(LD_HEARTBEAT, "Heartbeat: "
+      "Sent: %s / %s, Received: %s / %s. The "
+      "current accounting period ends on %s. That's %s left to go.",
+      acc_sent, acc_max, acc_rcvd, acc_max, end_buf, remaining);
+
+  tor_free(acc_rcvd);
+  tor_free(acc_sent);
+  tor_free(acc_max);
+  tor_free(remaining);
+}

+ 1 - 0
src/or/status.h

@@ -5,6 +5,7 @@
 #define TOR_STATUS_H
 
 int log_heartbeat(time_t now);
+void log_accounting(const time_t now, const or_options_t *options);
 
 #endif