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 3b9d6ae506ca8e11ebc4df62a7b8334dbb02948e..c9cb82d56f147efa3e2c6dc5fb868a7211c69f2c 100644 (file)
--- a/bigscreen/ccbs_bigscreen.cpp
+++ b/bigscreen/ccbs_bigscreen.cpp
@@ -30,6 +30,54 @@ public:
        }
 };
 
+/* 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");
@@ -54,6 +102,7 @@ int main(int argc, char **argv)
                // 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();