! Copyright (C) 2010 Dmitry Shubin.
! See http://factorcode.org/license.txt for BSD license.
-USING: accessors alien.c-types alien.data alien.destructors assocs
-biassocs classes.struct combinators destructors gdbm.ffi io.backend
-kernel libc literals math namespaces sequences serialize strings ;
+USING: accessors alien.c-types alien.data alien.destructors
+alien.enums alien.syntax classes.struct combinators destructors
+gdbm.ffi io.backend kernel libc locals math namespaces sequences
+serialize strings ;
IN: gdbm
+ENUM: gdbm-role reader writer wrcreat newdb ;
+
TUPLE: gdbm
- { name string }
+ { name string }
{ block-size integer }
- { flags integer initial: $ GDBM_WRCREAT }
- { mode integer initial: OCT: 644 } ;
-
-SINGLETONS:
- gdbm-no-error gdbm-malloc-error
- gdbm-block-size-error gdbm-file-open-error
- gdbm-file-write-error gdbm-file-seek-error
- gdbm-file-read-error gdbm-bad-magic-number
- gdbm-empty-database gdbm-cant-be-reader
- gdbm-cant-be-writer gdbm-reader-cant-delete
- gdbm-reader-cant-store gdbm-reader-cant-reorganize
- gdbm-unknown-update gdbm-item-not-found
- gdbm-reorganize-failed gdbm-cannot-replace
- gdbm-illegal-data gdbm-option-already-set
+ { role initial: wrcreat }
+ { sync boolean }
+ { nolock boolean }
+ { mode integer initial: 0o644 } ;
+
+: <gdbm> ( -- gdbm ) gdbm new ;
+
+ENUM: gdbm-error
+ gdbm-no-error
+ gdbm-malloc-error
+ gdbm-block-size-error
+ gdbm-file-open-error
+ gdbm-file-write-error
+ gdbm-file-seek-error
+ gdbm-file-read-error
+ gdbm-bad-magic-number
+ gdbm-empty-database
+ gdbm-cant-be-reader
+ gdbm-cant-be-writer
+ gdbm-reader-cant-delete
+ gdbm-reader-cant-store
+ gdbm-reader-cant-reorganize
+ gdbm-unknown-update
+ gdbm-item-not-found
+ gdbm-reorganize-failed
+ gdbm-cannot-replace
+ gdbm-illegal-data
+ gdbm-option-already-set
gdbm-illegal-option ;
-ERROR: gdbm-unknown-error error ;
-
<PRIVATE
-: error-table ( -- table )
- {
- { 0 gdbm-no-error }
- { 1 gdbm-malloc-error }
- { 2 gdbm-block-size-error }
- { 3 gdbm-file-open-error }
- { 4 gdbm-file-write-error }
- { 5 gdbm-file-seek-error }
- { 6 gdbm-file-read-error }
- { 7 gdbm-bad-magic-number }
- { 8 gdbm-empty-database }
- { 9 gdbm-cant-be-reader }
- { 10 gdbm-cant-be-writer }
- { 11 gdbm-reader-cant-delete }
- { 12 gdbm-reader-cant-store }
- { 13 gdbm-reader-cant-reorganize }
- { 14 gdbm-unknown-update }
- { 15 gdbm-item-not-found }
- { 16 gdbm-reorganize-failed }
- { 17 gdbm-cannot-replace }
- { 18 gdbm-illegal-data }
- { 19 gdbm-option-already-set }
- { 20 gdbm-illegal-option }
- } >biassoc ;
-
-: error>code ( error -- code )
- dup error-table value-at [ ] [ gdbm-unknown-error ] ?if ;
-
-: code>error ( code -- error ) error-table at ;
-
-: gdbm-throw ( -- * ) gdbm_errno code>error throw ;
+: gdbm-throw ( -- * ) gdbm_errno gdbm-error number>enum throw ;
: check-error ( ret -- ) 0 = [ gdbm-throw ] unless ;
-
SYMBOL: current-dbf
: dbf ( -- dbf ) current-dbf get ;
+: get-flag ( gdbm -- n )
+ [ role>> enum>number ]
+ [ sync>> GDBM_SYNC 0 ? ]
+ [ nolock>> GDBM_NOLOCK 0 ? ]
+ tri bitor bitor ;
+
: gdbm-open ( gdbm -- dbf )
{
[ name>> normalize-path ]
- [ block-size>> ] [ flags>> ] [ mode>> ]
+ [ block-size>> ] [ get-flag ] [ mode>> ]
} cleave f gdbm_open [ gdbm-throw ] unless* ;
DESTRUCTOR: gdbm-close
: gdbm-store ( key content flag -- )
[
- { [ dbf ] [ object>datum ] [ object>datum ] [ ] } spread
+ [ dbf ] 3dip
+ [ object>datum ] [ object>datum ] [ ] tri*
gdbm_store check-error
] with-destructors ;
+:: (setopt) ( value option -- )
+ [
+ int heap-size dup malloc &free :> ( size ptr )
+ value ptr 0 int set-alien-value
+ dbf option ptr size gdbm_setopt check-error
+ ] with-destructors ;
+
+: setopt ( value option -- )
+ [ GDBM_CACHESIZE = [ >c-bool ] unless ] keep (setopt) ;
+
PRIVATE>
-: gdbm-error-message ( error -- msg ) error>code gdbm_strerror ;
+: gdbm-info ( -- str ) gdbm_version ;
-: gdbm-replace ( key content -- ) GDBM_REPLACE gdbm-store ;
-: gdbm-insert ( key content -- ) GDBM_INSERT gdbm-store ;
+: gdbm-error-message ( error -- msg )
+ enum>number gdbm_strerror ;
-: gdbm-delete ( key -- )
+: replace ( key content -- ) GDBM_REPLACE gdbm-store ;
+: insert ( key content -- ) GDBM_INSERT gdbm-store ;
+
+: delete ( key -- )
[ dbf swap object>datum gdbm_delete check-error ]
with-destructors ;
-: gdbm-fetch* ( key -- content ? )
+: fetch* ( key -- content ? )
[ dbf swap object>datum gdbm_fetch datum>object* ]
with-destructors ;
-: gdbm-first-key* ( -- key ? )
+: first-key* ( -- key ? )
[ dbf gdbm_firstkey datum>object* ] with-destructors ;
-: gdbm-next-key* ( key -- key ? )
+: next-key* ( key -- next-key ? )
[ dbf swap object>datum gdbm_nextkey datum>object* ]
with-destructors ;
-: gdbm-fetch ( key -- content/f ) gdbm-fetch* drop ;
-: gdbm-first-key ( -- key/f ) gdbm-first-key* drop ;
-: gdbm-next-key ( key -- key/f ) gdbm-next-key* drop ;
+: fetch ( key -- content/f ) fetch* drop ;
+: first-key ( -- key/f ) first-key* drop ;
+: next-key ( key -- key/f ) next-key* drop ;
+
+:: each-key ( ... quot: ( ... key -- ... ) -- ... )
+ first-key*
+ [ [ next-key* ] [ quot keep ] do while ] when drop ; inline
+
+: each-value ( ... quot: ( ... value -- ... ) -- ... )
+ [ fetch ] prepose each-key ; inline
+
+: each-record ( ... quot: ( ... key value -- ... ) -- ... )
+ [ dup fetch ] prepose each-key ; inline
-: gdbm-reorganize ( -- ) dbf gdbm_reorganize check-error ;
+: reorganize ( -- ) dbf gdbm_reorganize check-error ;
-: gdbm-sync ( -- ) dbf gdbm_sync ;
+: synchronize ( -- ) dbf gdbm_sync ;
-: gdbm-exists ( key -- ? )
+: exists? ( key -- ? )
[ dbf swap object>datum gdbm_exists c-bool> ]
with-destructors ;
-! : gdbm-setopt ( option value size -- ret ) ;
+: set-cache-size ( size -- ) GDBM_CACHESIZE setopt ;
+: set-sync-mode ( ? -- ) GDBM_SYNCMODE setopt ;
+: set-block-pool ( ? -- ) GDBM_CENTFREE setopt ;
+: set-block-merging ( ? -- ) GDBM_COALESCEBLKS setopt ;
-: gdbm-fdesc ( -- desc ) dbf gdbm_fdesc ;
+: gdbm-file-descriptor ( -- desc ) dbf gdbm_fdesc ;
: with-gdbm ( gdbm quot -- )
[ gdbm-open &gdbm-close current-dbf set ] prepose curry
[ with-scope ] curry with-destructors ; inline
+
+:: with-gdbm-role ( name role quot -- )
+ <gdbm> name >>name role >>role quot with-gdbm ; inline
+
+: with-gdbm-reader ( name quot -- )
+ reader swap with-gdbm-role ; inline
+
+: with-gdbm-writer ( name quot -- )
+ writer swap with-gdbm-role ; inline
+