pg_data_encoder 0.0.2 → 0.0.3

data/lib/pg_data_encoder/encode_for_copy.rb

@@ -66,6 +66,45 @@ module PgDataEncoder
         buf = field.encode("UTF-8")
         io.write([buf.bytesize].pack("N"))
         io.write(buf)
+      when Array
+        array_io = StringIO.new
+        case field[0]
+        when String
+          array_io.write([1].pack("N"))  # unknown
+          array_io.write([0].pack("N"))  # unknown
+
+          array_io.write([25].pack("N"))  # I think is used to determine string data type
+          array_io.write([field.size].pack("N"))
+          array_io.write([1].pack("N"))   # forcing single dimention array for now
+          
+          field.each_with_index {|val, index|
+            buf = val.to_s.encode("UTF-8")
+            array_io.write([buf.bytesize].pack("N"))
+            array_io.write(buf)
+            
+          }
+        when Integer
+          array_io.write([1].pack("N"))  # unknown
+          array_io.write([0].pack("N"))  # unknown
+
+          array_io.write([23].pack("N"))  # I think is used to detemine int data type
+          array_io.write([field.size].pack("N"))
+          array_io.write([1].pack("N"))   # forcing single dimention array for now
+          
+          field.each_with_index {|val, index|
+            buf = [val.to_i].pack("N")
+            array_io.write([buf.bytesize].pack("N"))
+            array_io.write(buf)
+            
+          }
+        else
+          raise Exception.new("Arrays support int or string only")
+        end
+
+        io.write([array_io.pos].pack("N"))
+        
+        
+        io.write(array_io.string)
       when Hash
         raise Exception.new("Hash's can't contain hashes") if depth > 0
         hash_io = StringIO.new
@@ -89,4 +128,4 @@ module PgDataEncoder
     end
 
   end
-end
+end

data/lib/pg_data_encoder/version.rb

@@ -1,3 +1,3 @@
 module PgDataEncoder
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/spec/fixtures/3_col_array.txt

@@ -0,0 +1 @@
+1	hi	{hi,there,rubyist}

data/spec/verify_data_formats_spec.rb

@@ -27,6 +27,41 @@ describe "generating data" do
     str.should == existing_data
   end
 
+  it 'should encode array data correctly' do
+    encoder = PgDataEncoder::EncodeForCopy.new
+    encoder.add [1, "hello", ["hi", "jim"]]
+    encoder.close
+    io = encoder.get_io
+    existing_data = filedata("array_with_two.dat")
+    str = io.read
+    io.class.name.should == "StringIO"
+    str.force_encoding("ASCII-8BIT")
+    str.should == existing_data
+  end
+
+  it 'should encode array data from tempfile correctly' do
+    encoder = PgDataEncoder::EncodeForCopy.new(:use_tempfile => true)
+    encoder.add [1, "hi", ["hi", "there", "rubyist"]]
+    encoder.close
+    io = encoder.get_io
+    existing_data = filedata("3_column_array.dat")
+    str = io.read
+    io.class.name.should == "Tempfile"
+    str.force_encoding("ASCII-8BIT")
+    str.should == existing_data
+  end
+
+  it 'should encode integer array data from tempfile correctly' do
+    encoder = PgDataEncoder::EncodeForCopy.new(:use_tempfile => true)
+    encoder.add [[1,2,3]]
+    encoder.close
+    io = encoder.get_io
+    existing_data = filedata("intarray.dat")
+    str = io.read
+    io.class.name.should == "Tempfile"
+    str.force_encoding("ASCII-8BIT")
+    str.should == existing_data
+  end
 
   it 'should encode timestamp data correctly' do
     encoder = PgDataEncoder::EncodeForCopy.new
@@ -80,4 +115,4 @@ describe "generating data" do
     str.should == existing_data
   end
 
-end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pg_data_encoder
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-09-09 00:00:00.000000000 Z
+date: 2013-10-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -61,9 +61,13 @@ files:
 - lib/pg_data_encoder/encode_for_copy.rb
 - lib/pg_data_encoder/version.rb
 - pg_data_encoder.gemspec
+- spec/fixtures/3_col_array.txt
 - spec/fixtures/3_col_hstore.dat
 - spec/fixtures/3_col_hstore.txt
+- spec/fixtures/3_column_array.dat
+- spec/fixtures/array_with_two.dat
 - spec/fixtures/float.dat
+- spec/fixtures/intarray.dat
 - spec/fixtures/output.dat
 - spec/fixtures/timestamp.dat
 - spec/spec_helper.rb
@@ -94,9 +98,13 @@ specification_version: 3
 summary: for faster input of data into postgres you can use this to generate the binary
   import and run COPY FROM
 test_files:
+- spec/fixtures/3_col_array.txt
 - spec/fixtures/3_col_hstore.dat
 - spec/fixtures/3_col_hstore.txt
+- spec/fixtures/3_column_array.dat
+- spec/fixtures/array_with_two.dat
 - spec/fixtures/float.dat
+- spec/fixtures/intarray.dat
 - spec/fixtures/output.dat
 - spec/fixtures/timestamp.dat
 - spec/spec_helper.rb