113 lines
3.4 KiB
Factor
Executable File
113 lines
3.4 KiB
Factor
Executable File
! Copyright (C) 2008 Doug Coleman.
|
|
! See http://factorcode.org/license.txt for BSD license.
|
|
USING: arrays assocs classes db kernel namespaces
|
|
tuples words sequences slots math
|
|
math.parser io prettyprint db.types continuations
|
|
mirrors sequences.lib tools.walker combinators.lib ;
|
|
IN: db.tuples
|
|
|
|
: define-persistent ( class table columns -- )
|
|
>r dupd "db-table" set-word-prop dup r>
|
|
[ relation? ] partition swapd
|
|
dupd [ spec>tuple ] with map
|
|
"db-columns" set-word-prop
|
|
"db-relations" set-word-prop ;
|
|
|
|
: db-table ( class -- obj ) "db-table" word-prop ;
|
|
: db-columns ( class -- obj ) "db-columns" word-prop ;
|
|
: db-relations ( class -- obj ) "db-relations" word-prop ;
|
|
|
|
: set-primary-key ( key tuple -- )
|
|
[
|
|
class db-columns find-primary-key sql-spec-slot-name
|
|
] keep set-slot-named ;
|
|
|
|
! returns a sequence of prepared-statements
|
|
HOOK: create-sql-statement db ( class -- obj )
|
|
HOOK: drop-sql-statement db ( class -- obj )
|
|
|
|
HOOK: <insert-native-statement> db ( class -- obj )
|
|
HOOK: <insert-assigned-statement> db ( class -- obj )
|
|
|
|
HOOK: <update-tuple-statement> db ( class -- obj )
|
|
HOOK: <update-tuples-statement> db ( class -- obj )
|
|
|
|
HOOK: <delete-tuple-statement> db ( class -- obj )
|
|
HOOK: <delete-tuples-statement> db ( class -- obj )
|
|
|
|
HOOK: <select-by-slots-statement> db ( tuple -- tuple )
|
|
|
|
HOOK: row-column-typed db ( result-set n type -- sql )
|
|
HOOK: insert-tuple* db ( tuple statement -- )
|
|
|
|
: resulting-tuple ( row out-params -- tuple )
|
|
dup first sql-spec-class construct-empty [
|
|
[
|
|
>r [ sql-spec-type sql-type>factor-type ] keep
|
|
sql-spec-slot-name r> set-slot-named
|
|
] curry 2each
|
|
] keep ;
|
|
|
|
: query-tuples ( statement -- seq )
|
|
[ statement-out-params ] keep query-results [
|
|
[ sql-row swap resulting-tuple ] with query-map
|
|
] with-disposal ;
|
|
|
|
: query-modify-tuple ( tuple statement -- )
|
|
[ query-results [ sql-row ] with-disposal ] keep
|
|
statement-out-params rot [
|
|
>r [ sql-spec-type sql-type>factor-type ] keep
|
|
sql-spec-slot-name r> set-slot-named
|
|
] curry 2each ;
|
|
|
|
: sql-props ( class -- columns table )
|
|
dup db-columns swap db-table ;
|
|
|
|
: with-disposals ( seq quot -- )
|
|
over sequence? [
|
|
[ with-disposal ] curry each
|
|
] [
|
|
with-disposal
|
|
] if ;
|
|
|
|
: create-table ( class -- )
|
|
create-sql-statement [ execute-statement ] with-disposals ;
|
|
|
|
: drop-table ( class -- )
|
|
drop-sql-statement [ execute-statement ] with-disposals ;
|
|
|
|
: insert-native ( tuple -- )
|
|
dup class
|
|
db get db-insert-statements [ <insert-native-statement> ] cache
|
|
[ bind-tuple ] 2keep insert-tuple* ;
|
|
|
|
: insert-assigned ( tuple -- )
|
|
dup class
|
|
db get db-insert-statements [ <insert-assigned-statement> ] cache
|
|
[ bind-tuple ] keep execute-statement ;
|
|
|
|
: insert-tuple ( tuple -- )
|
|
break
|
|
dup class db-columns find-primary-key assigned-id? [
|
|
insert-assigned
|
|
] [
|
|
insert-native
|
|
] if ;
|
|
|
|
: update-tuple ( tuple -- )
|
|
dup class
|
|
db get db-update-statements [ <update-tuple-statement> ] cache
|
|
[ bind-tuple ] keep execute-statement ;
|
|
|
|
: delete-tuple ( tuple -- )
|
|
dup class
|
|
db get db-delete-statements [ <delete-tuple-statement> ] cache
|
|
[ bind-tuple ] keep execute-statement ;
|
|
|
|
: select-tuples ( tuple -- tuple )
|
|
dup dup class <select-by-slots-statement> [
|
|
[ bind-tuple ] keep query-tuples
|
|
] with-disposal ;
|
|
|
|
: select-tuple ( tuple -- tuple/f ) select-tuples ?first ;
|