r1683: *** empty log message ***

[clsql.git] / interfaces / postgresql-socket / postgresql-socket-sql.cl

diff --git a/interfaces/postgresql-socket/postgresql-socket-sql.cl b/interfaces/postgresql-socket/postgresql-socket-sql.cl
index b98836f93758826948993560f18e445d86c09311..0c99be07dfba6408aacf1bb9fa93c6532383c4b2 100644 (file)
--- a/interfaces/postgresql-socket/postgresql-socket-sql.cl
+++ b/interfaces/postgresql-socket/postgresql-socket-sql.cl
@@ -8,7 +8,7 @@
 ;;;;                Original code by Pierre R. Mai 
 ;;;; Date Started:  Feb 2002
 ;;;;
-;;;; $Id: postgresql-socket-sql.cl,v 1.4 2002/03/24 22:25:51 kevin Exp $
+;;;; $Id: postgresql-socket-sql.cl,v 1.8 2002/03/27 12:09:39 kevin Exp $
 ;;;;
 ;;;; This file, part of CLSQL, is Copyright (c) 2002 by Kevin M. Rosenberg
 ;;;; and Copyright (c) 1999-2001 by Pierre R. Mai
@@ -21,7 +21,6 @@
 (declaim (optimize (debug 3) (speed 3) (safety 1) (compilation-speed 0)))
 (in-package :cl-user)
 
-
 (defpackage :clsql-postgresql-socket
     (:use :common-lisp :clsql-sys :postgresql-socket)
     (:export #:postgresql-socket-database)
@@ -29,6 +28,75 @@
 
 (in-package :clsql-postgresql-socket)
 
+;; Field type conversion
+
+(defun make-type-list-for-auto (cursor)
+  (let* ((fields (postgresql-cursor-fields cursor))
+        (num-fields (length fields))
+        (new-types '()))
+    (dotimes (i num-fields)
+      (declare (fixnum i))
+      (push (canonical-field-type fields i) new-types))
+    (nreverse new-types)))
+
+(defun canonical-field-type (fields index)
+  "Extracts canonical field type from fields list"
+  (let ((oid (cadr (nth index fields))))
+    (case oid
+      ((#.pgsql-ftype#bytea
+       #.pgsql-ftype#int2
+       #.pgsql-ftype#int4)
+       :int32)
+      (#.pgsql-ftype#int8
+       :int64)
+      ((#.pgsql-ftype#float4
+       #.pgsql-ftype#float8)
+       :double)
+      (otherwise
+       t))))
+
+(defun canonicalize-types (types cursor)
+  (let ((auto-list (make-type-list-for-auto cursor)))
+    (cond
+      ((listp types)
+       (canonicalize-type-list types auto-list))
+      ((eq types :auto)
+       auto-list)
+      (t
+       nil))))
+
+(defun canonicalize-type-list (types auto-list)
+  "Ensure a field type list meets expectations.
+Duplicated from clsql-uffi package so that this interface
+doesn't depend on UFFI."
+  (let ((length-types (length types))
+       (new-types '()))
+    (loop for i from 0 below (length auto-list)
+         do
+         (if (>= i length-types)
+             (push t new-types) ;; types is shorted than num-fields
+             (push
+              (case (nth i types)
+                (:int
+                 (case (nth i auto-list)
+                   (:int32
+                    :int32)
+                   (:int64
+                    :int64)
+                   (t
+                    t)))
+                (:double
+                 (case (nth i auto-list)
+                   (:double
+                    :double)
+                   (t
+                    t)))
+                (t
+                 t))
+              new-types)))
+    (nreverse new-types)))
+
+
 (defun convert-to-clsql-warning (database condition)
   (warn 'clsql-database-warning :database database
        :message (postgresql-condition-message condition)))
@@ -110,7 +178,7 @@
   (close-postgresql-connection (database-connection database))
   t)
 
-(defmethod database-query (expression (database postgresql-socket-database) field-types)
+(defmethod database-query (expression (database postgresql-socket-database) types)
   (let ((connection (database-connection database)))
     (with-postgresql-handlers (database expression)
       (start-query-execution connection expression)
@@ -123,7 +191,8 @@
                 :expression expression
                 :errno 'missing-result
                 :error "Didn't receive result cursor for query."))
-       (loop for row = (read-cursor-row cursor)
+       (setq types (canonicalize-types types cursor))
+       (loop for row = (read-cursor-row cursor types)
              while row
              collect row
              finally
@@ -172,10 +241,10 @@
 (defstruct postgresql-socket-result-set
   (done nil)
   (cursor nil)
-  (field-types nil))
+  (types nil))
 
 (defmethod database-query-result-set (expression (database postgresql-socket-database) 
-                                     &key full-set field-types
+                                     &key full-set types
      )
   (declare (ignore full-set))
   (let ((connection (database-connection database)))
@@ -193,7 +262,7 @@
        (values (make-postgresql-socket-result-set
                 :done nil 
                 :cursor cursor
-                :field-types field-types)
+                :types (canonicalize-types types cursor))
                (length (postgresql-cursor-fields cursor)))))))
 
 (defmethod database-dump-result-set (result-set
@@ -210,7 +279,10 @@
                                    list)
   (let ((cursor (postgresql-socket-result-set-cursor result-set)))
     (with-postgresql-handlers (database)
-      (if (copy-cursor-row cursor list)
+      (if (copy-cursor-row cursor 
+                          list
+                          (postgresql-socket-result-set-types
+                           result-set))
          t
          (prog1 nil
            (setf (postgresql-socket-result-set-done result-set) t)