-! 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
-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 dup empty? [ drop ] [ peek rethrow ] if ;
+ [ 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
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
+]