aws_pipes 0.0.4 → 0.0.5

data/bin/aws_db

@@ -31,13 +31,25 @@ db = AWS::DynamoDB.new(
   :secret_access_key => AwsPipes.secret_access_key(opts)
 )
 
-begin
-  $stdin.sync = true
-  table = db.tables[table_name]
-  table.load_schema
-  while row = $stdin.gets
+$stdin.sync = true
+table = db.tables[table_name]
+table.load_schema
+
+while row = $stdin.gets
+  begin
     table.items.create Hash[col_names.zip(row.split "\t")]
+  rescue AWS::DynamoDB::Errors::ProvisionedThroughputExceededException
+    new_write_capacity = table.write_capacity_units * 2
+    warn "Exceeded provisioned throughput for table #{table_name}."
+    warn "Increasing write capacity to #{new_write_capacity}."
+    begin
+      table.provision_throughput :write_capacity_units => new_write_capacity
+    rescue AWS::DynamoDB::Errors::ResourceInUseException
+      # throughput provisioning in progress, give it time
+      sleep 10
+    end
+    retry
+  rescue Interrupt
+    exit 0
   end
-rescue Interrupt
-  exit 0
 end

data/lib/aws_pipes/version.rb

@@ -1,3 +1,3 @@
 module AwsPipes
-  VERSION = "0.0.4"
+  VERSION = "0.0.5"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aws_pipes
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-12-27 00:00:00.000000000 Z
+date: 2012-12-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk