git.sesse.net Git - cubemap/blob - stats.cpp

   1 #include <fcntl.h>
   2 #include <stddef.h>
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <string.h>
   6 #include <time.h>
   7 #include <unistd.h>
   8 #include <vector>
   9
  10 #include "client.h"
  11 #include "log.h"
  12 #include "serverpool.h"
  13 #include "stats.h"
  14 #include "util.h"
  15
  16 using namespace std;
  17
  18 extern ServerPool *servers;
  19
  20 StatsThread::StatsThread(const std::string &stats_file, int stats_interval)
  21         : stats_file(stats_file),
  22           stats_interval(stats_interval)
  23 {
  24 }
  25
  26 void StatsThread::do_work()
  27 {
  28         while (!should_stop()) {
  29                 int fd;
  30                 char *filename;
  31                 FILE *fp;
  32                 timespec now;
  33                 vector<ClientStats> client_stats;
  34
  35                 if (clock_gettime(CLOCK_MONOTONIC_COARSE, &now) == -1) {
  36                         log_perror("clock_gettime(CLOCK_MONOTONIC_COARSE)");
  37                         goto sleep;
  38                 }
  39
  40                 // Open a new, temporary file.
  41                 filename = strdup((stats_file + ".new.XXXXXX").c_str());
  42                 fd = mkostemp(filename, O_WRONLY);
  43                 if (fd == -1) {
  44                         log_perror(filename);
  45                         free(filename);
  46                         goto sleep;
  47                 }
  48
  49                 fp = fdopen(fd, "w");
  50                 if (fp == NULL) {
  51                         log_perror("fdopen");
  52                         safe_close(fd);
  53                         if (unlink(filename) == -1) {
  54                                 log_perror(filename);
  55                         }
  56                         free(filename);
  57                         goto sleep;
  58                 }
  59
  60                 client_stats = servers->get_client_stats();
  61                 for (size_t i = 0; i < client_stats.size(); ++i) {
  62                         fprintf(fp, "%s %d %d %s %d %llu %llu %llu\n",
  63                                 client_stats[i].remote_addr.c_str(),
  64                                 client_stats[i].sock,
  65                                 0,  // Used to be fwmark.
  66                                 client_stats[i].url.c_str(),
  67                                 int(now.tv_sec - client_stats[i].connect_time.tv_sec),  // Rather coarse.
  68                                 (long long unsigned)(client_stats[i].bytes_sent),
  69                                 (long long unsigned)(client_stats[i].bytes_lost),
  70                                 (long long unsigned)(client_stats[i].num_loss_events));
  71                 }
  72                 if (fclose(fp) == EOF) {
  73                         log_perror("fclose");
  74                         if (unlink(filename) == -1) {
  75                                 log_perror(filename);
  76                         }
  77                         free(filename);
  78                         goto sleep;
  79                 }
  80
  81                 if (rename(filename, stats_file.c_str()) == -1) {
  82                         log_perror("rename");
  83                         if (unlink(filename) == -1) {
  84                                 log_perror(filename);
  85                         }
  86                 }
  87                 free(filename);
  88
  89 sleep:
  90                 // Wait until we are asked to quit, stats_interval timeout,
  91                 // or a spurious signal. (The latter will cause us to write stats
  92                 // too often, but that's okay.)
  93                 timespec timeout_ts;
  94                 timeout_ts.tv_sec = stats_interval;
  95                 timeout_ts.tv_nsec = 0;
  96                 wait_for_wakeup(&timeout_ts);
  97         }
  98 }