Minor nit.

[cubemap] / main.cpp

diff --git a/main.cpp b/main.cpp
index dc52bdc29a17ab85f1ad1d5c8f23ef34a54c0604..69f0e9b7cb686b19293a4aef184e6b54941584b7 100644 (file)
--- a/main.cpp
+++ b/main.cpp
@@ -25,6 +25,7 @@
 #include "server.h"
 #include "serverpool.h"
 #include "input.h"
+#include "stats.h"
 #include "state.pb.h"
 
 using namespace std;
@@ -37,68 +38,6 @@ void hup(int ignored)
        hupped = true;
 }
 
-struct StatsThreadParameters {
-       string stats_file;
-       int stats_interval;
-};
-               
-void *stats_thread_run(void *arg)
-{
-       const StatsThreadParameters *parms = reinterpret_cast<StatsThreadParameters *>(arg);
-       while (!hupped) {
-               int fd;
-               FILE *fp;
-               time_t now;
-               vector<ClientStats> client_stats;
-
-               // Open a new, temporary file.
-               char *filename = strdup((parms->stats_file + ".new.XXXXXX").c_str());
-               fd = mkostemp(filename, O_WRONLY);
-               if (fd == -1) {
-                       perror(filename);
-                       free(filename);
-                       goto sleep;
-               }
-
-               fp = fdopen(fd, "w");
-               if (fp == NULL) {
-                       perror("fdopen");
-                       close(fd);
-                       unlink(filename);
-                       free(filename);
-                       goto sleep;
-               }
-
-               now = time(NULL);
-               client_stats = servers->get_client_stats();
-               for (size_t i = 0; i < client_stats.size(); ++i) {
-                       fprintf(fp, "%s %s %d %llu\n",
-                               client_stats[i].remote_addr.c_str(),
-                               client_stats[i].stream_id.c_str(),
-                               int(now - client_stats[i].connect_time),
-                               (long long unsigned)(client_stats[i].bytes_sent));
-               }
-               if (fclose(fp) == EOF) {
-                       perror("fclose");
-                       unlink(filename);
-                       free(filename);
-                       goto sleep;
-               }
-               
-               if (rename(filename, parms->stats_file.c_str()) == -1) {
-                       perror("rename");
-                       unlink(filename);
-               }
-
-sleep:
-               int left_to_sleep = parms->stats_interval;
-               do {
-                       left_to_sleep = sleep(left_to_sleep);
-               } while (left_to_sleep > 0 && !hupped);
-       }
-       return NULL;
-}
-
 // Serialize the given state to a file descriptor, and return the (still open)
 // descriptor.
 int make_tempfile(const CubemapStateProto &state)
@@ -319,11 +258,15 @@ int main(int argc, char **argv)
        if (stats_interval != -1 && stats_file.empty()) {
                fprintf(stderr, "WARNING: 'stats_interval' given, but no 'stats_file'. No statistics will be written.\n");
        }
+       StatsThread *stats_thread = NULL;
+       if (!stats_file.empty()) {
+               stats_thread = new StatsThread(stats_file, stats_interval);
+       }
 
        servers->run();
 
-       pthread_t acceptor_thread;
-       pthread_create(&acceptor_thread, NULL, acceptor_thread_run, reinterpret_cast<void *>(server_sock));
+       AcceptorThread acceptor_thread(server_sock);
+       acceptor_thread.run();
 
        // Find all streams in the configuration file, and create inputs for them.
        vector<Input *> inputs;
@@ -359,6 +302,9 @@ int main(int argc, char **argv)
                inputs.push_back(input);
        }
        
+       // All deserialized inputs should now have been taken care of, one way or the other.
+       assert(deserialized_inputs.empty());
+       
        if (is_reexec) {        
                // Put back the existing clients. It doesn't matter which server we
                // allocate them to, so just do round-robin. However, we need to add
@@ -368,16 +314,9 @@ int main(int argc, char **argv)
                }
        }
 
-       // All deserialized inputs should now have been taken care of, one way or the other.
-       assert(deserialized_inputs.empty());
-
        // Start writing statistics.
-       pthread_t stats_thread;
-       StatsThreadParameters stats_parameters;  // Must live for as long as the stats thread does.
-       if (!stats_file.empty()) {
-               stats_parameters.stats_file = stats_file;
-               stats_parameters.stats_interval = stats_interval;
-               pthread_create(&stats_thread, NULL, stats_thread_run, &stats_parameters);
+       if (stats_thread != NULL) {
+               stats_thread->run();
        }
 
        signal(SIGHUP, hup);
@@ -400,18 +339,10 @@ int main(int argc, char **argv)
        // OK, we've been HUPed. Time to shut down everything, serialize, and re-exec.
        gettimeofday(&serialize_start, NULL);
 
-       if (!stats_file.empty()) {
-               pthread_kill(stats_thread, SIGHUP);
-               if (pthread_join(stats_thread, NULL) == -1) {
-                       perror("pthread_join");
-                       exit(1);
-               }
-       }
-       pthread_kill(acceptor_thread, SIGHUP);
-       if (pthread_join(acceptor_thread, NULL) == -1) {
-               perror("pthread_join");
-               exit(1);
+       if (stats_thread != NULL) {
+               stats_thread->stop();
        }
+       acceptor_thread.stop();
 
        CubemapStateProto state;
        state.set_serialize_start_sec(serialize_start.tv_sec);