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