1 ! Copyright (C) 2007, 2008 Doug Coleman.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: arrays assocs alien alien.syntax continuations io
4 kernel math math.parser namespaces make prettyprint quotations
5 sequences debugger db db.postgresql.lib db.postgresql.ffi
6 db.tuples db.types tools.annotations math.ranges
7 combinators classes locals words tools.walker
8 nmake accessors random db.queries destructors db.tuples.private ;
12 TUPLE: postgresql-db < db
13 host port pgopts pgtty database username password ;
15 : <postgresql-db> ( -- postgresql-db )
16 postgresql-db new-db ;
18 TUPLE: postgresql-statement < statement ;
20 TUPLE: postgresql-result-set < result-set ;
22 M: postgresql-db db-open ( db -- db )
31 } cleave connect-postgres >>handle ;
33 M: postgresql-db dispose ( db -- )
36 M: postgresql-statement bind-statement* ( statement -- ) drop ;
38 GENERIC: postgresql-bind-conversion ( tuple object -- low-level-binding )
40 M: sql-spec postgresql-bind-conversion ( tuple spec -- object )
41 slot-name>> swap get-slot-named <low-level-binding> ;
43 M: literal-bind postgresql-bind-conversion ( tuple literal-bind -- object )
44 nip value>> <low-level-binding> ;
46 M: generator-bind postgresql-bind-conversion ( tuple generate-bind -- object )
47 dup generator-singleton>> eval-generator
48 [ swap slot-name>> rot set-slot-named ] [ <low-level-binding> ] bi ;
50 M: postgresql-statement bind-tuple ( tuple statement -- )
52 [ postgresql-bind-conversion ] with map
55 M: postgresql-result-set #rows ( result-set -- n )
58 M: postgresql-result-set #columns ( result-set -- n )
61 : result-handle-n ( result-set -- handle n )
62 [ handle>> ] [ n>> ] bi ;
64 M: postgresql-result-set row-column ( result-set column -- object )
65 [ result-handle-n ] dip pq-get-string ;
67 M: postgresql-result-set row-column-typed ( result-set column -- object )
68 dup pick out-params>> nth type>>
69 [ result-handle-n ] 2dip postgresql-column-typed ;
71 M: postgresql-statement query-results ( query -- result-set )
73 over [ bind-statement ] keep
74 do-postgresql-bound-statement
76 dup do-postgresql-statement
78 postgresql-result-set new-result-set
81 M: postgresql-result-set advance-row ( result-set -- )
82 [ 1+ ] change-n drop ;
84 M: postgresql-result-set more-rows? ( result-set -- ? )
85 [ n>> ] [ max>> ] bi < ;
87 M: postgresql-statement dispose ( query -- )
91 M: postgresql-result-set dispose ( result-set -- )
99 M: postgresql-statement prepare-statement ( statement -- )
101 [ db get handle>> f ] dip
102 [ sql>> ] [ in-params>> ] bi
103 length f PQprepare postgresql-error
106 M: postgresql-db <simple-statement> ( sql in out -- statement )
107 postgresql-statement new-statement ;
109 M: postgresql-db <prepared-statement> ( sql in out -- statement )
110 <simple-statement> dup prepare-statement ;
114 sql-counter [ inc ] [ get 0# ] bi ;
116 M: postgresql-db bind% ( spec -- )
119 M: postgresql-db bind# ( spec object -- )
120 [ bind-name% f swap type>> ] dip
123 : create-table-sql ( class -- statement )
126 "create table " 0% 0%
130 dup type>> lookup-create-type 0%
137 [ "," 0% ] [ column-name>> 0% ] interleave
141 : create-function-sql ( class -- statement )
144 "create function add_" 0% dup 0%
148 type>> lookup-type 0%
151 " returns bigint as '" 0%
156 over [ ", " 0% ] [ column-name>> 0% ] interleave
158 swap [ ", " 0% ] [ drop bind-name% ] interleave
160 "select currval(''" 0% 0% "_id_seq'');' language sql;" 0%
163 M: postgresql-db create-sql-statement ( class -- seq )
165 [ create-table-sql , ] keep
166 dup db-assigned? [ create-function-sql , ] [ drop ] if
169 : drop-function-sql ( class -- statement )
171 "drop function add_" 0% 0%
174 [ ", " 0% ] [ type>> lookup-type 0% ] interleave
178 : drop-table-sql ( table -- statement )
180 "drop table " 0% 0% drop
183 M: postgresql-db drop-sql-statement ( class -- seq )
185 [ drop-table-sql , ] keep
186 dup db-assigned? [ drop-function-sql , ] [ drop ] if
189 M: postgresql-db <insert-db-assigned-statement> ( class -- statement )
193 dup find-primary-key first 2,
195 [ ", " 0% ] [ bind% ] interleave
199 M: postgresql-db <insert-user-assigned-statement> ( class -- statement )
203 dup [ ", " 0% ] [ column-name>> 0% ] interleave
208 dup type>> +random-id+ = [
214 ] [ type>> ] bi <generator-bind> 1,
222 M: postgresql-db insert-tuple-set-key ( tuple statement -- )
225 M: postgresql-db persistent-table ( -- hashtable )
227 { +db-assigned-id+ { "integer" "serial" f } }
228 { +user-assigned-id+ { f f f } }
229 { +random-id+ { "bigint" "bigint" f } }
231 { +foreign-id+ { f f "references" } }
233 { +on-delete+ { f f "on delete" } }
234 { +restrict+ { f f "restrict" } }
235 { +cascade+ { f f "cascade" } }
236 { +set-null+ { f f "set null" } }
237 { +set-default+ { f f "set default" } }
239 { TEXT { "text" "text" f } }
240 { VARCHAR { "varchar" "varchar" f } }
241 { INTEGER { "integer" "integer" f } }
242 { BIG-INTEGER { "bigint" "bigint" f } }
243 { UNSIGNED-BIG-INTEGER { "bigint" "bigint" f } }
244 { SIGNED-BIG-INTEGER { "bigint" "bigint" f } }
245 { DOUBLE { "real" "real" f } }
246 { DATE { "date" "date" f } }
247 { TIME { "time" "time" f } }
248 { DATETIME { "timestamp" "timestamp" f } }
249 { TIMESTAMP { "timestamp" "timestamp" f } }
250 { BLOB { "bytea" "bytea" f } }
251 { FACTOR-BLOB { "bytea" "bytea" f } }
252 { URL { "varchar" "varchar" f } }
253 { +autoincrement+ { f f "autoincrement" } }
254 { +unique+ { f f "unique" } }
255 { +default+ { f f "default" } }
256 { +null+ { f f "null" } }
257 { +not-null+ { f f "not null" } }
258 { system-random-generator { f f f } }
259 { secure-random-generator { f f f } }
260 { random-generator { f f f } }
263 ERROR: no-compound-found string object ;
264 M: postgresql-db compound ( string object -- string' )
266 { "default" [ first number>string join-space ] }
267 { "varchar" [ first number>string paren append ] }
268 { "references" [ >reference-string ] }
269 [ drop no-compound-found ]