dynomite 1.2.7 → 2.0.0

data/lib/dynomite/migration/common.rb

@@ -1,86 +0,0 @@
-# Common methods to the *SecondaryIndex classes that handle gsi and lsi methods
-# as well a the Dsl class that handles create_table and update_table methods.
-class Dynomite::Migration::Dsl
-  module Common
-    # http://docs.aws.amazon.com/sdkforruby/api/Aws/DynamoDB/Types/KeySchemaElement.html
-    # partition_key is required
-    def partition_key(identifier)
-      @partition_key_identifier = identifier # for later use. useful for conventional_index_name
-      adjust_schema_and_attributes(identifier, "hash")
-    end
-
-    # sort_key is optional
-    def sort_key(identifier)
-      @sort_key_identifier = identifier # for later use. useful for conventional_index_name
-      adjust_schema_and_attributes(identifier, "range")
-    end
-
-    # Parameters:
-    #   identifier: "id:string" or "id"
-    #   key_type: "hash" or "range"
-    #
-    # Adjusts the parameters for create_table to add the
-    # partition_key and sort_key
-    def adjust_schema_and_attributes(identifier, key_type)
-      name, attribute_type = identifier.split(':')
-      attribute_type = "string" if attribute_type.nil?
-
-      partition_key = {
-        attribute_name: name,
-        key_type: key_type.upcase
-      }
-      @key_schema << partition_key
-
-      attribute_definition = {
-        attribute_name: name,
-        attribute_type: Dynomite::ATTRIBUTE_TYPES[attribute_type]
-      }
-      @attribute_definitions << attribute_definition
-    end
-
-    # t.provisioned_throughput(5) # both
-    # t.provisioned_throughput(:read, 5)
-    # t.provisioned_throughput(:write, 5)
-    # t.provisioned_throughput(:both, 5)
-    def provisioned_throughput(*params)
-      case params.size
-      when 0 # reader method
-        return @provisioned_throughput # early return
-      when 1
-        # @provisioned_throughput_set_called useful for update_table
-        # only provide a provisioned_throughput settings if explicitly called for update_table
-        @provisioned_throughput_set_called = true
-        arg = params[0]
-        if arg.is_a?(Hash)
-          # Case:
-          # provisioned_throughput(
-          #   read_capacity_units: 10,
-          #   write_capacity_units: 10
-          # )
-          @provisioned_throughput = arg # set directly
-          return # early return
-        else # assume parameter is an Integer
-          # Case: provisioned_throughput(10)
-          capacity_type = :both
-          capacity_units = arg
-        end
-      when 2
-        @provisioned_throughput_set_called = true
-        # Case: provisioned_throughput(:read, 5)
-        capacity_type, capacity_units = params
-      end
-
-      map = {
-        read: :read_capacity_units,
-        write: :write_capacity_units,
-      }
-
-      if capacity_type == :both
-        @provisioned_throughput[map[:read]] = capacity_units
-        @provisioned_throughput[map[:write]] = capacity_units
-      else
-        @provisioned_throughput[capacity_type] = capacity_units
-      end
-    end
-  end
-end

data/lib/dynomite/migration/dsl/base_secondary_index.rb

@@ -1,73 +0,0 @@
-# Base class for LocalSecondaryIndex and GlobalSecondaryIndex
-class Dynomite::Migration::Dsl
-  class BaseSecondaryIndex
-    include Common
-
-    attr_accessor :action, :key_schema, :attribute_definitions
-    attr_accessor :index_name
-    def initialize(action, index_name=nil, &block)
-      @action = action.to_sym
-      # for gsi action can be: :create, :update, :delete
-      # for lsi action is always: :create
-      @index_name = index_name
-      @block = block
-
-      # Dsl fills these atttributes in as methods are called within
-      # the block
-      @key_schema = []
-      @attribute_definitions = []
-      # default provisioned_throughput
-      @provisioned_throughput = {
-        read_capacity_units: 5,
-        write_capacity_units: 5
-      }
-    end
-
-    def index_name
-      @index_name || conventional_index_name
-    end
-
-    def conventional_index_name
-      # @partition_key_identifier and @sort_key_identifier are set as immediately
-      # when the partition_key and sort_key methods are called in the dsl block.
-      # Usually look like this:
-      #
-      #   @partition_key_identifier: post_id:string
-      #   @sort_key_identifier: updated_at:string
-      #
-      # We strip the :string portion in case it is provided
-      #
-      partition_key = @partition_key_identifier.split(':').first
-      sort_key = @sort_key_identifier.split(':').first if @sort_key_identifier
-      [partition_key, sort_key, "index"].compact.join('-')
-    end
-
-    def evaluate
-      return if @evaluated
-      @block.call(self) if @block
-      @evaluated = true
-    end
-
-    def params
-      evaluate # lazy evaluation: wait until as long as possible before evaluating code block
-
-      params = { index_name: index_name } # required for all actions
-
-      if @action == :create
-        params[:key_schema] = @key_schema # required for create action
-        # hardcode to ALL for now
-        params[:projection] = { # required
-          projection_type: "ALL", # accepts ALL, KEYS_ONLY, INCLUDE
-          # non_key_attributes: ["NonKeyAttributeName"],
-        }
-      end
-
-      if self.is_a?(GlobalSecondaryIndex) && [:create, :update].include?(@action)
-        # provisioned_throughput is required for gsi index definition
-        params[:provisioned_throughput] = @provisioned_throughput
-      end
-
-      params
-    end
-  end
-end

data/lib/dynomite/migration/dsl/global_secondary_index.rb

@@ -1,4 +0,0 @@
-class Dynomite::Migration::Dsl
-  class GlobalSecondaryIndex < BaseSecondaryIndex
-  end
-end

data/lib/dynomite/migration/dsl/local_secondary_index.rb

@@ -1,8 +0,0 @@
-class Dynomite::Migration::Dsl
-  class LocalSecondaryIndex < BaseSecondaryIndex
-    def initialize(index_name=nil, &block)
-      # Can only create local secondary index when creating a table
-      super(:create, index_name, &block)
-    end
-  end
-end

data/lib/dynomite/migration/executor.rb

@@ -1,38 +0,0 @@
-class Dynomite::Migration
-  class Executor
-    include Dynomite::DbConfig
-
-    # Examples:
-    #  Executor.new(:create_table, params) or
-    #  Executor.new(:update_table, params)
-    #
-    # The params are generated frmo the dsl.params
-    attr_accessor :table_name
-    def initialize(table_name, method_name, params)
-      @table_name = table_name
-      @method_name = method_name # create_table or update_table
-      @params = params
-    end
-
-    def run
-      begin
-        # Examples:
-        #   result = db.create_table(@params)
-        #   result = db.update_table(@params)
-
-        # Leaving this verbose output in here until this DSL is more hardened to help debug
-        unless ENV['DYNOMITE_ENV'] == 'test'
-          puts "Calling #{@method_name} with params:"
-          pp @params
-        end
-
-        return if ENV['DYNOMITE_DRY'] # dry run flag
-        result = db.send(@method_name, @params)
-
-        puts "DynamoDB Table: #{@table_name} Status: #{result.table_description.table_status}"
-      rescue Aws::DynamoDB::Errors::ServiceError => error
-        puts "Unable to #{@method_name.to_s.gsub('_',' ')}: #{error.message}".color(:red)
-      end
-    end
-  end
-end