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