Split up commits and requests to seperate functions. Changed Database * to Database &

1 files changed, 161 insertions, 144 deletions

diff --git a/server/src/server.cc b/server/src/server.cc
index 8bfd827..1bfa2b0 100644
--- a/server/src/server.cc
+++ b/server/src/server.cc
@@ -79,186 +79,203 @@ public:
   : Exception("Macro " + r.macro + " not found in course " + r.course) {}
 };
 
-static std::string handleTransaction(Transaction *transaction,
-                                     TCPSocket *pentominos_socket,
-                                     Database *db,
-                                     JournalWriter &journalwriter,
-                                     MacroList &macrolist)
+
+static std::string handleCommits(Transaction *transaction, Database &db, 
+                                 JournalWriter &journalwriter, MacroList &macrolist)
 {
   std::string answer;
-  answer += "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n";
-  answer += "<pracro version=\"1.0\">\n";
-
-  try {
-    //
-    // Handle commits
-    //
-    {
-      Commits::iterator i = transaction->commits.begin();
-      while(i != transaction->commits.end()) {
-        Commit &commit = *i;
 
-        MacroParser mp(macrolist.getLatestVersion(commit.macro));
-        mp.parse();
-        Macro *macro = mp.getMacro();
-
-        std::string resume = resume_parser(macro->resume, commit);
-        commit.fields["journal.resume"] = resume;
-        db->commitTransaction(transaction->user, transaction->cpr, *macro, commit.fields);
-        
-        bool store_in_journal = true;
-        // We always need to store in journal!
-        // macro->resume.attributes.find("store_in_journal") != macro->resume.attributes.end() &&
-        // macro->resume.attributes["store_in_journal"] == "true";
-
-        if(resume != "" && store_in_journal) {
-          journalwriter.addEntry(*transaction, commit, resume, commit.course);
-        }
-
-        i++;
-      }
+  Commits::iterator i = transaction->commits.begin();
+  while(i != transaction->commits.end()) {
+    Commit &commit = *i;
+    
+    MacroParser mp(macrolist.getLatestVersion(commit.macro));
+    mp.parse();
+    Macro *macro = mp.getMacro();
+    
+    std::string resume = resume_parser(macro->resume, commit);
+    commit.fields["journal.resume"] = resume;
+    db.commitTransaction(transaction->user, transaction->cpr, *macro, commit.fields);
+    
+    if(resume != "") {
+      journalwriter.addEntry(*transaction, commit, resume, commit.course);
     }
-        
-    //
-    // Handle requests
-    //
-    Requests::iterator i = transaction->requests.begin();
-    while(i != transaction->requests.end()) {
-      Request &request = *i;
-
-      PRACRO_DEBUG(server, "Handling request - macro: %s, course: %s\n",
-             request.macro.c_str(), request.course.c_str());
-
-      // Read and parse the template file.
-      TemplateParser tp(request.course);
-      tp.parse();
-      
-      Template *templ = tp.getTemplate();
+    
+    i++;
+  }
 
-      answer += "  <course name=\"";
-      answer += templ->course.attributes["name"];
-      answer += "\" title=\"";
-      answer += templ->course.attributes["title"];
-      answer += "\">\n";
-      
-      bool foundmacro = false;
+  return answer;
+}
 
-      // Generate the macro and return it to the client
-      std::vector< Macro >::iterator mi2 = templ->course.macros.begin();
-      while(mi2 != templ->course.macros.end()) {
-        Macro &macro = (*mi2);
 
-        if(macro.isHeader) {
-          answer += "    <header caption=\"" + macro.attributes["caption"] + "\"/>\n";
-          mi2++;
-          continue;
-        }
-
-        bool completed = db->checkMacro(transaction->cpr, macro.attributes["name"]);
+static std::string handleRequest(Transaction *transaction,
+                                 TCPSocket *pentominos_socket,
+                                 Database &db,
+                                 JournalWriter &journalwriter,
+                                 MacroList &macrolist)
+{
+  std::string answer;
 
-        answer += "    <macro completed=";
-        if(completed) answer += "\"true\"";
-        else answer += "\"false\"";
+  Requests::iterator i = transaction->requests.begin();
+  while(i != transaction->requests.end()) {
+    Request &request = *i;
+    
+    PRACRO_DEBUG(server, "Handling request - macro: %s, course: %s\n",
+                 request.macro.c_str(), request.course.c_str());
+    
+    // Read and parse the template file.
+    TemplateParser tp(request.course);
+    tp.parse();
+    
+    Template *templ = tp.getTemplate();
+    
+    answer += "  <course name=\"";
+    answer += templ->course.attributes["name"];
+    answer += "\" title=\"";
+    answer += templ->course.attributes["title"];
+    answer += "\">\n";
+    
+    bool foundmacro = false;
+    
+    // Generate the macro and return it to the client
+    std::vector< Macro >::iterator mi2 = templ->course.macros.begin();
+    while(mi2 != templ->course.macros.end()) {
+      Macro &macro = (*mi2);
 
-        std::map< std::string, std::string >::iterator ai = macro.attributes.begin();
-        while(ai != macro.attributes.end()) {
-          std::string name = ai->first;
-          std::string value = ai->second;
-          answer += " "+name+"=\"" + value + "\"";
-          ai++;
-        }
+      if(macro.isHeader) {
+        answer += "    <header caption=\"" + macro.attributes["caption"] + "\"/>\n";
+        mi2++;
+        continue;
+      }
 
-        if(macro.attributes["name"] == request.macro || 
-           (macro.attributes.find("static") != macro.attributes.end() &&
-            macro.attributes["static"] == "true")
-           ) {
-          foundmacro = true;
+      bool completed = db.checkMacro(transaction->cpr, macro.attributes["name"]);
 
-          MacroParser mp(macrolist.getLatestVersion(macro.attributes["name"]));
-          mp.parse();
-          Macro *m = mp.getMacro();
-          answer += " caption=\"" + m->window.attributes["caption"] + "\"";
-          answer += ">\n";
+      answer += "    <macro completed=";
+      if(completed) answer += "\"true\"";
+      else answer += "\"false\"";
 
-          LUAQueryMapper lqm;
+      std::map< std::string, std::string >::iterator ai = macro.attributes.begin();
+      while(ai != macro.attributes.end()) {
+        std::string name = ai->first;
+        std::string value = ai->second;
+        answer += " "+name+"=\"" + value + "\"";
+        ai++;
+      }
 
-          ////////////////////////
-          std::vector< Query >::iterator qi = m->queries.begin();
-          while(qi != m->queries.end()) {
+      if(macro.attributes["name"] == request.macro || 
+         (macro.attributes.find("static") != macro.attributes.end() &&
+          macro.attributes["static"] == "true")
+         ) {
+        foundmacro = true;
 
-            Query &query = *qi;
-            std::string service = query.attributes["service"];
+        MacroParser mp(macrolist.getLatestVersion(macro.attributes["name"]));
+        mp.parse();
+        Macro *m = mp.getMacro();
+        answer += " caption=\"" + m->window.attributes["caption"] + "\"";
+        answer += ">\n";
 
-            if(service == "pentominos") {
-              // Send the queries to Pentominos (if any)
-              QueryHandlerPentominos qh(pentominos_socket, transaction->cpr);
+        LUAQueryMapper lqm;
 
-              QueryResult queryresult = qh.exec(*qi);
-              lqm.addQueryResult(queryresult);
-            }
+        ////////////////////////
+        std::vector< Query >::iterator qi = m->queries.begin();
+        while(qi != m->queries.end()) {
 
-            if(service == "pracro") {
-              // Send the queries to Pentominos (if any)
-              QueryHandlerPracro qh(db, transaction->cpr);
+          Query &query = *qi;
+          std::string service = query.attributes["service"];
 
-              QueryResult queryresult = qh.exec(*qi);
-              lqm.addQueryResult(queryresult);
-            }
+          if(service == "pentominos") {
+            // Send the queries to Pentominos (if any)
+            QueryHandlerPentominos qh(pentominos_socket, transaction->cpr);
 
-            qi++;
+            QueryResult queryresult = qh.exec(*qi);
+            lqm.addQueryResult(queryresult);
           }
 
-          // Handle scripts
-          if(m->scripts.size()) {
-            answer += "      <scripts>\n";
-            
-            std::vector< Script >::iterator spi = m->scripts.begin();
-            while(spi != m->scripts.end()) {
-              answer += "        <script language=\"" + spi->attributes["language"] 
-                + "\" name=\"" + spi->attributes["name"] + "\">\n";
-              answer += xml_encode(spi->attributes["code"]);
-              answer += "\n        </script>\n";
-              spi++;
-            }
-            answer += "      </scripts>\n";
-          }
+          if(service == "pracro") {
+            // Send the queries to Pentominos (if any)
+            QueryHandlerPracro qh(db, transaction->cpr);
 
-          answer += widgetgenerator(transaction->cpr, *m, lqm, db);
-        } else {
-          // only find macro title
-          MacroParser mp(macrolist.getLatestVersion(macro.attributes["name"]));
-          mp.parse();
-          Macro *m = mp.getMacro();
-          answer += " caption=\"" + m->window.attributes["caption"] + "\"";
-          answer += ">\n";
+            QueryResult queryresult = qh.exec(*qi);
+            lqm.addQueryResult(queryresult);
+          }
 
+          qi++;
         }
 
-        if(completed) {
-          answer += "      <resume>";
-          answer += db->getResume(transaction->cpr, macro, time(NULL) - Conf::db_max_ttl);
-          answer += "</resume>\n";
+        // Handle scripts
+        if(m->scripts.size()) {
+          answer += "      <scripts>\n";
+            
+          std::vector< Script >::iterator spi = m->scripts.begin();
+          while(spi != m->scripts.end()) {
+            answer += "        <script language=\"" + spi->attributes["language"] 
+              + "\" name=\"" + spi->attributes["name"] + "\">\n";
+            answer += xml_encode(spi->attributes["code"]);
+            answer += "\n        </script>\n";
+            spi++;
+          }
+          answer += "      </scripts>\n";
         }
 
-        answer += "    </macro>\n";
-        mi2++;
+        answer += widgetgenerator(transaction->cpr, *m, lqm, db);
+      } else {
+        // only find macro title
+        MacroParser mp(macrolist.getLatestVersion(macro.attributes["name"]));
+        mp.parse();
+        Macro *m = mp.getMacro();
+        answer += " caption=\"" + m->window.attributes["caption"] + "\"";
+        answer += ">\n";
 
       }
-      
-      if(foundmacro == false && request.macro != "")
-        throw NotFoundException(request);
 
-      answer += "  </course>\n";
+      if(completed) {
+        answer += "      <resume>";
+        answer += db.getResume(transaction->cpr, macro, time(NULL) - Conf::db_max_ttl);
+        answer += "</resume>\n";
+      }
+
+      answer += "    </macro>\n";
+      mi2++;
 
-      i++;
     }
+      
+    if(foundmacro == false && request.macro != "")
+      throw NotFoundException(request);
+
+    answer += "  </course>\n";
+
+    i++;
+  }
 
-    answer += "</pracro>\n";
+  return answer;
+}
 
+static std::string handleTransaction(Transaction *transaction,
+                                     TCPSocket *pentominos_socket,
+                                     Database &db,
+                                     JournalWriter &journalwriter,
+                                     MacroList &macrolist)
+{
+  std::string answer;
+  answer += "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n";
+  answer += "<pracro version=\"1.0\">\n";
+
+  try {
+    answer += handleCommits(transaction, db, journalwriter, macrolist);
   } catch( std::exception &e ) {
-    answer = error_box(xml_encode(e.what()));
+    PRACRO_ERR(server, "Commit error: %s\n", e.what());
+    return error_box(xml_encode(e.what()));
   }
 
+  try {
+    answer += handleRequest(transaction, pentominos_socket, db, journalwriter, macrolist);
+  } catch( std::exception &e ) {
+    PRACRO_ERR(server, "Request error: %s\n", e.what());
+    return error_box(xml_encode(e.what()));
+  }
+
+  answer += "</pracro>\n";
+
   PRACRO_DEBUG(server, "Done handling transaction\n");
   PRACRO_DEBUG(serverxml, "%s\n", answer.c_str());
   return answer;
@@ -272,7 +289,7 @@ static void handleConnection(TCPSocket *socket)
   pentominos_socket.connect(Conf::pentominos_addr, Conf::pentominos_port);
 #endif/*WITHOUT_PENTOMINOS*/
 
-  Database *db = new Database("pgsql", Conf::database_addr, "", Conf::database_user, Conf::database_passwd, "");
+  Database db("pgsql", Conf::database_addr, "", Conf::database_user, Conf::database_passwd, "");
 
   JournalWriter journalwriter(Conf::journal_commit_addr.c_str(), Conf::journal_commit_port);