pg_data_encoder 0.0.5 → 0.0.6

data/README.md

@@ -71,6 +71,7 @@ or
   * Boolean
   * Floats (double precision)
   * Timestamp
+  * Date
   * Array (integer and string single dimension)
 
 ## Contributing

data/lib/pg_data_encoder/encode_for_copy.rb

@@ -1,7 +1,8 @@
 require 'tempfile'
 require 'stringio'
 module PgDataEncoder
-  POSTGRES_EPOCH_DATE = (Time.utc(2000,1,1).to_f * 1_000_000).to_i
+  POSTGRES_EPOCH_TIME = (Time.utc(2000,1,1).to_f * 1_000_000).to_i
+
   class EncodeForCopy
     def initialize(options = {})
       @options = options
@@ -127,7 +128,11 @@ module PgDataEncoder
         io.write([hash_io.pos].pack("N"))  # assumed identifier for hstore column
         io.write(hash_io.string)
       when Time
-        buf = [(field.to_f * 1_000_000 - POSTGRES_EPOCH_DATE).to_i].pack("L!>")
+        buf = [(field.to_f * 1_000_000 - POSTGRES_EPOCH_TIME).to_i].pack("L!>")
+        io.write([buf.bytesize].pack("N"))
+        io.write(buf)
+      when Date
+        buf = [(field - Date.new(2000,1,1)).to_i].pack("N")
         io.write([buf.bytesize].pack("N"))
         io.write(buf)
       else

data/lib/pg_data_encoder/version.rb

@@ -1,3 +1,3 @@
 module PgDataEncoder
-  VERSION = "0.0.5"
+  VERSION = "0.0.6"
 end

data/spec/verify_data_formats_spec.rb

@@ -123,6 +123,42 @@ describe "generating data" do
     str.should == existing_data
   end
 
+  it 'should encode date data correctly' do
+    encoder = PgDataEncoder::EncodeForCopy.new
+    encoder.add [Date.parse("1900-12-03")]
+    encoder.close
+    io = encoder.get_io
+    existing_data = filedata("date.dat")
+    str = io.read
+    io.class.name.should == "StringIO"
+    str.force_encoding("ASCII-8BIT")
+    str.should == existing_data
+  end
+
+  it 'should encode date data correctly for years > 2000' do
+    encoder = PgDataEncoder::EncodeForCopy.new
+    encoder.add [Date.parse("2033-01-12")]
+    encoder.close
+    io = encoder.get_io
+    existing_data = filedata("date2000.dat")
+    str = io.read
+    io.class.name.should == "StringIO"
+    str.force_encoding("ASCII-8BIT")
+    str.should == existing_data
+  end
+
+  it 'should encode date data correctly in the 70s' do
+    encoder = PgDataEncoder::EncodeForCopy.new
+    encoder.add [Date.parse("1971-12-11")]
+    encoder.close
+    io = encoder.get_io
+    existing_data = filedata("date2.dat")
+    str = io.read
+    io.class.name.should == "StringIO"
+    str.force_encoding("ASCII-8BIT")
+    str.should == existing_data
+  end
+
   it 'should encode timestamp data correctly' do
     encoder = PgDataEncoder::EncodeForCopy.new
     encoder.add [Time.parse("2013-06-11 15:03:54.62605 UTC")]

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pg_data_encoder
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.6
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-10-14 00:00:00.000000000 Z
+date: 2013-10-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -70,6 +70,9 @@ files:
 - spec/fixtures/array_with_two2.dat
 - spec/fixtures/big_str_array.dat
 - spec/fixtures/big_str_array2.dat
+- spec/fixtures/date.dat
+- spec/fixtures/date2.dat
+- spec/fixtures/date2000.dat
 - spec/fixtures/falseclass.dat
 - spec/fixtures/float.dat
 - spec/fixtures/intarray.dat
@@ -115,6 +118,9 @@ test_files:
 - spec/fixtures/array_with_two2.dat
 - spec/fixtures/big_str_array.dat
 - spec/fixtures/big_str_array2.dat
+- spec/fixtures/date.dat
+- spec/fixtures/date2.dat
+- spec/fixtures/date2000.dat
 - spec/fixtures/falseclass.dat
 - spec/fixtures/float.dat
 - spec/fixtures/intarray.dat