Added the atom class and did some more client work, it will close the socket

now when the end has been reached.

4 files changed, 116 insertions, 0 deletions

diff --git a/src/atom.cpp b/src/atom.cpp
new file mode 100644
index 0000000..f966bfc
--- /dev/null
+++ b/src/atom.cpp
@@ -0,0 +1 @@
+#include "bu/atom.h"
diff --git a/src/atom.h b/src/atom.h
new file mode 100644
index 0000000..731e08b
--- /dev/null
+++ b/src/atom.h
@@ -0,0 +1,93 @@
+#ifndef ATOM_H
+#define ATOM_H
+
+#include <stdint.h>
+#include <memory>
+#include "bu/exceptions.h"
+
+namespace Bu
+{
+        /**
+         *
+         */
+        template <typename t, typename talloc=std::allocator<t> >
+        class Atom
+        {
+        private:
+                typedef struct Atom<t, talloc> MyType;
+
+        public:
+                Atom() :
+                        pData( NULL )
+                {
+                }
+
+                virtual ~Atom()
+                {
+                        clear();
+                }
+
+                bool isSet() const
+                {
+                        return (pData != NULL);
+                }
+
+                void set( const t &val )
+                {
+                        clear();
+                        pData = ta.allocate( 1 );
+                        ta.construct( pData, val );
+                }
+
+                t &get()
+                {
+                        if( !pData )
+                                throw Bu::ExceptionBase("Not set");
+                        return *pData;
+                }
+
+                const t &get() const
+                {
+                        if( !pData )
+                                throw Bu::ExceptionBase("Not set");
+                        return *pData;
+                }
+
+                void clear()
+                {
+                        if( pData )
+                        {
+                                ta.destroy( pData );
+                                ta.deallocate( pData, 1 );
+                                pData = NULL;
+                        }
+                }
+
+                operator const t &() const
+                {
+                        if( !pData )
+                                throw Bu::ExceptionBase("Not set");
+                        return *pData;
+                }
+                
+                operator t &()
+                {
+                        if( !pData )
+                                throw Bu::ExceptionBase("Not set");
+                        return *pData;
+                }
+
+                MyType &operator =( const t &oth )
+                {
+                        set( oth );
+
+                        return *this;
+                }
+
+        private:
+                t *pData;
+                talloc ta;
+        };
+}
+
+#endif
diff --git a/src/client.cpp b/src/client.cpp
index cf96424..6d7d81c 100644
--- a/src/client.cpp
+++ b/src/client.cpp
@@ -47,6 +47,11 @@ void Bu::Client::processInput()
                 }
         }
 
+        if( nTotal == 0 )
+        {
+                pSocket->close();
+        }
+
         if( pProto && nTotal )
         {
                 pProto->onNewData( this );
diff --git a/src/tests/atom.cpp b/src/tests/atom.cpp
new file mode 100644
index 0000000..cf076b1
--- /dev/null
+++ b/src/tests/atom.cpp
@@ -0,0 +1,17 @@
+#include "bu/atom.h"
+#include <stdio.h>
+#include <stdlib.h>
+
+int main()
+{
+        Bu::Atom<int> aInt;
+        Bu::Atom<char *> aStr;
+
+        aStr.set("Hey there, dude");
+        aInt.set( 55 );
+        int me = aInt;
+        aInt = 12;
+        printf("%d, %d\n", aInt.get(), me );
+        printf("%s\n", aStr.get() );
+}
+