Fix the transactor so it actually works.

[ccbs] / bigscreen / ccbs_bigscreen.cpp

diff --git a/bigscreen/ccbs_bigscreen.cpp b/bigscreen/ccbs_bigscreen.cpp
index 778720f9e90aa97037dcb67db3ce1dc4acbad52a..09aa59cdedc504eeaddc30806e283f31a0f06789 100644 (file)
--- a/bigscreen/ccbs_bigscreen.cpp
+++ b/bigscreen/ccbs_bigscreen.cpp
@@ -1,10 +1,111 @@
-#include <stdio.h>
+#include <cstdio>
 #include <unistd.h>
+#include <pqxx/pqxx>
 #include "glwindow.h"
 
+/* A trigger that sets a flag whenever it's trigged. */
+class FlagTrigger : pqxx::trigger {
+private:
+       bool flag;
+       
+public:
+       FlagTrigger(pqxx::connection_base &conn, const PGSTD::string &name)
+               : pqxx::trigger(conn, name), flag(false) {}
+       virtual ~FlagTrigger() throw () {}
+       
+       virtual void operator() (int pid)
+       {
+               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;
+       }
+};
+
+class Tournament {
+public:
+       int id;
+       std::string name;
+};
+
+/* A transactor that fetches the current tournament and some information about it. */
+class FetchCurrentTournament : public pqxx::transactor<> {
+private:
+       Tournament *tourn;
+
+public:
+       FetchCurrentTournament(Tournament *tourn) : tourn(tourn) {}
+       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);
+
+                       tourn->id = tournament["tournament"].as(tourn->id);
+                       tourn->name = tournament["tournamentname"].as(tourn->name);
+               } catch (PGSTD::out_of_range &e) {
+                       tourn->id = -1;
+                       tourn->name = "";
+               }
+       }
+};
+
+void init(pqxx::connection &conn)
+{
+       Tournament tourn;
+       conn.perform(FetchCurrentTournament(&tourn));
+
+       if (tourn.id == -1) {
+               std::fprintf(stderr, "No active tournament\n");
+       } else {
+               std::fprintf(stderr, "Current tournament is %d (name: '%s')\n",
+                       tourn.id, tourn.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);
-       sleep(5);
+       try {
+               pqxx::connection conn("dbname=ccbs host=altersex.samfundet.no user=ccbs password=GeT|>>B_");
+               FlagTrigger tournament_changed(conn, "active_tournament");
+               
+               // 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");
+               }
+       } catch (const std::exception &e) {
+               std::fprintf(stderr, "Exception: %s\n", e.what());
+               exit(1);
+       }
+       
        return 0;
 }