do_postgres 0.2.0

data/ext/postgres_c.i

@@ -0,0 +1,73 @@
+%module postgres_c
+%{
+#include "libpq-fe.h"
+%}
+
+%include "/opt/local/include/postgresql82/libpq-fe.h"
+%include "/opt/local/include/postgresql82/postgres_ext.h"
+
+#define BOOLOID 16
+#define BYTEAOID 17
+#define CHAROID 18
+#define NAMEOID 19
+#define INT8OID 20
+#define INT2OID 21
+#define INT2VECTOROID 22
+#define INT4OID 23
+#define REGPROCOID 24
+#define TEXTOID 25
+#define OIDOID 26
+#define TIDOID 27
+#define XIDOID 28
+#define CIDOID 29
+#define OIDVECTOROID 30
+#define PG_TYPE_RELTYPE_OID 71
+#define PG_ATTRIBUTE_RELTYPE_OID 75
+#define PG_PROC_RELTYPE_OID 81
+#define PG_CLASS_RELTYPE_OID 83
+#define POINTOID 600
+#define LSEGOID 601
+#define PATHOID 602
+#define BOXOID 603
+#define POLYGONOID 604
+#define LINEOID 628
+#define FLOAT4OID 700
+#define FLOAT8OID 701
+#define ABSTIMEOID 702
+#define RELTIMEOID 703
+#define TINTERVALOID 704
+#define UNKNOWNOID 705
+#define CIRCLEOID 718
+#define CASHOID 790
+#define MACADDROID 829
+#define INETOID 869
+#define CIDROID 650
+#define INT4ARRAYOID 1007
+#define ACLITEMOID 1033
+#define BPCHAROID 1042
+#define VARCHAROID 1043
+#define DATEOID 1082
+#define TIMEOID 1083
+#define TIMESTAMPOID 1114
+#define TIMESTAMPTZOID 1184
+#define INTERVALOID 1186
+#define TIMETZOID 1266
+#define BITOID 1560
+#define VARBITOID 1562
+#define NUMERICOID 1700
+#define REFCURSOROID 1790
+#define REGPROCEDUREOID 2202
+#define REGOPEROID 2203
+#define REGOPERATOROID 2204
+#define REGCLASSOID 2205
+#define REGTYPEOID 2206
+#define RECORDOID 2249
+#define CSTRINGOID 2275
+#define ANYOID 2276
+#define ANYARRAYOID 2277
+#define VOIDOID 2278
+#define TRIGGEROID 2279
+#define LANGUAGE_HANDLEROID 2280
+#define INTERNALOID 2281
+#define OPAQUEOID 2282
+#define ANYELEMENTOID 2283

data/lib/do_postgres.rb

@@ -0,0 +1,196 @@
+require 'postgres_c'
+require 'data_objects'
+
+module DataObject
+  module Postgres
+    TYPES = Hash[*Postgres_c.constants.select {|x| x.include?("OID")}.map {|x| [Postgres_c.const_get(x), x.gsub(/_?OID$/, "")]}.flatten]
+    QUOTE_STRING = "'"
+    QUOTE_COLUMN = "\""
+    
+    class Connection < DataObject::Connection
+      attr_reader :db
+      
+      def initialize(connection_string)
+        @state = STATE_CLOSED        
+        @connection_string = connection_string
+      end
+      
+      def open
+        @db = Postgres_c.PQconnectdb(@connection_string)
+        if Postgres_c.PQstatus(@db) != Postgres_c::CONNECTION_OK
+          raise ConnectionFailed, "The connection with connection string #{@connection_string} failed\n#{Postgres_c.PQerrorMessage(@db)}"
+        end
+        @state = STATE_OPEN
+        true
+      end
+      
+      def close
+        Postgres_c.PQfinish(@db)
+        @state = STATE_CLOSED
+        true
+      end
+      
+      def create_command(text)
+        Command.new(self, text)
+      end
+      
+    end
+    
+    class Reader < DataObject::Reader
+      
+      def initialize(db, reader)
+        @reader = reader
+        case Postgres_c.PQresultStatus(reader)
+        when Postgres_c::PGRES_COMMAND_OK
+          @records_affected = Postgres_c.PQcmdTuples(reader).to_i
+          close
+        when Postgres_c::PGRES_TUPLES_OK
+          @fields, @field_types = [], []
+          @field_count = Postgres_c.PQnfields(@reader)
+          i = 0
+          while(i < @field_count)
+            @field_types.push(Postgres_c.PQftype(@reader, i))
+            @fields.push(Postgres_c.PQfname(@reader, i))
+            i += 1
+          end
+          @rows = Postgres_c.PQntuples(@reader)
+          @has_rows = @rows > 0
+          @cursor = 0
+          @state = STATE_OPEN
+        end
+      end
+      
+      def real_close
+        Postgres_c.PQclear(@reader)
+      end
+      
+      def data_type_name(col)
+        
+      end
+      
+      def name(col)
+        super
+        Postgres_c.PQfname(@reader, col)
+      end
+      
+      def get_index(name)
+        super
+        @fields.index(name)
+      end
+
+      def null?(idx)
+        super
+        Postgres_c.PQgetisnull(@reader, @cursor, idx) != 0
+      end
+      
+      def item(idx)
+        super
+        val = Postgres_c.PQgetvalue(@reader, @cursor, idx)
+        typecast(val, @field_types[idx])
+      end
+      
+      def each
+        return unless has_rows?
+        
+        while(true) do
+          yield
+          break unless self.next
+        end
+      end
+      
+      def next
+        super   
+        if @cursor >= @rows - 1
+          @cursor = nil
+          close
+          return nil
+        end
+        @cursor += 1
+        true
+      end
+      
+      protected
+      def native_type(col)
+        TYPES[Postgres_c.PQftype(@reader, col)]
+      end
+      
+      def typecast(val, field_type)
+        return nil if val.nil?
+        case TYPES[field_type]
+          when "BOOL"
+            val == "t"
+          when "INT2", "INT4", "OID", "TID", "XID", "CID", "INT8"
+            val.to_i
+          when "FLOAT4", "FLOAT8", "NUMERIC", "CASH"
+            val.to_f
+          when "TIMESTAMP", "TIMETZ", "TIMESTAMPTZ"
+            DateTime.parse(val) rescue nil
+          when "TIME"
+            DateTime.parse(val).to_time rescue nil
+          when "DATE"
+            Date.parse(val) rescue nil
+          else
+            val
+        end
+      end
+      
+    end
+    
+    class ResultData < DataObject::ResultData
+      
+      def last_insert_row
+        @last_insert_row ||= begin
+          reader = @conn.create_command("select lastval()").execute_reader
+          reader.item(0).to_i
+        rescue QueryError
+          raise NoInsertError, "You tried to get the last inserted row without doing an insert\n#{Postgres_c.PQerrorMessage(@conn.db)}"
+        ensure
+          reader and reader.close
+        end
+      end      
+      
+    end
+    
+    class Command < DataObject::Command
+      
+      def execute_reader(*args)
+        super
+        sql = escape_sql(args)
+        @connection.logger.debug { sql }
+        ptr = Postgres_c.PQexec(@connection.db, sql)
+        unless [Postgres_c::PGRES_COMMAND_OK, Postgres_c::PGRES_TUPLES_OK].include?(Postgres_c.PQresultStatus(ptr))
+          raise QueryError, "Your query failed.\n#{Postgres_c.PQerrorMessage(@connection.db)}QUERY: \"#{sql}\""
+        else
+          reader = Reader.new(@connection.db, ptr)
+          if block_given?
+            return_value = yield(reader)
+            reader.close
+            return_value
+          else
+            reader
+          end
+        end
+      end
+      
+      def execute_non_query(*args)
+        super
+        sql = escape_sql(args)
+        @connection.logger.debug { sql }
+        results = Postgres_c.PQexec(@connection.db, sql)
+        status = Postgres_c.PQresultStatus(results)
+        if status == Postgres_c::PGRES_TUPLES_OK
+          Postgres_c.PQclear(results)
+          raise QueryError, "Your query failed or you tried to execute a SELECT query through execute_non_reader\n#{Postgres_c.PQerrorMessage(@connection.db)}\nQUERY: \"#{sql}\""
+        elsif status != Postgres_c::PGRES_COMMAND_OK
+          Postgres_c.PQclear(results)
+          raise QueryError, "Your query failed.\n#{Postgres_c.PQerrorMessage(@connection.db)}\nQUERY: \"#{sql}\""
+        end
+        rows_affected = Postgres_c.PQcmdTuples(results).to_i
+        Postgres_c.PQclear(results)
+        ResultData.new(@connection, rows_affected)
+      end
+      
+    end
+    
+  end
+end

metadata

@@ -0,0 +1,63 @@
+--- !ruby/object:Gem::Specification 
+rubygems_version: 0.9.4
+specification_version: 1
+name: do_postgres
+version: !ruby/object:Gem::Version 
+  version: 0.2.0
+date: 2007-11-11 00:00:00 -08:00
+summary: A DataObject.rb driver for postgres
+require_paths: 
+- lib
+email: wycats@gmail.com
+homepage: http://dataobjects.devjavu.com
+rubyforge_project: 
+description: A DataObject.rb driver for postgres
+autorequire: do_postgres
+default_executable: 
+bindir: bin
+has_rdoc: true
+required_ruby_version: !ruby/object:Gem::Version::Requirement 
+  requirements: 
+  - - ">"
+    - !ruby/object:Gem::Version 
+      version: 0.0.0
+  version: 
+platform: ruby
+signing_key: 
+cert_chain: 
+post_install_message: 
+authors: 
+- Yehuda Katz
+files: 
+- LICENSE
+- README
+- Rakefile
+- TODO
+- lib/do_postgres.rb
+- ext/extconf.rb
+- ext/postgres_c.c
+- ext/postgres_c.i
+test_files: []
+
+rdoc_options: []
+
+extra_rdoc_files: 
+- README
+- LICENSE
+- TODO
+executables: []
+
+extensions: []
+
+requirements: []
+
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: data_objects
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Version::Requirement 
+    requirements: 
+    - - ">"
+      - !ruby/object:Gem::Version 
+        version: 0.0.0
+    version: