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 db.private
8 nmake accessors random db.queries destructors db.tuples.private
9 db.postgresql db.errors.postgresql splitting ;
12 TUPLE: postgresql-db host port pgopts pgtty database username password ;
14 : <postgresql-db> ( -- postgresql-db )
19 TUPLE: postgresql-db-connection < db-connection ;
20 : <postgresql-db-connection> ( handle -- db-connection )
21 postgresql-db-connection new-db-connection
26 TUPLE: postgresql-statement < statement ;
28 TUPLE: postgresql-result-set < result-set ;
30 M: postgresql-db db-open ( db -- db-connection )
39 } cleave connect-postgres <postgresql-db-connection> ;
41 M: postgresql-db-connection db-close ( handle -- ) PQfinish ;
43 M: postgresql-statement bind-statement* ( statement -- ) drop ;
45 GENERIC: postgresql-bind-conversion ( tuple object -- low-level-binding )
47 M: sql-spec postgresql-bind-conversion ( tuple spec -- object )
48 slot-name>> swap get-slot-named <low-level-binding> ;
50 M: literal-bind postgresql-bind-conversion ( tuple literal-bind -- object )
51 nip value>> <low-level-binding> ;
53 M: generator-bind postgresql-bind-conversion ( tuple generate-bind -- object )
54 dup generator-singleton>> eval-generator
55 [ swap slot-name>> rot set-slot-named ] [ <low-level-binding> ] bi ;
57 M: postgresql-statement bind-tuple ( tuple statement -- )
60 [ postgresql-bind-conversion ] with map
64 M: postgresql-result-set #rows ( result-set -- n )
67 M: postgresql-result-set #columns ( result-set -- n )
70 : result-handle-n ( result-set -- handle n )
71 [ handle>> ] [ n>> ] bi ;
73 M: postgresql-result-set row-column ( result-set column -- object )
74 [ result-handle-n ] dip pq-get-string ;
76 M: postgresql-result-set row-column-typed ( result-set column -- object )
77 dup pick out-params>> nth type>>
78 [ result-handle-n ] 2dip postgresql-column-typed ;
80 M: postgresql-statement query-results ( query -- result-set )
82 over [ bind-statement ] keep
83 do-postgresql-bound-statement
85 dup do-postgresql-statement
87 postgresql-result-set new-result-set
90 M: postgresql-result-set advance-row ( result-set -- )
91 [ 1+ ] change-n drop ;
93 M: postgresql-result-set more-rows? ( result-set -- ? )
94 [ n>> ] [ max>> ] bi < ;
96 M: postgresql-statement dispose ( query -- )
100 M: postgresql-result-set dispose ( result-set -- )
108 M: postgresql-statement prepare-statement ( statement -- )
110 [ db-connection get handle>> f ] dip
111 [ sql>> ] [ in-params>> ] bi
112 length f PQprepare postgresql-error
115 M: postgresql-db-connection <simple-statement> ( sql in out -- statement )
116 postgresql-statement new-statement ;
118 M: postgresql-db-connection <prepared-statement> ( sql in out -- statement )
119 <simple-statement> dup prepare-statement ;
123 sql-counter [ inc ] [ get 0# ] bi ;
125 M: postgresql-db-connection bind% ( spec -- )
128 M: postgresql-db-connection bind# ( spec object -- )
129 [ bind-name% f swap type>> ] dip
132 : create-table-sql ( class -- statement )
135 "create table " 0% 0%
139 dup type>> lookup-create-type 0%
146 [ "," 0% ] [ column-name>> 0% ] interleave
150 : create-function-sql ( class -- statement )
152 [ dup remove-id ] dip
153 "create function add_" 0% dup 0%
157 type>> lookup-type 0%
160 " returns bigint as '" 0%
165 over [ ", " 0% ] [ column-name>> 0% ] interleave
167 swap [ ", " 0% ] [ drop bind-name% ] interleave
169 "select currval(''" 0% 0% "_" 0%
170 find-primary-key first column-name>> 0%
171 "_seq'');' language sql;" 0%
174 M: postgresql-db-connection create-sql-statement ( class -- seq )
176 [ create-table-sql , ] keep
177 dup db-assigned? [ create-function-sql , ] [ drop ] if
180 : drop-function-sql ( class -- statement )
182 "drop function add_" 0% 0%
185 [ ", " 0% ] [ type>> lookup-type 0% ] interleave
189 : drop-table-sql ( table -- statement )
191 "drop table " 0% 0% drop
194 M: postgresql-db-connection drop-sql-statement ( class -- seq )
196 [ drop-table-sql , ] keep
197 dup db-assigned? [ drop-function-sql , ] [ drop ] if
200 M: postgresql-db-connection <insert-db-assigned-statement> ( class -- statement )
204 dup find-primary-key first 2,
206 [ ", " 0% ] [ bind% ] interleave
210 M: postgresql-db-connection <insert-user-assigned-statement> ( class -- statement )
214 dup [ ", " 0% ] [ column-name>> 0% ] interleave
219 dup type>> +random-id+ = [
225 ] [ type>> ] bi <generator-bind> 1,
233 M: postgresql-db-connection insert-tuple-set-key ( tuple statement -- )
236 M: postgresql-db-connection persistent-table ( -- hashtable )
238 { +db-assigned-id+ { "integer" "serial" f } }
239 { +user-assigned-id+ { f f f } }
240 { +random-id+ { "bigint" "bigint" f } }
242 { +foreign-id+ { f f "references" } }
244 { +on-update+ { f f "on update" } }
245 { +on-delete+ { f f "on delete" } }
246 { +restrict+ { f f "restrict" } }
247 { +cascade+ { f f "cascade" } }
248 { +set-null+ { f f "set null" } }
249 { +set-default+ { f f "set default" } }
251 { TEXT { "text" "text" f } }
252 { VARCHAR { "varchar" "varchar" f } }
253 { INTEGER { "integer" "integer" f } }
254 { BIG-INTEGER { "bigint" "bigint" f } }
255 { UNSIGNED-BIG-INTEGER { "bigint" "bigint" f } }
256 { SIGNED-BIG-INTEGER { "bigint" "bigint" f } }
257 { DOUBLE { "real" "real" f } }
258 { DATE { "date" "date" f } }
259 { TIME { "time" "time" f } }
260 { DATETIME { "timestamp" "timestamp" f } }
261 { TIMESTAMP { "timestamp" "timestamp" f } }
262 { BLOB { "bytea" "bytea" f } }
263 { FACTOR-BLOB { "bytea" "bytea" f } }
264 { URL { "varchar" "varchar" f } }
265 { +autoincrement+ { f f "autoincrement" } }
266 { +unique+ { f f "unique" } }
267 { +default+ { f f "default" } }
268 { +null+ { f f "null" } }
269 { +not-null+ { f f "not null" } }
270 { system-random-generator { f f f } }
271 { secure-random-generator { f f f } }
272 { random-generator { f f f } }
275 ERROR: no-compound-found string object ;
276 M: postgresql-db-connection compound ( string object -- string' )
278 { "default" [ first number>string " " glue ] }
279 { "varchar" [ first number>string "(" ")" surround append ] }
280 { "references" [ >reference-string ] }
281 [ drop no-compound-found ]
284 M: postgresql-db-connection parse-db-error
285 "\n" split dup length {
286 { 1 [ first parse-postgresql-sql-error ] }
289 [ parse-postgresql-sql-error ] 2dip
290 postgresql-location >>location