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