dynamo-record 0.2.0 → 0.3.0

data/lib/dynamo-record/task_helpers/cleanup.rb

@@ -1,19 +0,0 @@
-module DynamoRecord
-  module TaskHelpers
-    class Cleanup
-      def self.run
-        raise 'Task not available on production' if Rails.env.production?
-        Dir[Rails.root.join('app/models/*.rb').to_s].each do |filename|
-          delete_by_class(filename)
-        end
-      end
-
-      def self.delete_by_class(filename)
-        klass = File.basename(filename, '.rb').camelize.constantize
-        return unless klass.included_modules.include? DynamoRecord::Model
-        Rails.logger.info "Deleting all items in table: #{klass}"
-        klass.scan.each(&:delete!)
-      end
-    end
-  end
-end

data/lib/dynamo-record/task_helpers/drop_all_tables.rb

@@ -1,17 +0,0 @@
-module DynamoRecord
-  module TaskHelpers
-    class DropAllTables
-      def self.run(override = false)
-        raise 'Task not available on production' if Rails.env.production?
-        env = Rails.env
-        dynamodb = Aws::DynamoDB::Client.new
-        tables = dynamodb.list_tables
-        tables.table_names.map do |t|
-          next unless t.include?(env) || override
-          dt = dynamodb.delete_table(table_name: t)
-          dt ? "Deleted: #{t}" : "Delete failed: #{t}"
-        end
-      end
-    end
-  end
-end

data/lib/dynamo-record/task_helpers/drop_table.rb

@@ -1,11 +0,0 @@
-module DynamoRecord
-  module TaskHelpers
-    class DropTable
-      def self.run(table_name)
-        dynamodb = Aws::DynamoDB::Client.new
-        resp = dynamodb.delete_table(table_name: table_name)
-        "Deleted: #{resp.table_description.table_name}"
-      end
-    end
-  end
-end

data/lib/dynamo-record/task_helpers/list_tables.rb

@@ -1,13 +0,0 @@
-module DynamoRecord
-  module TaskHelpers
-    class ListTables
-      def self.run
-        dynamodb = Aws::DynamoDB::Client.new
-        tables = dynamodb.list_tables
-        tables.table_names.select do |tn|
-          tn.starts_with?(Rails.configuration.dynamo['prefix'])
-        end
-      end
-    end
-  end
-end

data/lib/dynamo-record/task_helpers/migration_runner.rb

@@ -1,70 +0,0 @@
-module DynamoRecord
-  module TaskHelpers
-    class MigrationRunner
-      def self.run(path = 'db/dynamo_migrate')
-        constants = []
-        filename_regexp = /\A([0-9]+)_([_a-z0-9]*)\.?([_a-z0-9]*)?\.rb\z/
-
-        # Sorts the files located in `db/dynamo_migrate` to ensure order is preserved
-        Dir[Rails.root.join("#{path}/*.rb")].sort.each do |f|
-          migration = migration(f, filename_regexp, constants)
-
-          # starts the migration
-          yield "Migrating: #{migration}"
-
-          begin
-            status = table_config_check(migration)
-            yield status if status
-
-            status = up(migration)
-            yield status if status
-
-            status = update(migration)
-            yield status if status
-          rescue => e
-            yield "Migration failed: #{e}"
-          end
-        end
-      end
-
-      def self.migration(f, filename_regexp, constants)
-        raise "Non-numeric prefix: #{f}" if File.basename(f).scan(filename_regexp).first.nil?
-        require f
-
-        # finds the constant that was added on the require statement above
-        migration_sym = (DynamoMigrate.constants - constants).first
-        migration = DynamoMigrate.const_get(migration_sym)
-        constants.push migration_sym
-        migration
-      end
-
-      def self.status_message(status)
-        case status
-        when :exists
-          'Table already exists'
-        when :migrated
-          'Migration successful'
-        else
-          raise 'Migration failed'
-        end
-      end
-
-      def self.up(migration)
-        return unless migration.respond_to? :up
-        status_message migration.up
-      end
-
-      def self.table_config_check(migration)
-        return unless migration.respond_to? :table_config
-        status_message migration.table_config_check
-      end
-
-      def self.update(migration)
-        return unless migration.respond_to? :update
-        status = migration.update
-        return 'Migration successful' if status == :updated
-        status
-      end
-    end
-  end
-end

data/lib/dynamo-record/task_helpers/scale.rb

@@ -1,86 +0,0 @@
-module DynamoRecord
-  module TaskHelpers
-    class Scale
-      attr_reader :model, :attribute_selector, :new_throughput
-      attr_reader :migration, :existing_throughput, :model_name
-
-      def initialize(model_name, attribute_selector, new_throughput)
-        @model_name = model_name
-        @attribute_selector = attribute_selector
-        @new_throughput = new_throughput
-      end
-
-      def run
-        return description if [model_name, attribute_selector, new_throughput].any?(&:nil?)
-
-        @model = model_name.constantize
-        @migration = Aws::Record::TableMigration.new(model)
-        @existing_throughput = model.provisioned_throughput
-
-        update_throughput
-        success_message
-      end
-
-      private
-
-      def success_message
-        "Successfully updated #{model.table_name} throughput to #{update_instructions[:provisioned_throughput]}"
-      end
-
-      def update_throughput
-        raise_attribute_error if !read? && !write?
-
-        migration.update!(update_instructions)
-      end
-
-      def update_instructions
-        {
-          provisioned_throughput: {
-            write_capacity_units: (write? && new_throughput) || existing_write,
-            read_capacity_units: (read? && new_throughput) || existing_read
-          }
-        }
-      end
-
-      def existing_write
-        existing_throughput[:write_capacity_units]
-      end
-
-      def existing_read
-        existing_throughput[:read_capacity_units]
-      end
-
-      def both?
-        attribute_selector.to_sym == :both
-      end
-
-      def read?
-        both? || attribute_selector.to_sym == :read
-      end
-
-      def write?
-        both? || attribute_selector.to_sym == :write
-      end
-
-      def raise_attribute_error
-        raise ArgumentError, 'You didn\'t provide an appropriate attribute selection. We accept [:both, :read, :write]'
-      end
-
-      def description
-        <<~DESCRIPTION
-        --------------------------------------------------------------------------------
-          Here's some usage information:
-            Scale a dynamo table.  Requires three inputs.
-              - ModelName
-                ruby class of the model
-              - attribute
-                valid values include "both", "read" and "write"
-              - new_throughput
-                numerical value for the new read/write capacity units
-            Example: `rake dynamo:scale[MySuperDynamoModel,both,50]`
-          --------------------------------------------------------------------------------
-        DESCRIPTION
-      end
-    end
-  end
-end

data/lib/model_existence_validator.rb

@@ -1,7 +0,0 @@
-require 'active_model'
-class ModelExistenceValidator < ActiveModel::EachValidator
-  def validate_each(record, attribute, value)
-    return if options[:model].exists? value
-    record.errors[attribute] << (options[:message] || "#{attribute}:#{value} is not a valid #{options[:model]}")
-  end
-end