r9229: new tests
[clsql.git] / base / loop-extension.lisp
1 ;;;; -*- Mode: LISP; Syntax: ANSI-Common-Lisp; Base: 10 -*-
2 ;;;; *************************************************************************
3 ;;;; FILE IDENTIFICATION
4 ;;;;
5 ;;;; Name: loop-extension.lisp
6 ;;;; Purpose: Extensions to the Loop macro for CLSQL
7 ;;;;
8 ;;;; Copyright (c) 2001-2004 Kevin Rosenberg and (c) 1999-2001 Pierre R. Mai
9 ;;;;
10 ;;;; $Id$
11 ;;;; *************************************************************************
12
13 (in-package #:cl-user)
14
15 #+(or allegro sbcl)
16 (eval-when (:compile-toplevel :load-toplevel :execute)
17   (defpackage #:ansi-loop 
18     (:import-from #+sbcl #:sb-loop #+allegro #:excl
19                   #:loop-error
20                   #:*loop-epilogue*
21                   #:*loop-ansi-universe* 
22                   #:add-loop-path)))
23
24 #+(or allegro sbcl)
25 (defun ansi-loop::loop-gentemp (&optional (pref 'loopva-))
26   (gensym (string pref)))
27
28 #+(or cmu scl sbcl openmcl allegro)
29 (defun loop-record-iteration-path (variable data-type prep-phrases)
30   (let ((in-phrase nil)
31         (from-phrase nil))
32     (loop for (prep . rest) in prep-phrases
33           do
34           (case prep
35             ((:in :of)
36              (when in-phrase
37                (ansi-loop::loop-error
38                 "Duplicate OF or IN iteration path: ~S." (cons prep rest)))
39              (setq in-phrase rest))
40             ((:from)
41              (when from-phrase
42                (ansi-loop::loop-error
43                 "Duplicate FROM iteration path: ~S." (cons prep rest)))
44              (setq from-phrase rest))
45             (t
46              (ansi-loop::loop-error
47               "Unknown preposition: ~S." prep))))
48     (unless in-phrase
49       (ansi-loop::loop-error "Missing OF or IN iteration path."))
50     (unless from-phrase
51       (setq from-phrase '(clsql-base:*default-database*)))
52     (cond
53       ;; Object query resulting in a list of returned object instances
54       ((and (consp in-phrase)
55             (consp (car in-phrase))
56             (consp (second (car in-phrase)))
57             (eq 'quote (first (second (car in-phrase))))
58             (symbolp (second (second (car in-phrase)))))
59        (ansi-loop::loop-error "object query not yet supported"))
60       
61       ((consp variable)
62        (let ((query-var (ansi-loop::loop-gentemp 'loop-record-))
63              (db-var (ansi-loop::loop-gentemp 'loop-record-database-))
64              (result-set-var (ansi-loop::loop-gentemp
65                               'loop-record-result-set-))
66              (step-var (ansi-loop::loop-gentemp 'loop-record-step-)))
67          (push `(when ,result-set-var
68                   (clsql-base:database-dump-result-set ,result-set-var ,db-var))
69                ansi-loop::*loop-epilogue*)
70          `(((,variable nil ,@(and data-type (list data-type)))
71             (,query-var ,(first in-phrase))
72             (,db-var ,(first from-phrase))
73             (,result-set-var nil)
74             (,step-var nil))
75            ((multiple-value-bind (%rs %cols)
76                 (clsql-base:database-query-result-set ,query-var ,db-var :result-types :auto)
77               (setq ,result-set-var %rs ,step-var (make-list %cols))))
78            ()
79            ()
80            (not (clsql-base:database-store-next-row ,result-set-var ,db-var ,step-var))
81            (,variable ,step-var)
82            (not ,result-set-var)
83            ()
84            (not (clsql-base:database-store-next-row ,result-set-var ,db-var ,step-var))
85            (,variable ,step-var))))
86       (t
87        (let ((query-var (ansi-loop::loop-gentemp 'loop-record-))
88              (db-var (ansi-loop::loop-gentemp 'loop-record-database-))
89              (result-set-var (ansi-loop::loop-gentemp
90                               'loop-record-result-set-)))
91          (push `(when ,result-set-var
92                  (clsql-base:database-dump-result-set ,result-set-var ,db-var))
93                ansi-loop::*loop-epilogue*)
94          `(((,variable nil ,@(and data-type (list data-type)))
95             (,query-var ,(first in-phrase))
96             (,db-var ,(first from-phrase))
97             (,result-set-var nil))
98            ((multiple-value-bind (%rs %cols)
99                 (clsql-base:database-query-result-set ,query-var ,db-var :result-types :auto)
100               (setq ,result-set-var %rs ,variable (make-list %cols))))
101            ()
102            ()
103            (not (clsql-base:database-store-next-row ,result-set-var ,db-var ,variable))
104            ()
105            (not ,result-set-var)
106            ()
107            (not (clsql-base:database-store-next-row ,result-set-var ,db-var ,variable))
108            ()))))))
109
110 #+(or cmu scl sbcl openmcl allegro)
111 (ansi-loop::add-loop-path '(record records tuple tuples)
112                           'loop-record-iteration-path
113                           ansi-loop::*loop-ansi-universe*
114                           :preposition-groups '((:of :in) (:from))
115                           :inclusive-permitted nil)
116
117 #+lispworks 
118 (eval-when (:compile-toplevel :load-toplevel :execute)
119   (in-package loop))
120
121 #+lispworks
122 (cl-user::define-loop-method (record records tuple tuples) clsql-loop-method 
123   (in of from))
124
125 #+lispworks
126 (defun clsql-loop-method (method-name iter-var iter-var-data-type 
127                           prep-phrases inclusive? allowed-preps 
128                           method-specific-data)
129   (declare (ignore method-name inclusive? allowed-preps method-specific-data))
130   (let ((in-phrase nil)
131         (from-phrase nil))
132     (loop for (prep . rest) in prep-phrases
133           do
134           (cond
135             ((or (eq prep 'in) (eq prep 'of))
136              (when in-phrase
137                (error
138                 "Duplicate OF or IN iteration path: ~S." (cons prep rest)))
139              (setq in-phrase rest))
140             ((eq prep 'from)
141              (when from-phrase
142                (error
143                 "Duplicate FROM iteration path: ~S." (cons prep rest)))
144              (setq from-phrase rest))
145             (t
146              (error
147               "Unknown preposition: ~S." prep))))
148     (unless in-phrase
149       (error "Missing OF or IN iteration path."))
150     (unless from-phrase
151       (setq from-phrase '(clsql-base:*default-database*)))
152
153     (cond
154       ;; Object query resulting in a list of returned object instances
155       ((and (consp in-phrase)
156             (consp (car in-phrase))
157             (consp (second (car in-phrase)))
158             (eq 'quote (first (second (car in-phrase))))
159             (symbolp (second (second (car in-phrase)))))
160        (loop-error "object query not yet supported"))
161       
162       ((consp iter-var)
163        (let ((query-var (gensym "LOOP-RECORD-"))
164              (db-var (gensym "LOOP-RECORD-DATABASE-"))
165              (result-set-var (gensym "LOOP-RECORD-RESULT-SET-"))
166              (step-var (gensym "LOOP-RECORD-STEP-")))
167          (values
168           t
169           nil
170           `(,@(mapcar (lambda (v) `(,v nil)) iter-var)
171             (,query-var ,in-phrase)
172             (,db-var ,(first from-phrase))
173             (,result-set-var nil)
174             (,step-var nil))
175           `((multiple-value-bind (%rs %cols)
176                 (clsql-base:database-query-result-set ,query-var ,db-var :result-types :auto)
177               (setq ,result-set-var %rs ,step-var (make-list %cols))))
178           ()
179           ()
180           `((unless (clsql-base:database-store-next-row ,result-set-var ,db-var ,step-var)
181               (when ,result-set-var
182                 (clsql-base:database-dump-result-set ,result-set-var ,db-var))
183               t))
184           `(,iter-var ,step-var)
185           `((unless (clsql-base:database-store-next-row ,result-set-var ,db-var ,step-var)
186               (when ,result-set-var
187                 (clsql-base:database-dump-result-set ,result-set-var ,db-var))
188               t))
189           `(,iter-var ,step-var)
190           ()
191           ())))
192       (t
193        (let ((query-var (gensym "LOOP-RECORD-"))
194              (db-var (gensym "LOOP-RECORD-DATABASE-"))
195              (result-set-var (gensym "LOOP-RECORD-RESULT-SET-")))
196          (values
197           t
198           nil
199           `((,iter-var nil ,iter-var-data-type) (,query-var ,in-phrase)
200             (,db-var ,(first from-phrase))
201             (,result-set-var nil))
202           `((multiple-value-bind (%rs %cols)
203                 (clsql-base:database-query-result-set ,query-var ,db-var :result-types :auto)
204               (setq ,result-set-var %rs ,iter-var (make-list %cols))))
205           ()
206           ()
207           `((unless (clsql-base:database-store-next-row ,result-set-var ,db-var ,iter-var)
208               (when ,result-set-var
209                 (clsql-base:database-dump-result-set ,result-set-var ,db-var))
210               t))
211            ()
212           `((unless (clsql-base:database-store-next-row ,result-set-var ,db-var ,iter-var)
213               (when ,result-set-var
214                 (clsql-base:database-dump-result-set ,result-set-var ,db-var))
215               t))
216           ()
217           ()
218           ()))))))
219