Updates to protocolgats...it should work now.

2 files changed, 46 insertions, 1 deletions

diff --git a/src/protocolgats.cpp b/src/protocolgats.cpp
index fbb25a5..2ef6079 100644
--- a/src/protocolgats.cpp
+++ b/src/protocolgats.cpp
@@ -1,14 +1,53 @@
 #include "gats/protocolgats.h"
+#include "gats/gatsstream.h"
 
-Gats::ProtocolGats::ProtocolGats()
+#include <bu/client.h>
+
+Gats::ProtocolGats::ProtocolGats() :
+        pStream( NULL ),
+        pUsedClient( NULL )
 {
 }
 
 Gats::ProtocolGats::~ProtocolGats()
 {
+        delete pStream;
+        pStream = NULL;
+}
+
+void Gats::ProtocolGats::onNewConnection( Bu::Client *pClient )
+{
+        if( pStream == NULL )
+        {
+                pStream = new Gats::GatsStream( *pClient );
+                pUsedClient = pClient;
+        }
 }
 
 void Gats::ProtocolGats::onNewData( Bu::Client *pClient )
 {
+        if( pStream == NULL )
+        {
+                pStream = new Gats::GatsStream( *pClient );
+                pUsedClient = pClient;
+        }
+        else if( pClient != pUsedClient )
+        {
+                throw Bu::ExceptionBase("ProtocolGats requires that you maintain a "
+                        "1:1 relationship between client and protocol objects.");
+        }
+
+        for(;;)
+        {
+                Gats::Object *pObj = pStream->readObject();
+                if( pObj == NULL )
+                        break;
+                onNewObject( pObj );
+        }
+}
+
+void Gats::ProtocolGats::writeObject( Gats::Object *pObj )
+{
+        pStream->writeObject( pObj );
 }
 
diff --git a/src/protocolgats.h b/src/protocolgats.h
index dcd0c55..e261ab5 100644
--- a/src/protocolgats.h
+++ b/src/protocolgats.h
@@ -6,6 +6,7 @@
 namespace Gats
 {
         class Object;
+        class GatsStream;
 
         class ProtocolGats : public Bu::Protocol
         {
@@ -13,11 +14,16 @@ namespace Gats
                 ProtocolGats();
                 virtual ~ProtocolGats();
 
+                virtual void onNewConnection( Bu::Client *pClient );
                 virtual void onNewData( Bu::Client *pClient );
 
                 virtual void onNewObject( Gats::Object *pObj )=0;
 
+                void writeObject( Gats::Object *pObj );
+
         private:
+                Gats::GatsStream *pStream;
+                Bu::Client *pUsedClient;
         };
 };