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       #+ignore
  59       ((and (consp (first in-phrase))
  60             (consp (second (first in-phrase)))
  61             (eq 'quote (first (second (first in-phrase))))
  62             (symbolp (second (second (first in-phrase)))))
  63
  64        (let ((result-var (ansi-loop::loop-gentemp
  65                               'loop-record-result-))
  66              (step-var (ansi-loop::loop-gentemp 'loop-record-step-)))
  67          `(((,variable nil ,@(and data-type (list data-type)))
  68             (,result-var ,(first in-phrase))
  69             (,step-var nil))
  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 (consp (car in-phrase))
 164             (consp (second (car in-phrase)))
 165             (eq 'quote (first (second (car in-phrase))))
 166             (symbolp (second (second (car in-phrase)))))
 167        (loop-error "object query not yet supported"))
 168
 169       ((consp iter-var)
 170        (let ((query-var (gensym "LOOP-RECORD-"))
 171              (db-var (gensym "LOOP-RECORD-DATABASE-"))
 172              (result-set-var (gensym "LOOP-RECORD-RESULT-SET-"))
 173              (step-var (gensym "LOOP-RECORD-STEP-")))
 174          (values
 175           t
 176           nil
 177           `(,@(mapcar (lambda (v) `(,v nil)) iter-var)
 178             (,query-var ,in-phrase)
 179             (,db-var ,(first from-phrase))
 180             (,result-set-var nil)
 181             (,step-var nil))
 182           `((multiple-value-bind (%rs %cols)
 183                 (clsql-base:database-query-result-set ,query-var ,db-var :result-types :auto)
 184               (setq ,result-set-var %rs ,step-var (make-list %cols))))
 185           ()
 186           ()
 187           `((unless (clsql-base:database-store-next-row ,result-set-var ,db-var ,step-var)
 188               (when ,result-set-var
 189                 (clsql-base:database-dump-result-set ,result-set-var ,db-var))
 190               t))
 191           `(,iter-var ,step-var)
 192           `((unless (clsql-base:database-store-next-row ,result-set-var ,db-var ,step-var)
 193               (when ,result-set-var
 194                 (clsql-base:database-dump-result-set ,result-set-var ,db-var))
 195               t))
 196           `(,iter-var ,step-var)
 197           ()
 198           ()))))))
 199