X-Git-Url: http://git.kpe.io/?a=blobdiff_plain;f=base%2Floop-extension.lisp;h=526bafd863c6684af279afa42f9294b052b7c7eb;hb=550cc94ed84c6cfc61830c25fac2b39e92393b06;hp=a78086baf5adce1e12476b1959512ad61284251f;hpb=7e20a5285824bbbd2aa6fae61da73465ed2e8557;p=clsql.git

diff --git a/base/loop-extension.lisp b/base/loop-extension.lisp
index a78086b..526bafd 100644
--- a/base/loop-extension.lisp
+++ b/base/loop-extension.lisp
@@ -49,10 +49,42 @@
       (ansi-loop::loop-error "Missing OF or IN iteration path."))
     (unless from-phrase
       (setq from-phrase '(clsql-base:*default-database*)))
+
+    (unless (consp variable)
+      (setq variable (list variable)))
+
     (cond
       ;; Object query resulting in a list of returned object instances
-      ((consp in-phrase)
-       (error "object query not yet supported"))
+      ((and (consp (first in-phrase))
+	    (string-equal "select" (symbol-name (caar 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 ,(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-))
@@ -69,7 +101,7 @@
 	    (,result-set-var nil)
 	    (,step-var nil))
 	   ((multiple-value-bind (%rs %cols)
-		(clsql-base:database-query-result-set ,query-var ,db-var)
+		(clsql-base:database-query-result-set ,query-var ,db-var :result-types :auto)
 	      (setq ,result-set-var %rs ,step-var (make-list %cols))))
 	   ()
 	   ()
@@ -78,30 +110,7 @@
 	   (not ,result-set-var)
 	   ()
 	   (not (clsql-base: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
-		 (clsql-base: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 nil))
-	   ((multiple-value-bind (%rs %cols)
-		(clsql-base:database-query-result-set ,query-var ,db-var)
-	      (setq ,result-set-var %rs ,variable (make-list %cols))))
-	   ()
-	   ()
-	   (not (clsql-base:database-store-next-row ,result-set-var ,db-var ,variable))
-	   ()
-	   (not ,result-set-var)
-	   ()
-	   (not (clsql-base:database-store-next-row ,result-set-var ,db-var ,variable))
-	   ()))))))
+	   (,variable ,step-var)))))))
 
 #+(or cmu scl sbcl openmcl allegro)
 (ansi-loop::add-loop-path '(record records tuple tuples)
@@ -146,10 +155,43 @@
     (unless from-phrase
       (setq from-phrase '(clsql-base:*default-database*)))
 
+    (unless (consp iter-var)
+      (setq iter-var (list iter-var)))
+
     (cond
       ;; Object query resulting in a list of returned object instances
-      ((consp in-phrase)
-       (error "Object query not yet supported."))
+      ((and (string-equal "select" (symbol-name (car 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 ,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-"))
@@ -165,7 +207,7 @@
 	    (,result-set-var nil)
 	    (,step-var nil))
 	  `((multiple-value-bind (%rs %cols)
-		(clsql-base:database-query-result-set ,query-var ,db-var)
+		(clsql-base:database-query-result-set ,query-var ,db-var :result-types :auto)
 	      (setq ,result-set-var %rs ,step-var (make-list %cols))))
 	  ()
 	  ()
@@ -180,32 +222,5 @@
 	      t))
 	  `(,iter-var ,step-var)
 	  ()
-	  ())))
-      (t
-       (let ((query-var (gensym "LOOP-RECORD-"))
-	     (db-var (gensym "LOOP-RECORD-DATABASE-"))
-	     (result-set-var (gensym "LOOP-RECORD-RESULT-SET-")))
-	 (values
-	  t
-	  nil
-	  `((,iter-var nil ,iter-var-data-type) (,query-var ,in-phrase)
-	    (,db-var ,(first from-phrase))
-	    (,result-set-var nil))
-	  `((multiple-value-bind (%rs %cols)
-		(clsql-base:database-query-result-set ,query-var ,db-var)
-	      (setq ,result-set-var %rs ,iter-var (make-list %cols))))
-	  ()
-	  ()
-	  `((unless (clsql-base:database-store-next-row ,result-set-var ,db-var ,iter-var)
-	      (when ,result-set-var
-		(clsql-base:database-dump-result-set ,result-set-var ,db-var))
-	      t))
-	   ()
-	  `((unless (clsql-base:database-store-next-row ,result-set-var ,db-var ,iter-var)
-	      (when ,result-set-var
-		(clsql-base:database-dump-result-set ,result-set-var ,db-var))
-	      t))
-	  ()
-	  ()
 	  ()))))))