Updates to protocolgats...it should work now.

1 files changed, 40 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 );
 }