X-Git-Url: http://git.kpe.io/?p=clsql.git;a=blobdiff_plain;f=sql%2Floop-extension.lisp;h=32a356c28fdfcb11bc609648cfe9709c03b210bf;hp=c3a2b9f2e930f37be93d1cce60f903d369321955;hb=HEAD;hpb=edd1963395a5b5e5f91ef975fcd329975ae367e2

diff --git a/sql/loop-extension.lisp b/sql/loop-extension.lisp
index c3a2b9f..32a356c 100644
--- a/sql/loop-extension.lisp
+++ b/sql/loop-extension.lisp
@@ -2,97 +2,246 @@
 ;;;; *************************************************************************
 ;;;; FILE IDENTIFICATION
 ;;;;
-;;;; Name:          loop-extension.lisp
-;;;; 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$
-;;;;
-;;;; The functions in this file were orignally distributed in the
-;;;; MaiSQL package in the file sql/sql.cl
 ;;;; *************************************************************************
 
-(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)))
+
+#+(or allegro sbcl)
+(defun ansi-loop::loop-gentemp (&optional (pref 'loopva-))
+  (gensym (string pref)))
 
-;;;; MIT-LOOP extension
+#+clisp
+(eval-when (:compile-toplevel :load-toplevel :execute)
+  (when (find-package "ANSI-LOOP") (push :clisp-aloop cl:*features*)))
 
-#+(or cmu scl)
+#+(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))
+        (from-phrase nil))
     (loop for (prep . rest) in prep-phrases
-	  do
-	  (case prep
-	    ((:in :of)
-	     (when in-phrase
-	       (ansi-loop::loop-error
-		"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)))
-	     (setq from-phrase rest))
-	    (t
-	     (ansi-loop::loop-error
-	      "Unknown preposition: ~S." prep))))
+          do
+          (case prep
+            ((:in :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))
+            ((: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
-      (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-))
-	     (result-set-var (ansi-loop::loop-gentemp
-			      '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))
-	       ansi-loop::*loop-epilogue*)
-	 `(((,variable nil ,data-type) (,query-var ,(first in-phrase))
-	    (,db-var ,(first from-phrase))
-	    (,result-set-var nil)
-	    (,step-var nil))
-	   ((multiple-value-bind (%rs %cols)
-		(database-query-result-set ,query-var ,db-var)
-	      (setq ,result-set-var %rs ,step-var (make-list %cols))))
-	   ()
-	   ()
-	   (not (database-store-next-row ,result-set-var ,db-var ,step-var))
-	   (,variable ,step-var)
-	   (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))
-	   ()))))))
-
-#+(or cmu scl)
+             (db-var (ansi-loop::loop-gentemp 'loop-record-database-))
+             (result-set-var (ansi-loop::loop-gentemp
+                              '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))
+               ansi-loop::*loop-epilogue*)
+         `(((,variable nil ,@(and data-type (list data-type)))
+            (,query-var ,(first in-phrase))
+            (,db-var ,(first from-phrase))
+            (,result-set-var t)
+            (,step-var nil))
+           ((multiple-value-bind (%rs %cols)
+                (database-query-result-set ,query-var ,db-var :result-types :auto)
+              (setq ,result-set-var %rs ,step-var (make-list %cols))))
+           ()
+           ()
+           (not (database-store-next-row ,result-set-var ,db-var ,step-var))
+           (,variable ,step-var)
+           (not ,result-set-var)
+           ()
+           (not (database-store-next-row ,result-set-var ,db-var ,step-var))
+           (,variable ,step-var)))))))
+
+#+(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)
+                          '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 iter-var-data-type 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*)))
+