r9228: add tests for result-types for map-query and do-query, ignore incorrect select...
[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       #+ignore
55       ((consp (car in-phrase))
56        (ansi-loop::loop-error "object query not yet supported"))
57       
58       ((consp variable)
59        (let ((query-var (ansi-loop::loop-gentemp 'loop-record-))
60              (db-var (ansi-loop::loop-gentemp 'loop-record-database-))
61              (result-set-var (ansi-loop::loop-gentemp
62                               'loop-record-result-set-))
63              (step-var (ansi-loop::loop-gentemp 'loop-record-step-)))
64          (push `(when ,result-set-var
65                   (clsql-base:database-dump-result-set ,result-set-var ,db-var))
66                ansi-loop::*loop-epilogue*)
67          `(((,variable nil ,@(and data-type (list data-type)))
68             (,query-var ,(first in-phrase))
69             (,db-var ,(first from-phrase))
70             (,result-set-var nil)
71             (,step-var nil))
72            ((multiple-value-bind (%rs %cols)
73                 (clsql-base:database-query-result-set ,query-var ,db-var)
74               (setq ,result-set-var %rs ,step-var (make-list %cols))))
75            ()
76            ()
77            (not (clsql-base:database-store-next-row ,result-set-var ,db-var ,step-var))
78            (,variable ,step-var)
79            (not ,result-set-var)
80            ()
81            (not (clsql-base:database-store-next-row ,result-set-var ,db-var ,step-var))
82            (,variable ,step-var))))
83       (t
84        (let ((query-var (ansi-loop::loop-gentemp 'loop-record-))
85              (db-var (ansi-loop::loop-gentemp 'loop-record-database-))
86              (result-set-var (ansi-loop::loop-gentemp
87                               'loop-record-result-set-)))
88          (push `(when ,result-set-var
89                  (clsql-base:database-dump-result-set ,result-set-var ,db-var))
90                ansi-loop::*loop-epilogue*)
91          `(((,variable nil ,@(and data-type (list data-type)))
92             (,query-var ,(first in-phrase))
93             (,db-var ,(first from-phrase))
94             (,result-set-var nil))
95            ((multiple-value-bind (%rs %cols)
96                 (clsql-base:database-query-result-set ,query-var ,db-var)
97               (setq ,result-set-var %rs ,variable (make-list %cols))))
98            ()
99            ()
100            (not (clsql-base:database-store-next-row ,result-set-var ,db-var ,variable))
101            ()
102            (not ,result-set-var)
103            ()
104            (not (clsql-base:database-store-next-row ,result-set-var ,db-var ,variable))
105            ()))))))
106
107 #+(or cmu scl sbcl openmcl allegro)
108 (ansi-loop::add-loop-path '(record records tuple tuples)
109                           'loop-record-iteration-path
110                           ansi-loop::*loop-ansi-universe*
111                           :preposition-groups '((:of :in) (:from))
112                           :inclusive-permitted nil)
113
114 #+lispworks 
115 (eval-when (:compile-toplevel :load-toplevel :execute)
116   (in-package loop))
117
118 #+lispworks
119 (cl-user::define-loop-method (record records tuple tuples) clsql-loop-method 
120   (in of from))
121
122 #+lispworks
123 (defun clsql-loop-method (method-name iter-var iter-var-data-type 
124                           prep-phrases inclusive? allowed-preps 
125                           method-specific-data)
126   (declare (ignore method-name inclusive? allowed-preps method-specific-data))
127   (let ((in-phrase nil)
128         (from-phrase nil))
129     (loop for (prep . rest) in prep-phrases
130           do
131           (cond
132             ((or (eq prep 'in) (eq prep 'of))
133              (when in-phrase
134                (error
135                 "Duplicate OF or IN iteration path: ~S." (cons prep rest)))
136              (setq in-phrase rest))
137             ((eq prep 'from)
138              (when from-phrase
139                (error
140                 "Duplicate FROM iteration path: ~S." (cons prep rest)))
141              (setq from-phrase rest))
142             (t
143              (error
144               "Unknown preposition: ~S." prep))))
145     (unless in-phrase
146       (error "Missing OF or IN iteration path."))
147     (unless from-phrase
148       (setq from-phrase '(clsql-base:*default-database*)))
149
150     (cond
151       ;; Object query resulting in a list of returned object instances
152       #+ignore
153       ((consp (car in-phrase))
154        (error "Object query not yet supported."))
155       
156       ((consp iter-var)
157        (let ((query-var (gensym "LOOP-RECORD-"))
158              (db-var (gensym "LOOP-RECORD-DATABASE-"))
159              (result-set-var (gensym "LOOP-RECORD-RESULT-SET-"))
160              (step-var (gensym "LOOP-RECORD-STEP-")))
161          (values
162           t
163           nil
164           `(,@(mapcar (lambda (v) `(,v nil)) iter-var)
165             (,query-var ,in-phrase)
166             (,db-var ,(first from-phrase))
167             (,result-set-var nil)
168             (,step-var nil))
169           `((multiple-value-bind (%rs %cols)
170                 (clsql-base:database-query-result-set ,query-var ,db-var)
171               (setq ,result-set-var %rs ,step-var (make-list %cols))))
172           ()
173           ()
174           `((unless (clsql-base:database-store-next-row ,result-set-var ,db-var ,step-var)
175               (when ,result-set-var
176                 (clsql-base:database-dump-result-set ,result-set-var ,db-var))
177               t))
178           `(,iter-var ,step-var)
179           `((unless (clsql-base:database-store-next-row ,result-set-var ,db-var ,step-var)
180               (when ,result-set-var
181                 (clsql-base:database-dump-result-set ,result-set-var ,db-var))
182               t))
183           `(,iter-var ,step-var)
184           ()
185           ())))
186       (t
187        (let ((query-var (gensym "LOOP-RECORD-"))
188              (db-var (gensym "LOOP-RECORD-DATABASE-"))
189              (result-set-var (gensym "LOOP-RECORD-RESULT-SET-")))
190          (values
191           t
192           nil
193           `((,iter-var nil ,iter-var-data-type) (,query-var ,in-phrase)
194             (,db-var ,(first from-phrase))
195             (,result-set-var nil))
196           `((multiple-value-bind (%rs %cols)
197                 (clsql-base:database-query-result-set ,query-var ,db-var)
198               (setq ,result-set-var %rs ,iter-var (make-list %cols))))
199           ()
200           ()
201           `((unless (clsql-base:database-store-next-row ,result-set-var ,db-var ,iter-var)
202               (when ,result-set-var
203                 (clsql-base:database-dump-result-set ,result-set-var ,db-var))
204               t))
205            ()
206           `((unless (clsql-base:database-store-next-row ,result-set-var ,db-var ,iter-var)
207               (when ,result-set-var
208                 (clsql-base:database-dump-result-set ,result-set-var ,db-var))
209               t))
210           ()
211           ()
212           ()))))))
213