core: trim using lists with lint.vocabs tool

[factor.git] / core / destructors / destructors.factor

diff --git a/core/destructors/destructors.factor b/core/destructors/destructors.factor
old mode 100755 (executable)
new mode 100644 (file)
index afc956f..1ed1ce8
--- a/core/destructors/destructors.factor
+++ b/core/destructors/destructors.factor
@@ -1,30 +1,65 @@
-! Copyright (C) 2007, 2008 Doug Coleman, Slava Pestov.
+! Copyright (C) 2007, 2010 Doug Coleman, Slava Pestov.
 ! See http://factorcode.org/license.txt for BSD license.
-USING: accessors continuations kernel namespaces make
-sequences vectors ;
+USING: accessors assocs continuations kernel namespaces
+sequences sets ;
 IN: destructors
 
-TUPLE: disposable disposed ;
+SYMBOL: disposables
+
+ERROR: already-unregistered disposable ;
+
+SYMBOL: debug-leaks?
+
+<PRIVATE
+
+SLOT: continuation
+
+: register-disposable ( obj -- )
+    debug-leaks? get-global [ current-continuation >>continuation ] when
+    disposables get adjoin ;
+
+: unregister-disposable ( obj -- )
+    dup disposables get ?delete [ drop ] [ already-unregistered ] if ;
+
+PRIVATE>
+
+TUPLE: disposable < identity-tuple
+{ disposed boolean }
+continuation ;
+
+: new-disposable ( class -- disposable )
+    new dup register-disposable ; inline
 
 GENERIC: dispose* ( disposable -- )
 
 ERROR: already-disposed disposable ;
 
-: check-disposed ( disposable -- )
-    dup disposed>> [ already-disposed ] [ drop ] if ; inline
+: check-disposed ( disposable -- disposable )
+    dup disposed>> [ already-disposed ] when ; inline
 
 GENERIC: dispose ( disposable -- )
 
-M: object dispose
-    dup disposed>> [ drop ] [ t >>disposed dispose* ] if ;
+: unless-disposed ( disposable quot -- )
+    [ dup disposed>> [ drop ] ] dip if ; inline
 
-: dispose-each ( seq -- )
+M: object dispose [ t >>disposed dispose* ] unless-disposed ;
+
+M: disposable dispose
     [
-        [ [ dispose ] curry [ , ] recover ] each
-    ] { } make [ peek rethrow ] unless-empty ;
+        [ unregister-disposable ]
+        [ call-next-method ]
+        bi
+    ] unless-disposed ;
+
+: dispose-to ( obj accum -- )
+    [ dispose ] [ push ] bi-curry* recover ; inline
+
+: dispose-each ( seq -- )
+    V{ } clone [ [ dispose-to ] curry each ] keep
+    [ last rethrow ] unless-empty ;
 
 : with-disposal ( object quot -- )
-    over [ dispose ] curry [ ] cleanup ; inline
+    over [ dispose ] curry finally ; inline
 
 <PRIVATE
 
@@ -47,10 +82,21 @@ PRIVATE>
     dup error-destructors get push ; inline
 
 : with-destructors ( quot -- )
-    [
-        V{ } clone always-destructors set
-        V{ } clone error-destructors set
+    H{ } clone
+    V{ } clone always-destructors pick set-at
+    V{ } clone error-destructors pick set-at [
         [ do-always-destructors ]
         [ do-error-destructors ]
         cleanup
-    ] with-scope ; inline
+    ] with-variables ; inline
+
+STARTUP-HOOK: [
+    HS{ } clone disposables set-global
+    V{ } clone always-destructors set-global
+    V{ } clone error-destructors set-global
+]
+
+SHUTDOWN-HOOK: [
+    do-always-destructors
+    do-error-destructors
+]