Added more helper guys to atom, it seems weird, but I think it's ok...

2 files changed, 22 insertions, 0 deletions

diff --git a/src/atom.h b/src/atom.h
index f876274..a0469b6 100644
--- a/src/atom.h
+++ b/src/atom.h
@@ -84,6 +84,20 @@ namespace Bu
                         return *this;
                 }
 
+                t *operator ->()
+                {
+                        if( !pData )
+                                throw Bu::ExceptionBase("Not set");
+                        return pData;
+                }
+
+                t &operator *()
+                {
+                        if( !pData )
+                                throw Bu::ExceptionBase("Not set");
+                        return *pData;
+                }
+
         private:
                 t *pData;
                 talloc ta;
diff --git a/src/tests/atom.cpp b/src/tests/atom.cpp
index cf076b1..2077bfd 100644
--- a/src/tests/atom.cpp
+++ b/src/tests/atom.cpp
@@ -2,10 +2,18 @@
 #include <stdio.h>
 #include <stdlib.h>
 
+typedef struct bob
+{
+        int a, b;
+} bob;
 int main()
 {
         Bu::Atom<int> aInt;
         Bu::Atom<char *> aStr;
+        Bu::Atom<bob> aBob;
+
+        aBob = bob();
+        aBob->a = 5;
 
         aStr.set("Hey there, dude");
         aInt.set( 55 );