Added a transactor for getting info from the current tournament.

[ccbs] / bigscreen / ccbs_bigscreen.cpp

diff --git a/bigscreen/ccbs_bigscreen.cpp b/bigscreen/ccbs_bigscreen.cpp
index 03a94962ddab87ff4c0fe0200ed8154bfe7649bb..c9cb82d56f147efa3e2c6dc5fb868a7211c69f2c 100644 (file)
--- a/bigscreen/ccbs_bigscreen.cpp
+++ b/bigscreen/ccbs_bigscreen.cpp
@@ -3,43 +3,112 @@
 #include <pqxx/pqxx>
 #include "glwindow.h"
 
-/*
- * A trigger that exits whenever it's trigged (used when we change big things
- * such as starting a new tournament, and it's not really worth doing a clean
- * change).
- */
-class ExitTrigger : pqxx::trigger {
+/* A trigger that sets a flag whenever it's trigged. */
+class FlagTrigger : pqxx::trigger {
+private:
+       bool flag;
+       
 public:
-       ExitTrigger(pqxx::connection_base &conn, const PGSTD::string &name)
-               : pqxx::trigger(conn, name) {}
-       virtual ~ExitTrigger() throw () {}
+       FlagTrigger(pqxx::connection_base &conn, const PGSTD::string &name)
+               : pqxx::trigger(conn, name), flag(false) {}
+       virtual ~FlagTrigger() throw () {}
        
        virtual void operator() (int pid)
        {
-               std::fprintf(stderr, "Received an exit trigger from pid %u\n", pid);
-               exit(0);
+               flag = true;
+               std::fprintf(stderr, "Received a flag trigger from pid %u\n", pid);
+       }
+
+       bool get_flag() const
+       {
+               return flag;
+       }
+
+       void reset_flag()
+       {
+               flag = false;
+       }
+};
+
+/* A transactor that fetches the current tournament and some information about it. */
+class FetchCurrentTournament : public pqxx::transactor<> {
+private:
+       int id;
+       std::string name;
+
+public:
+       FetchCurrentTournament() {}
+       void operator() (pqxx::transaction<> &t)
+       {
+               pqxx::result res( t.exec("SELECT * FROM bigscreen.active_tournament NATURAL JOIN tournaments") );
+               try {
+                       pqxx::result::tuple tournament = res.at(0);
+
+                       id = tournament["tournament"].as(id);
+                       name = tournament["tournament"].as(name);
+               } catch (PGSTD::out_of_range &e) {
+                       std::fprintf(stderr, "foof\n");
+                       
+                       id = -1;
+                       name = "";
+               }
+       }
+
+       int get_tournament_id() const
+       {
+               return id;
+       }
+       
+       std::string get_tournament_name() const
+       {
+               return name;
        }
 };
 
+void init(pqxx::connection &conn)
+{
+       FetchCurrentTournament fct;
+       conn.perform(fct);
+
+       if (fct.get_tournament_id() == -1) {
+               std::fprintf(stderr, "No active tournament\n");
+       } else {
+               std::fprintf(stderr, "Current tournament is %d (name: '%s')\n",
+                       fct.get_tournament_id(), fct.get_tournament_name().c_str());
+       }
+}
+
+void main_loop(pqxx::connection &conn)
+{
+       pqxx::work t(conn, "trx");
+
+       // fetch all songs
+       pqxx::result res( t.exec("SELECT * FROM songs") );
+       for (pqxx::result::const_iterator i = res.begin(); i != res.end(); ++i) {
+               // std::fprintf(stderr, "%s\n", i["title"].c_str());
+       }
+       t.commit();
+       
+       sleep(1);
+}
+
 int main(int argc, char **argv)
 {
        GLWindow glw("CCBS bigscreen", 800, 600, 32, false, 16, -1);
        try {
                pqxx::connection conn("dbname=ccbs host=altersex.samfundet.no user=ccbs password=GeT|>>B_");
-               ExitTrigger et(conn, "bs_tournament_changed");
+               FlagTrigger tournament_changed(conn, "active_tournament");
                
-               sleep(1);
-
-               pqxx::work t(conn, "trx");
-
-               // fetch all songs
-               pqxx::result res( t.exec("SELECT * FROM songs") );
-               for (pqxx::result::const_iterator i = res.begin(); i != res.end(); ++i) {
-                       std::fprintf(stderr, "%s\n", i["title"].c_str());
+               // when active_tournament is changed, we destroy everything and start from scratch
+               for ( ;; ) {
+                       tournament_changed.reset_flag();
+                       init(conn);
+                       do {
+                               main_loop(conn);
+                               conn.get_notifs();
+                       } while (!tournament_changed.get_flag());
+                       std::fprintf(stderr, "active_tournament changed, resetting...\n");
                }
-               t.commit();
-               conn.get_notifs();
-               sleep(1);
        } catch (const std::exception &e) {
                std::fprintf(stderr, "Exception: %s\n", e.what());
                exit(1);