Sha256: 431ebd052ed7ab12d48609ea912b379355729aff05dd12c35fa7605915f366c4
Contents?: true
Size: 1.87 KB
Versions: 105
Compression:
Stored size: 1.87 KB
Contents
require "active_support/core_ext/string"
class Dynomite::Migration
# jets dynamodb:generate posts --partition-key id:string
class Generator
include Dynomite::DbConfig
attr_reader :migration_name, :table_name
def initialize(migration_name, options)
@migration_name = migration_name
@options = options
end
def generate
puts "Generating migration for #{@table_name}" unless @options[:quiet]
return if @options[:noop]
create_migration
end
def create_migration
FileUtils.mkdir_p(File.dirname(migration_path))
IO.write(migration_path, migration_code)
puts "Migration file created: #{migration_path}. \nTo run:"
puts " jets dynamodb:migrate #{migration_path}"
end
def migration_code
path = File.expand_path("../templates/#{table_action}.rb", __FILE__)
result = Dynomite::Erb.result(path,
migration_class_name: migration_class_name,
table_name: table_name,
partition_key: @options[:partition_key],
sort_key: @options[:sort_key],
provisioned_throughput: @options[:provisioned_throughput] || 5,
)
end
def table_action
@options[:table_action] || conventional_table_action
end
def conventional_table_action
@migration_name.include?("update") ? "update_table" : "create_table"
end
def table_name
@options[:table_name] || conventional_table_name
end
# create_posts => posts
# update_posts => posts
def conventional_table_name
@migration_name.sub(/^(create|update)_/, '')
end
def migration_class_name
"#{@migration_name}_migration".classify # doesnt include timestamp
end
def migration_path
"#{Dynomite.app_root}dynamodb/migrate/#{timestamp}-#{@migration_name}_migration.rb"
end
def timestamp
@timestamp ||= Time.now.strftime("%Y%m%d%H%M%S")
end
end
end
Version data entries
105 entries across 105 versions & 2 rubygems