rdo-postgres 0.0.6 → 0.0.7

data/ext/rdo_postgres/types.h

@@ -0,0 +1,74 @@
+/*
+ * RDO Postgres Driver.
+ * Copyright © 2012 Chris Corbyn.
+ *
+ * See LICENSE file for details.
+ */
+
+/*
+ * Definitions for types taken from postgres/catalog/pg_type.h.
+ *
+ * Since not all the types we care about have named defines, they exist here.
+ */
+
+// integers
+#define RDO_PG_INT2OID 21
+#define RDO_PG_INT4OID 23
+#define RDO_PG_INT8OID 20
+
+// floats
+#define RDO_PG_FLOAT4OID 700
+#define RDO_PG_FLOAT8OID 701
+
+// precision decimals
+#define RDO_PG_NUMERICOID 1700
+
+// boolean
+#define RDO_PG_BOOLOID 16
+
+// bytea
+#define RDO_PG_BYTEAOID 17
+
+// dates
+#define RDO_PG_DATEOID 1082
+
+// timestamps
+#define RDO_PG_TIMESTAMPOID   1114
+#define RDO_PG_TIMESTAMPTZOID 1184
+
+// text/char
+#define RDO_PG_TEXTOID    25
+#define RDO_PG_CHAROID    18
+#define RDO_PG_VARCHAROID 1043
+#define RDO_PG_BPCHAROID  1042
+
+// text[]/char[]
+#define RDO_PG_TEXTARRAYOID    1009
+#define RDO_PG_CHARARRAYOID    1002
+#define RDO_PG_BPCHARARRAYOID  1014
+#define RDO_PG_VARCHARARRAYOID 1015
+
+// integer[]
+#define RDO_PG_INT2ARRAYOID 1005
+#define RDO_PG_INT4ARRAYOID 1007
+#define RDO_PG_INT8ARRAYOID 1016
+
+// float[]
+#define RDO_PG_FLOAT4ARRAYOID 1021
+#define RDO_PG_FLOAT8ARRAYOID 1022
+
+// numeric[]/decimal[]
+#define RDO_PG_NUMERICARRAYOID 1231
+
+// boolean[]
+#define RDO_PG_BOOLARRAYOID 1000
+
+// bytea[]
+#define RDO_PG_BYTEAARRAYOID 1001
+
+// date[]
+#define RDO_PG_DATEARRAYOID 1182
+
+// timestamp[]/timestamptz[]
+#define RDO_PG_TIMESTAMPARRAYOID   1115
+#define RDO_PG_TIMESTAMPTZARRAYOID 1185

data/lib/rdo/postgres.rb

@@ -6,9 +6,21 @@
 ##
 
 require "rdo"
+
 require "rdo/postgres/version"
 require "rdo/postgres/driver"
 
+require "rdo/postgres/array"
+require "rdo/postgres/array/text"
+require "rdo/postgres/array/integer"
+require "rdo/postgres/array/float"
+require "rdo/postgres/array/numeric"
+require "rdo/postgres/array/boolean"
+require "rdo/postgres/array/bytea"
+require "rdo/postgres/array/date"
+require "rdo/postgres/array/timestamp"
+require "rdo/postgres/array/timestamp_tz"
+
 # c extension
 require "rdo_postgres/rdo_postgres"
 

data/lib/rdo/postgres/array.rb

@@ -0,0 +1,84 @@
+##
+# RDO PostgreSQL driver.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module RDO
+  module Postgres
+    # Utility class used to handle PostgreSQL arrays.
+    #
+    # The default implementation assumes only String support.
+    # Subclasses override #parse_value and #format_value for typed-arrays.
+    #
+    # @example Turn a Ruby Array into a PostgreSQL array String
+    #   RDO::Postgres::Array.new(original).to_s
+    #   # => {"John Smith","Sarah Doe"}
+    #
+    # @example Turn a PostgreSQL array String into a Ruby Array
+    #   RDO::Postgres::Array.parse('{"John Smith","Sarah Doe"}').to_a
+    #   # => ["John Smith", "Sarah Doe"]
+    class Array < ::Array
+      class << self
+        # Read a PostgreSQL array in its string form.
+        #
+        # @param [String] str
+        #   an array string from postgresql
+        #
+        # @return [Array]
+        #   a Ruby Array for this string
+        def parse(str)
+          new.tap do |a|
+            a.replace(str[1...-1].split(",").map(&a.method(:parse_value_or_null)))
+          end
+        end
+      end
+
+      # Convert the Array to the format used by PostgreSQL.
+      #
+      # @return [String]
+      #   a postgresql array string
+      def to_s
+        "{#{map(&method(:format_value_or_null)).join(",")}}"
+      end
+
+      # Format an individual element in the Array for building into a String.
+      #
+      # The default implementation wraps quotes around the element.
+      #
+      # @param [Object] v
+      #   the Ruby type in the Array
+      #
+      # @return [String]
+      #   a String used to build the formatted array
+      def format_value(v)
+        %Q{"#{v.to_s.gsub('\\', '\\\\\\\\').gsub('"', '\\\\"')}"}
+      end
+
+      # Parse an individual element from the array.
+      #
+      # The default implementation parses as if it were text.
+      # Subclasses should override this.
+      #
+      # @param [String] s
+      #   the string form of the array element
+      #
+      # @return [Object]
+      #   the Ruby value
+      def parse_value(s)
+        s[0] == '"' ? s[1...-1].gsub(/\\(.)/, "\\1") : s
+      end
+
+      private
+
+      def format_value_or_null(v)
+        v.nil? ? "NULL" : format_value(v)
+      end
+
+      def parse_value_or_null(s)
+        s == "NULL" ? nil : parse_value(s)
+      end
+    end
+  end
+end

data/lib/rdo/postgres/array/boolean.rb

@@ -0,0 +1,21 @@
+##
+# RDO PostgreSQL driver.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module RDO
+  module Postgres
+    # Ruby handling for boolean[] type in PostgreSQL.
+    class Array::Boolean < Array
+      def parse_value(s)
+        s[0] == "t"
+      end
+
+      def format_value(v)
+        (!!v).to_s
+      end
+    end
+  end
+end

data/lib/rdo/postgres/array/bytea.rb

@@ -0,0 +1,22 @@
+##
+# RDO PostgreSQL driver.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module RDO
+  module Postgres
+    # Ruby handling for bytea[] type in PostgreSQL.
+    class Array::Bytea < Array
+      def parse_value(s)
+        # defined in ext/rdo_postgres/arrays.c
+      end
+
+      def format_value(v)
+        # defined in ext/rdo_postgres/arrays.c
+      end
+    end
+  end
+end
+

data/lib/rdo/postgres/array/date.rb

@@ -0,0 +1,17 @@
+##
+# RDO PostgreSQL driver.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module RDO
+  module Postgres
+    # Ruby handling for date[] type in PostgreSQL.
+    class Array::Date < Array
+      def parse_value(s)
+        RDO::Util.date(s)
+      end
+    end
+  end
+end

data/lib/rdo/postgres/array/float.rb

@@ -0,0 +1,30 @@
+##
+# RDO PostgreSQL driver.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module RDO
+  module Postgres
+    # Ruby handling for float[] type in PostgreSQL.
+    class Array::Float < Array
+      def parse_value(s)
+        case s
+        when "Infinity"
+          ::Float::INFINITY
+        when "-Infinity"
+          -::Float::INFINITY
+        when "NaN"
+          ::Float::NAN
+        else
+          s.to_f
+        end
+      end
+
+      def format_value(v)
+        v.to_s
+      end
+    end
+  end
+end

data/lib/rdo/postgres/array/integer.rb

@@ -0,0 +1,21 @@
+##
+# RDO PostgreSQL driver.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module RDO
+  module Postgres
+    # Ruby handling for integer[] type in PostgreSQL.
+    class Array::Integer < Array
+      def parse_value(s)
+        s.to_i
+      end
+
+      def format_value(v)
+        v.to_s
+      end
+    end
+  end
+end

data/lib/rdo/postgres/array/numeric.rb

@@ -0,0 +1,23 @@
+##
+# RDO PostgreSQL driver.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+require "bigdecimal"
+
+module RDO
+  module Postgres
+    # Ruby handling for numeric[] type in PostgreSQL.
+    class Array::Numeric < Array
+      def parse_value(s)
+        BigDecimal.new(s)
+      end
+
+      def format_value(v)
+        v.to_s
+      end
+    end
+  end
+end

data/lib/rdo/postgres/array/text.rb

@@ -0,0 +1,15 @@
+##
+# RDO PostgreSQL driver.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module RDO
+  module Postgres
+    # Ruby handling for text[] type in PostgreSQL.
+    class Array::Text < Array
+      # abstract class behaves as text anyway
+    end
+  end
+end

data/lib/rdo/postgres/array/timestamp.rb

@@ -0,0 +1,17 @@
+##
+# RDO PostgreSQL driver.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module RDO
+  module Postgres
+    # Ruby handling for timestamp[] type in PostgreSQL.
+    class Array::Timestamp < Array
+      def parse_value(s)
+        RDO::Util.date_time_without_zone(s)
+      end
+    end
+  end
+end

data/lib/rdo/postgres/array/timestamp_tz.rb

@@ -0,0 +1,17 @@
+##
+# RDO PostgreSQL driver.
+# Copyright © 2012 Chris Corbyn.
+#
+# See LICENSE file for details.
+##
+
+module RDO
+  module Postgres
+    # Ruby handling for timestamptz[] type in PostgreSQL.
+    class Array::TimestampTZ < Array
+      def parse_value(s)
+        RDO::Util.date_time_with_zone(s)
+      end
+    end
+  end
+end

data/lib/rdo/postgres/version.rb

@@ -7,6 +7,6 @@
 
 module RDO
   module Postgres
-    VERSION = "0.0.6"
+    VERSION = "0.0.7"
   end
 end

data/spec/postgres/array/boolean_spec.rb

@@ -0,0 +1,74 @@
+require "spec_helper"
+
+describe RDO::Postgres::Array::Boolean do
+  it "is a kind of ::Array" do
+    RDO::Postgres::Array::Boolean.new.should be_a_kind_of(::Array)
+  end
+
+  describe "#to_s" do
+    context "with an empty array" do
+      let(:arr) { RDO::Postgres::Array::Boolean[] }
+
+      it "returns {}" do
+        arr.to_s.should == '{}'
+      end
+    end
+
+    context "with an array of Booleans" do
+      let(:arr) { RDO::Postgres::Array::Boolean[true, false, false] }
+
+      it "uses truth literals" do
+        arr.to_s.should == '{true,false,false}'
+      end
+
+      context "containing nil" do
+        let(:arr) { RDO::Postgres::Array::Boolean[true, nil] }
+
+        it "uses NULL" do
+          arr.to_s.should == '{true,NULL}'
+        end
+      end
+    end
+  end
+
+  describe "#to_a" do
+    let(:arr) { RDO::Postgres::Array::Bytea["\x00\x11", "\x22\x33", "\x44\x55"] }
+
+    it "returns a Ruby ::Array" do
+      arr.to_a.should == ["\x00\x11", "\x22\x33", "\x44\x55"]
+    end
+  end
+
+  describe ".parse" do
+    let(:str) { '{}' }
+    let(:arr) { RDO::Postgres::Array::Bytea.parse(str) }
+
+    it "returns a RDO::Postgres::Array::Bytea" do
+      arr.should be_a_kind_of(RDO::Postgres::Array::Bytea)
+    end
+
+    context "with an empty array string" do
+      let(:str) { '{}' }
+
+      it "returns an empty Array" do
+        arr.should be_empty
+      end
+    end
+
+    context "with an array of byteas" do
+      let(:str) { '{"\\\\x0011","\\\\x2233"}' }
+
+      it "returns an Array of Strings" do
+        arr.to_a.should == ["\x00\x11", "\x22\x33"]
+      end
+    end
+
+    context "with an array containing NULL" do
+      let(:str) { '{NULL,NULL,"\\\\x0011"}' }
+
+      it "uses nil as the value" do
+        arr.to_a.should == [nil, nil, "\x00\x11"]
+      end
+    end
+  end
+end