r11060: fix loop error on clisp. RESULT-SET should be T at start of loop to avoid...
[clsql.git] / sql / loop-extension.lisp
index d19dfb65ac080ce77815743c45db13b803e6d880..802533092c71b36a0c5ac8fa1655219ce622c674 100644 (file)
@@ -2,23 +2,33 @@
 ;;;; *************************************************************************
 ;;;; FILE IDENTIFICATION
 ;;;;
-;;;; Name:          loop-extension.cl
-;;;; Purpose:       Extensions to the Loop macro for CMUCL
-;;;; Programmer:    Pierre R. Mai
+;;;; Name:    loop-extension.lisp
+;;;; Purpose: Extensions to the Loop macro for CLSQL
 ;;;;
-;;;; Copyright (c) 1999-2001 Pierre R. Mai
+;;;; Copyright (c) 2001-2006 Kevin Rosenberg and (c) 1999-2001 Pierre R. Mai
 ;;;;
-;;;; $Id: loop-extension.lisp,v 1.1 2002/09/30 10:19:23 kevin Exp $
-;;;;
-;;;; The functions in this file were orignally distributed in the
-;;;; MaiSQL package in the file sql/sql.cl
+;;;; $Id$
 ;;;; *************************************************************************
 
-(in-package :cl-user)
+(in-package #:clsql-sys)
+
+#+(or allegro sbcl)
+(eval-when (:compile-toplevel :load-toplevel :execute)
+  (defpackage #:ansi-loop
+    (:import-from #+sbcl #:sb-loop #+allegro #:excl
+                 #:*loop-epilogue*
+                 #:*loop-ansi-universe*
+                 #:add-loop-path)))
 
-;;;; MIT-LOOP extension
+#+(or allegro sbcl)
+(defun ansi-loop::loop-gentemp (&optional (pref 'loopva-))
+  (gensym (string pref)))
 
-#+cmu
+#+clisp
+(eval-when (:compile-toplevel :load-toplevel :execute)
+  (when (find-package "ANSI-LOOP") (push :clisp-aloop cl:*features*)))
+
+#+(or allegro clisp-aloop cmu openmcl sbcl scl)
 (defun loop-record-iteration-path (variable data-type prep-phrases)
   (let ((in-phrase nil)
        (from-phrase nil))
          (case prep
            ((:in :of)
             (when in-phrase
-              (ansi-loop::loop-error
-               "Duplicate OF or IN iteration path: ~S." (cons prep rest)))
+              (error 'clsql:sql-user-error
+                     :message
+                     (format nil
+                             "Duplicate OF or IN iteration path: ~S."
+                             (cons prep rest))))
             (setq in-phrase rest))
            ((:from)
             (when from-phrase
-              (ansi-loop::loop-error
-               "Duplicate FROM iteration path: ~S." (cons prep rest)))
+              (error 'clsql:sql-user-error
+                     :message
+                     (format nil
+                             "Duplicate FROM iteration path: ~S."
+                             (cons prep rest))))
             (setq from-phrase rest))
            (t
-            (ansi-loop::loop-error
-             "Unknown preposition: ~S." prep))))
+            (error 'clsql:sql-user-error
+                   :message
+                   (format nil"Unknown preposition: ~S." prep)))))
     (unless in-phrase
-      (ansi-loop::loop-error "Missing OF or IN iteration path."))
+      (error 'clsql:sql-user-error
+            :message "Missing OF or IN iteration path."))
     (unless from-phrase
       (setq from-phrase '(*default-database*)))
+
+    (unless (consp variable)
+      (setq variable (list variable)))
+
     (cond
+     ;; object query
+     ((and (consp (first in-phrase))
+          (string-equal "sql-query" (symbol-name (caar in-phrase)))
+          (consp (second (first in-phrase)))
+          (eq 'quote (first (second (first in-phrase))))
+          (symbolp (second (second (first in-phrase)))))
+
+       (let ((result-var (ansi-loop::loop-gentemp
+                             'loop-record-result-))
+            (step-var (ansi-loop::loop-gentemp 'loop-record-step-)))
+        `(((,variable nil ,@(and data-type (list data-type)))
+           (,result-var (query ,(first in-phrase)))
+           (,step-var nil))
+          ()
+          ()
+          ()
+          (if (null ,result-var)
+              t
+              (progn
+                (setq ,step-var (first ,result-var))
+                (setq ,result-var (rest ,result-var))
+                nil))
+          (,variable ,step-var)
+          (null ,result-var)
+          ()
+          (if (null ,result-var)
+              t
+              (progn
+                (setq ,step-var (first ,result-var))
+                (setq ,result-var (rest ,result-var))
+                nil))
+          (,variable ,step-var))))
+
       ((consp variable)
        (let ((query-var (ansi-loop::loop-gentemp 'loop-record-))
             (db-var (ansi-loop::loop-gentemp 'loop-record-database-))
                              'loop-record-result-set-))
             (step-var (ansi-loop::loop-gentemp 'loop-record-step-)))
         (push `(when ,result-set-var
-                (database-dump-result-set ,result-set-var ,db-var))
+                 (database-dump-result-set ,result-set-var ,db-var))
               ansi-loop::*loop-epilogue*)
-        `(((,variable nil ,data-type) (,query-var ,(first in-phrase))
+        `(((,variable nil ,@(and data-type (list data-type)))
+           (,query-var ,(first in-phrase))
            (,db-var ,(first from-phrase))
-           (,result-set-var nil)
+           (,result-set-var t)
            (,step-var nil))
           ((multiple-value-bind (%rs %cols)
-               (database-query-result-set ,query-var ,db-var)
+               (database-query-result-set ,query-var ,db-var :result-types :auto)
              (setq ,result-set-var %rs ,step-var (make-list %cols))))
           ()
           ()
           (not ,result-set-var)
           ()
           (not (database-store-next-row ,result-set-var ,db-var ,step-var))
-          (,variable ,step-var))))
-      (t
-       (let ((query-var (ansi-loop::loop-gentemp 'loop-record-))
-            (db-var (ansi-loop::loop-gentemp 'loop-record-database-))
-            (result-set-var (ansi-loop::loop-gentemp
-                             'loop-record-result-set-)))
-        (push `(when ,result-set-var
-                (database-dump-result-set ,result-set-var ,db-var))
-              ansi-loop::*loop-epilogue*)
-        `(((,variable nil ,data-type) (,query-var ,(first in-phrase))
-           (,db-var ,(first from-phrase))
-           (,result-set-var nil))
-          ((multiple-value-bind (%rs %cols)
-               (database-query-result-set ,query-var ,db-var)
-             (setq ,result-set-var %rs ,variable (make-list %cols))))
-          ()
-          ()
-          (not (database-store-next-row ,result-set-var ,db-var ,variable))
-          ()
-          (not ,result-set-var)
-          ()
-          (not (database-store-next-row ,result-set-var ,db-var ,variable))
-          ()))))))
+          (,variable ,step-var)))))))
 
-#+cmu
+#+(or allegro clisp-aloop cmu openmcl sbcl scl)
 (ansi-loop::add-loop-path '(record records tuple tuples)
                          'loop-record-iteration-path
                          ansi-loop::*loop-ansi-universe*
                          :preposition-groups '((:of :in) (:from))
                          :inclusive-permitted nil)
+
+
+#+lispworks
+(cl-user::define-loop-method (loop::record loop::records loop::tuple loop::tuples)
+    clsql-loop-method
+  (loop::in loop::of loop::from))
+
+#+lispworks
+(defun clsql-loop-method (method-name iter-var iter-var-data-type
+                         prep-phrases inclusive? allowed-preps
+                         method-specific-data)
+  (declare (ignore method-name inclusive? allowed-preps method-specific-data))
+  (let ((in-phrase nil)
+       (from-phrase nil))
+    (loop for (prep . rest) in prep-phrases
+         do
+         (cond
+           ((or (eq prep 'loop::in) (eq prep 'loop::of))
+            (when in-phrase
+              (error 'clsql:sql-user-error
+                     :message
+                     (format nil "Duplicate OF or IN iteration path: ~S."
+                             (cons prep rest))))
+            (setq in-phrase rest))
+           ((eq prep 'loop::from)
+            (when from-phrase
+              (error 'clsql:sql-user-error
+                     :message
+                     (format nil "Duplicate FROM iteration path: ~S."
+                             (cons prep rest))))
+            (setq from-phrase rest))
+           (t
+            (error 'clsql:sql-user-error
+                   :message (format nil "Unknown preposition: ~S." prep)))))
+    (unless in-phrase
+      (error 'clsql:sql-user-error
+            :message "Missing OF or IN iteration path."))
+    (unless from-phrase
+      (setq from-phrase '(clsql:*default-database*)))
+
+    (unless (consp iter-var)
+      (setq iter-var (list iter-var)))
+
+    (cond
+     ;; object query
+     ((and (consp in-phrase)
+          (string-equal "sql-query" (symbol-name (car in-phrase)))
+          (consp (second in-phrase))
+          (eq 'quote (first (second in-phrase)))
+          (symbolp (second (second in-phrase))))
+
+       (let ((result-var (gensym "LOOP-RECORD-RESULT-"))
+            (step-var (gensym "LOOP-RECORD-STEP-")))
+        (values
+         t
+         nil
+         `(,@(mapcar (lambda (v) `(,v nil)) iter-var)
+             (,result-var (clsql:query ,in-phrase))
+             (,step-var nil))
+         ()
+         ()
+         ()
+         `((if (null ,result-var)
+               t
+               (progn
+                 (setq ,step-var (first ,result-var))
+                 (setq ,result-var (rest ,result-var))
+                 nil)))
+         `(,iter-var ,step-var)
+         `((if (null ,result-var)
+               t
+               (progn
+                 (setq ,step-var (first ,result-var))
+                 (setq ,result-var (rest ,result-var))
+                 nil)))
+          `(,iter-var ,step-var)
+          ()
+          ()
+          )))
+
+      ((consp iter-var)
+       (let ((query-var (gensym "LOOP-RECORD-"))
+            (db-var (gensym "LOOP-RECORD-DATABASE-"))
+            (result-set-var (gensym "LOOP-RECORD-RESULT-SET-"))
+            (step-var (gensym "LOOP-RECORD-STEP-")))
+        (values
+         t
+         nil
+         `(,@(mapcar (lambda (v) `(,v nil)) iter-var)
+           (,query-var ,in-phrase)
+           (,db-var ,(first from-phrase))
+           (,result-set-var nil)
+           (,step-var nil))
+         `((multiple-value-bind (%rs %cols)
+               (clsql-sys:database-query-result-set ,query-var ,db-var :result-types :auto)
+             (setq ,result-set-var %rs ,step-var (make-list %cols))))
+         ()
+         ()
+         `((unless (clsql-sys:database-store-next-row ,result-set-var ,db-var ,step-var)
+             (when ,result-set-var
+               (clsql-sys:database-dump-result-set ,result-set-var ,db-var))
+             t))
+         `(,iter-var ,step-var)
+         `((unless (clsql-sys:database-store-next-row ,result-set-var ,db-var ,step-var)
+             (when ,result-set-var
+               (clsql-sys:database-dump-result-set ,result-set-var ,db-var))
+             t))
+         `(,iter-var ,step-var)
+         ()
+         ()))))))
+
+
+#+clisp-aloop
+(eval-when (:compile-toplevel :load-toplevel :execute)
+  (setq cl:*features* (delete :clisp-aloop cl:*features*)))
+