lookup_table.rb in alephant-lookup-2.1.0

- old
+ new

@@ -1,6 +1,6 @@
-require "aws-sdk"
+require "aws-sdk-dynamodb"
 require "thread"
 require "timeout"
 
 require "alephant/support/dynamodb/table"
 require "alephant/logger"
@@ -10,32 +10,28 @@
     class LookupTable < ::Alephant::Support::DynamoDB::Table
       include Logger
       attr_reader :table_name, :client
 
       def initialize(table_name)
+        options = {}
+        options.merge!({endpoint: ENV['AWS_DYNAMO_DB_ENDPOINT']}) if ENV['AWS_DYNAMO_DB_ENDPOINT']
         @mutex      = Mutex.new
-        @client     = AWS::DynamoDB::Client::V20120810.new
+        @client     = Aws::DynamoDB::Client.new(options)
         @table_name = table_name
         logger.info(
           "event"     => "LookupTableInitialized",
           "tableName" => table_name,
           "method"    => "#{self.class}#initialize"
         )
       end
 
       def write(component_key, version, location)
         client.put_item({
-          :table_name => table_name,
-          :item => {
-            'component_key' => {
-              'S' => component_key.to_s
-            },
-            'batch_version' => {
-              'N' => version.to_s
-            },
-            'location' => {
-              'S' => location.to_s
-            }
+          table_name: table_name,
+          item: {
+            'component_key' => component_key.to_s,
+            'batch_version' => version,
+            'location'      => location.to_s
           }
         }).tap do
           logger.info(
             "event"        => "LookupLocationWritten",
             "componentKey" => component_key,