Sticky drops from the registry eagerly now

CHANGELOG.md

@@ -7,6 +7,8 @@ All notable changes to similar are documented here.
 * `Fragile` no longer boxes internally.
 * `Sticky` and `SemiSticky` now require the use of stack tokens.
   For more information see [#26](https://github.com/mitsuhiko/fragile/issues/26)
+* `Sticky` now tries to drop entries from the thread local registry eagerly
+  if it's dropped on the right thread.
 
 ## 1.2.1
 

src/registry.rs

@@ -34,6 +34,11 @@ mod slab_impl {
         let _ = thread_id;
         REGISTRY.with(|registry| unsafe { (*registry.get()).0.remove(item_id) })
     }
+
+    pub fn try_remove(item_id: ItemId, thread_id: NonZeroUsize) -> Option<Entry> {
+        let _ = thread_id;
+        REGISTRY.with(|registry| unsafe { (*registry.get()).0.try_remove(item_id) })
+    }
 }
 
 #[cfg(not(feature = "slab"))]
@@ -76,6 +81,10 @@ mod map_impl {
         REGISTRY
             .with(|registry| unsafe { (*registry.get()).0.remove(&(thread_id, item_id)).unwrap() })
     }
+
+    pub fn try_remove(item_id: ItemId, thread_id: NonZeroUsize) -> Option<Entry> {
+        REGISTRY.with(|registry| unsafe { (*registry.get()).0.remove(&(thread_id, item_id)) })
+    }
 }
 
 #[cfg(feature = "slab")]

src/sticky.rs

@@ -35,12 +35,24 @@ pub struct Sticky<T: 'static> {
 
 impl<T> Drop for Sticky<T> {
     fn drop(&mut self) {
+        // if the type needs dropping we can only do so on the
+        // right thread.  worst case we leak the value until the
+        // thread dies.
         if mem::needs_drop::<T>() {
             unsafe {
                 if self.is_valid() {
                     self.unsafe_take_value();
                 }
             }
+
+        // otherwise we take the liberty to drop the value
+        // right here and now.  We can however only do that if
+        // we are on the right thread.  If we are not, we again
+        // need to wait for the thread to shut down.
+        } else if let Some(entry) = registry::try_remove(self.item_id, self.thread_id) {
+            unsafe {
+                (entry.drop)(entry.ptr);
+            }
         }
     }
 }