fluent-plugin-dynamodb-alt 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: def6475b444e76ca66d619f4fea1f814a4c64431
-  data.tar.gz: 3f83b0bd39fe53f813f2b8273b5e2e39392145da
+  metadata.gz: c7a92c9fe9a41c6c1c7360d796a22cf7fd235259
+  data.tar.gz: 090c2a1de0e08e5730119dcc49e896556493f8b1
 SHA512:
-  metadata.gz: 552cd086584a6edcf30139cb6b6ea4b9081c2913592cd1381e3abe360a1f52958b4a0a4e2f62597d16ee4bc2a98c46536047a552403c792fbcf1f55e5fd04b07
-  data.tar.gz: 6eab1ab0f0540f685264fb2ec7190e2c177ebd7a41ea31dbb149b09aa63390fe17e3727fb2279dc00eb7310cebcb0da392b2c3796c3569407524fb47aa795cec
+  metadata.gz: 84e9d9b2747a0bf6570efaa56e1ec49892032572cfa1380cc424bd30e5e9622ec3b9f1d11eca669a0189d2dc5d96ee3e1578b35b8a997af544dcb043d01e6be7
+  data.tar.gz: e73846a99a68e9033ee3a92a9300b7c987d148b5464e8655e9ecbcf6fadabe843356641d845483015ea591f59938c2cc12bebc528045f6ec596cf44020db59f5

data/fluent-plugin-dynamodb-alt.gemspec

@@ -4,7 +4,7 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 
 Gem::Specification.new do |spec|
   spec.name          = 'fluent-plugin-dynamodb-alt'
-  spec.version       = '0.1.2'
+  spec.version       = '0.1.3'
   spec.authors       = ['Genki Sugawara']
   spec.email         = ['sgwr_dts@yahoo.co.jp']
   spec.summary       = %q{Fluent plugin to output to DynamoDB.}

data/lib/fluent/plugin/out_dynamodb_alt.rb

@@ -28,6 +28,7 @@ class Fluent::DynamodbAltOutput < Fluent::BufferedOutput
     super
     require 'aws-sdk-core'
     require 'parallel'
+    require 'set'
     require 'stringio'
   end
 
@@ -111,6 +112,7 @@ class Fluent::DynamodbAltOutput < Fluent::BufferedOutput
 
   def put_record(record)
     convert_binary!(record)
+    convert_set!(record)
 
     item = {
       :table_name => @table_name,
@@ -220,4 +222,14 @@ class Fluent::DynamodbAltOutput < Fluent::BufferedOutput
 
     return record
   end
+
+  def convert_set!(record)
+    record.each do |key, val|
+      if val.kind_of?(Array)
+        record[key] = Set.new(val)
+      end
+    end
+
+    return record
+  end
 end

data/spec/out_dynamodb_alt_spec.rb

@@ -479,5 +479,26 @@ describe Fluent::DynamodbAltOutput do
         ]
       end
     }
+
+    context('set') {
+      it do
+        run_driver do |d|
+          d.emit({'id' => '12345678-1234-1234-1234-123456789001', 'timestamp' => 1409534625001, 'set' => [1, 2, 3]}, time)
+          d.emit({'id' => '12345678-1234-1234-1234-123456789002', 'timestamp' => 1409534625002, 'set' => ['1', '2', '3']}, time)
+          d.emit({'id' => '12345678-1234-1234-1234-123456789003', 'timestamp' => 1409534625003, 'set' => ['A', 'B', 'C']}, time)
+        end
+
+        rows = select_all.map do |row|
+          row['set'].sort!
+          row
+        end
+
+        expect(rows).to match_array [
+          {"id"=>"12345678-1234-1234-1234-123456789001", "timestamp"=>1409534625001, "set"=>[1, 2, 3]},
+          {"id"=>"12345678-1234-1234-1234-123456789002", "timestamp"=>1409534625002, "set"=>['1', '2', '3']},
+          {"id"=>"12345678-1234-1234-1234-123456789003", "timestamp"=>1409534625003, "set"=>['A', 'B', 'C']},
+        ]
+      end
+    }
   }
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: fluent-plugin-dynamodb-alt
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Genki Sugawara
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-09-04 00:00:00.000000000 Z
+date: 2014-09-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: fluentd