1 files changed, 8 insertions, 2 deletions
diff --git a/server/src/connection.h b/server/src/connection.h
index f91ddc1..61997da 100644
--- a/server/src/connection.h
+++ b/server/src/connection.h
@@ -33,9 +33,11 @@
 #include "transaction.h"
 #include "transactionparser.h"
 
+class Session;
+
 class Connection {
 public:
-  Connection(Environment &e, std::string sessionid, bool commit);
+  Connection(Environment &e, std::string sessionid, bool commit, bool discard);
   ~Connection();
 
   bool handle(const char *data, size_t size);
@@ -44,8 +46,12 @@ public:
   std::string getSessionID();
 
 private:
+  void commit(Session *session);
+  void discard(Session *session);
+
   std::string sessionid;
-  bool commit;
+  bool docommit;
+  bool dodiscard;
   Environment &env;
 
   Transaction transaction;