Add support for early destruction of TLS object

6 files changed, 66 insertions, 5 deletions

diff --git a/odb/details/posix/tls.hxx b/odb/details/posix/tls.hxx
index d23a09a..0db3d35 100644
--- a/odb/details/posix/tls.hxx
+++ b/odb/details/posix/tls.hxx
@@ -23,6 +23,9 @@ namespace odb
       T&
       get () const;
 
+      void
+      free ();
+
     private:
       tls (const tls&);
       tls& operator= (const tls&);
@@ -74,6 +77,13 @@ namespace odb
     }
 
     template <typename T>
+    inline void
+    tls_free (tls<T>& t)
+    {
+      t.free ();
+    }
+
+    template <typename T>
     inline T*
     tls_get (const tls<T*>& t)
     {
diff --git a/odb/details/posix/tls.txx b/odb/details/posix/tls.txx
index 8cc78a4..189248b 100644
--- a/odb/details/posix/tls.txx
+++ b/odb/details/posix/tls.txx
@@ -32,9 +32,7 @@ namespace odb
       if (e != 0 || error_ != 0)
         throw posix_exception (e ? e : error_);
 
-      void* v (pthread_getspecific (key_));
-
-      if (v != 0)
+      if (void* v = pthread_getspecific (key_))
         return *static_cast<T*> (v);
 
       std::auto_ptr<T> p (new T);
@@ -49,6 +47,24 @@ namespace odb
 
     template <typename T>
     void tls<T>::
+    free ()
+    {
+      int e (pthread_once (&once_, key_init));
+
+      if (e != 0 || error_ != 0)
+        throw posix_exception (e ? e : error_);
+
+      if (void* v = pthread_getspecific (key_))
+      {
+        if (e = pthread_setspecific (key_, 0))
+          throw posix_exception (e);
+
+        delete static_cast<T*> (v);
+      }
+    }
+
+    template <typename T>
+    void tls<T>::
     key_init ()
     {
       error_ = pthread_key_create (&key_, destructor);
diff --git a/odb/details/tls.hxx b/odb/details/tls.hxx
index d2627be..2ed68ab 100644
--- a/odb/details/tls.hxx
+++ b/odb/details/tls.hxx
@@ -26,6 +26,15 @@ namespace odb
       return x;
     }
 
+    // If early destructions is possible, destroy the object and free
+    // any allocated resources.
+    //
+    template <typename T>
+    inline void
+    tls_free (T& x)
+    {
+    }
+
     template <typename T>
     inline T*
     tls_get (T* p)
diff --git a/odb/details/win32/tls.cxx b/odb/details/win32/tls.cxx
index d6dc2d8..29eb5fb 100644
--- a/odb/details/win32/tls.cxx
+++ b/odb/details/win32/tls.cxx
@@ -18,6 +18,10 @@
 #include <odb/details/win32/tls-init.hxx>
 #include <odb/details/win32/exceptions.hxx>
 
+#ifdef _MSC_VER
+#  pragma warning (disable:4200) // zero-sized array in struct
+#endif
+
 using namespace std;
 
 namespace odb
diff --git a/odb/details/win32/tls.hxx b/odb/details/win32/tls.hxx
index b1dce0f..3354723 100644
--- a/odb/details/win32/tls.hxx
+++ b/odb/details/win32/tls.hxx
@@ -39,6 +39,9 @@ namespace odb
       T&
       get () const;
 
+      void
+      free ();
+
     private:
       tls (const tls&);
       tls& operator= (const tls&);
@@ -88,6 +91,13 @@ namespace odb
     }
 
     template <typename T>
+    inline void
+    tls_free (tls<T>& t)
+    {
+      t.free ();
+    }
+
+    template <typename T>
     inline T*
     tls_get (const tls<T*>& t)
     {
diff --git a/odb/details/win32/tls.txx b/odb/details/win32/tls.txx
index f20b6a9..025e7cb 100644
--- a/odb/details/win32/tls.txx
+++ b/odb/details/win32/tls.txx
@@ -26,9 +26,8 @@ namespace odb
     get () const
     {
       once_.call (key_init);
-      void* v (_get (key_));
 
-      if (v != 0)
+      if (void* v = _get (key_))
         return *static_cast<T*> (v);
 
       auto_ptr<T> p (new T);
@@ -41,6 +40,19 @@ namespace odb
 
     template <typename T>
     void tls<T>::
+    free ()
+    {
+      once_.call (key_init);
+
+      if (void* v = _get (key_))
+      {
+        _set (key_, 0);
+        delete static_cast<T*> (v);
+      }
+    }
+
+    template <typename T>
+    void tls<T>::
     key_init ()
     {
       key_ = _allocate (destructor);