shearwater 0.0.0 → 0.1.3

data/lib/shearwater/cassandra_cql_backend.rb

@@ -0,0 +1,44 @@
+module Shearwater
+
+  class CassandraCqlBackend
+
+    def initialize(connection, column_family = 'schema_migrations')
+      @connection, @column_family = connection, column_family
+    end
+
+    def migrated!(id)
+      execute(
+        "INSERT INTO #{@column_family} (version, migrated_at) VALUES (?, ?)",
+        id, Time.now
+      )
+    end
+
+    def rolled_back!(id)
+      execute("DELETE FROM #{@column_family} WHERE version = ?", id)
+    end
+
+    def migrated?(id)
+      !!execute(
+        "SELECT migrated_at FROM #{@column_family} WHERE version = ?", id
+      ).fetch_row.to_hash['migrated_at']
+    end
+
+    def last_migration
+      rows = []
+      execute(
+        "SELECT version, migrated_at FROM #{@column_family}"
+      ).fetch { |row| rows << row.to_hash }
+      row = rows.reject { |row| row['migrated_at'].nil? }.sort { |row1, row2| row1['version'].to_i <=> row2['version'].to_i }.last
+      row['version'].to_i if row
+    end
+
+    private
+
+    def execute(*args)
+      @connection.execute(*args)
+    end
+
+  end
+
+end
+

data/lib/shearwater/in_memory_backend.rb

@@ -0,0 +1,25 @@
+require 'set'
+
+module Shearwater
+  class InMemoryBackend
+    def initialize
+      @migrations = SortedSet[]
+    end
+
+    def migrated!(id)
+      @migrations << id
+    end
+
+    def rolled_back!(id)
+      @migrations.delete(id)
+    end
+
+    def migrated?(id)
+      @migrations.include?(id)
+    end
+
+    def last_migration
+      @migrations.to_a.last
+    end
+  end
+end

data/lib/shearwater/migrator.rb

@@ -0,0 +1,55 @@
+module Shearwater
+
+  class Migrator
+
+    def initialize(migrations_dir, backend, options = {})
+      @migrations_dir, @backend = migrations_dir, backend
+      @verbose = !!options[:verbose]
+    end
+
+    def migrate
+      migrations.keys.sort.each do |id|
+        unless @backend.migrated?(id)
+          migration = migrations[id]
+          say "Migrating #{migration.class.name}"
+          migration.up
+          @backend.migrated!(id)
+        end
+      end
+    end
+
+    def rollback(step = 1)
+      step.times do
+        id = @backend.last_migration
+        migration = migrations[id]
+        say "Rolling back #{migration.class.name}"
+        migration.down
+        @backend.rolled_back!(id)
+      end
+    end
+
+    private
+
+    def migrations
+      @migrations ||= {}.tap do |migrations|
+        Dir.glob(File.join(@migrations_dir, '**', '*.rb')).each do |file|
+          if /(\d+)_(\w+)\.rb/ =~ file
+            id, name = $1.to_i, $2
+            load file
+            class_name = name.split('_').map { |s| s.capitalize }.join
+            migration = ::Object.const_get(class_name).new
+            migrations[id] = migration
+          end
+        end
+      end
+    end
+
+    private
+
+    def say(message)
+      puts message if @verbose
+    end
+
+  end
+
+end

data/lib/shearwater/redis_backend.rb

@@ -0,0 +1,28 @@
+module Shearwater
+
+  class RedisBackend
+
+    def initialize(redis, key)
+      @redis, @key = redis, key
+    end
+
+    def migrated!(id)
+      @redis.zadd(@key, id, id.to_s)
+    end
+
+    def rolled_back!(id)
+      @redis.zrem(@key, id.to_s)
+    end
+
+    def migrated?(id)
+      @redis.zcount(@key, id, id) > 0
+    end
+
+    def last_migration
+      id = @redis.zrange(@key, -1, -1).first
+      id.to_i if id
+    end
+
+  end
+
+end

data/lib/shearwater/version.rb

@@ -0,0 +1,5 @@
+module Shearwater
+
+  VERSION = '0.1.3'
+
+end

data/lib/shearwater.rb

@@ -0,0 +1,5 @@
+require 'shearwater/in_memory_backend'
+require 'shearwater/migrator'
+
+module Shearwater
+end

data/spec/environment.rb

@@ -0,0 +1,7 @@
+require 'bundler'
+
+Bundler.require(:default, :test)
+
+Dir.glob(File.join(File.dirname(__FILE__), 'support', '**', '*.rb')).each do |f|
+  require f
+end

data/spec/examples/cassandra_cql_backend_spec.rb

@@ -0,0 +1,29 @@
+require File.expand_path('../spec_helper', __FILE__)
+require 'shearwater/cassandra_cql_backend'
+
+class CassandraStubResults
+  def initialize(row_hashes)
+    @row_hashes = row_hashes
+  end
+
+  def fetch
+    @row_hashes.each do |row_hash|
+      yield row_hash
+    end
+  end
+end
+
+describe Shearwater::CassandraCqlBackend do
+  describe "#last_migration" do
+    it "ignores 'range ghosts' when identifying the last migration" do
+      connection = stub('connection')
+      stubbed_cassandra_results = CassandraStubResults.new([
+        { 'version' => 1, 'migrated_at' => 2342342 },
+        { 'version' => 2, 'migrated_at' => nil }
+      ])
+      connection.stub_chain(:execute).and_return(stubbed_cassandra_results)
+      backend = Shearwater::CassandraCqlBackend.new(connection)
+      backend.last_migration.should == 1
+    end
+  end
+end

data/spec/examples/migrator_spec.rb

@@ -0,0 +1,58 @@
+require File.expand_path('../spec_helper', __FILE__)
+
+describe Shearwater::Migrator do
+  let(:backend) { Shearwater::InMemoryBackend.new }
+  let(:tracker) { Shearwater::SpecSupport::MigrationTracker }
+
+  let :migrator do
+    Shearwater::Migrator.new(File.expand_path('../../migrations', __FILE__), backend)
+  end
+
+  describe '#migrate' do
+    it 'should run all migrations from fresh' do
+      migrator.migrate
+      [1, 2, 3].each { |i| tracker.should have_migrated(i) }
+    end
+
+    it 'should run migrations in order of id' do
+      migrator.migrate
+      tracker.events.map(&:id).should == [1, 2, 3]
+    end
+
+    it 'should not run migrations that backend reports have already been run' do
+      backend.migrated!(1)
+      migrator.migrate
+      tracker.events.map(&:id).should == [2, 3]
+    end
+
+    it 'should report run to backend' do
+      migrator.migrate
+      1.upto(3) { |i| backend.migrated?(i).should be_true }
+    end
+  end
+
+  describe '#rollback' do
+    before do
+      1.upto(3) { |i| backend.migrated!(i) }
+    end
+
+    it 'should roll back most recent change' do
+      migrator.rollback
+      tracker.should have_rolled_back(3)
+      tracker.should_not have_rolled_back(2)
+      tracker.should_not have_rolled_back(1)
+    end
+
+    it 'should mark migration as rolled back in backend' do
+      migrator.rollback
+      backend.migrated?(3).should be_false
+    end
+
+    it 'should run multiple migrations if passed' do
+      migrator.rollback(2)
+      tracker.should have_rolled_back(3)
+      tracker.should have_rolled_back(2)
+      tracker.should_not have_rolled_back(1)
+    end
+  end
+end

data/spec/examples/spec_helper.rb

@@ -0,0 +1,7 @@
+require File.expand_path('../../environment', __FILE__)
+
+RSpec.configure do |config|
+  config.before :each do
+    Shearwater::SpecSupport::MigrationTracker.clear!
+  end
+end

data/spec/migrations/001_migration_1.rb

@@ -0,0 +1,9 @@
+class Migration1
+  def up
+    Shearwater::SpecSupport::MigrationTracker.migrated!(1)
+  end
+
+  def down
+    Shearwater::SpecSupport::MigrationTracker.rolled_back!(1)
+  end
+end

data/spec/migrations/002_migration_2.rb

@@ -0,0 +1,10 @@
+class Migration2
+  def up
+    Shearwater::SpecSupport::MigrationTracker.migrated!(2)
+  end
+
+  def down
+    Shearwater::SpecSupport::MigrationTracker.rolled_back!(2)
+  end
+end
+

data/spec/migrations/003_migration_3.rb

@@ -0,0 +1,9 @@
+class Migration3
+  def up
+    Shearwater::SpecSupport::MigrationTracker.migrated!(3)
+  end
+
+  def down
+    Shearwater::SpecSupport::MigrationTracker.rolled_back!(3)
+  end
+end

data/spec/support/migration_tracker.rb

@@ -0,0 +1,43 @@
+module Shearwater
+
+  module SpecSupport
+
+    module MigrationTracker
+
+      extend self
+
+      Event = Struct.new(:id, :action)
+
+      def migrated!(id)
+        events << Event.new(id, :migrate)
+      end
+
+      def rolled_back!(id)
+        events << Event.new(id, :rollback)
+      end
+
+      def has_migrated?(id)
+        event?(id, :migrate)
+      end
+
+      def has_rolled_back?(id)
+        event?(id, :rollback)
+      end
+
+      def clear!
+        events.clear
+      end
+
+      def events
+        @events ||= []
+      end
+
+      def event?(id, action)
+        events.any? { |e| e.id == id && e.action == action }
+      end
+
+    end
+
+  end
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: shearwater
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 0.1.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-04-13 00:00:00.000000000Z
+date: 2012-11-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &10110340 !ruby/object:Gem::Requirement
+  requirement: !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -21,21 +21,31 @@ dependencies:
         version: '2.0'
   type: :development
   prerelease: false
-  version_requirements: *10110340
-- !ruby/object:Gem::Dependency
-  name: ruby-debug19
-  requirement: &10109880 !ruby/object:Gem::Requirement
+  version_requirements: !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
       - !ruby/object:Gem::Version
-        version: '0.11'
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: debugger
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :development
   prerelease: false
-  version_requirements: *10109880
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: yard
-  requirement: &10109420 !ruby/object:Gem::Requirement
+  requirement: !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -43,7 +53,12 @@ dependencies:
         version: '0.6'
   type: :development
   prerelease: false
-  version_requirements: *10109420
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.6'
 description: ! 'Shearwater is a tiny framework for managing migrations in an everything-agnostic
 
   way. It provides a pluggable backend architecture for storing which migrations
@@ -57,7 +72,21 @@ email: mat.a.brown@gmail.com
 executables: []
 extensions: []
 extra_rdoc_files: []
-files: []
+files:
+- lib/shearwater.rb
+- lib/shearwater/cassandra_cql_backend.rb
+- lib/shearwater/redis_backend.rb
+- lib/shearwater/in_memory_backend.rb
+- lib/shearwater/migrator.rb
+- lib/shearwater/version.rb
+- spec/environment.rb
+- spec/examples/spec_helper.rb
+- spec/examples/migrator_spec.rb
+- spec/examples/cassandra_cql_backend_spec.rb
+- spec/support/migration_tracker.rb
+- spec/migrations/002_migration_2.rb
+- spec/migrations/003_migration_3.rb
+- spec/migrations/001_migration_1.rb
 homepage: 
 licenses:
 - MIT
@@ -79,9 +108,12 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.15
+rubygems_version: 1.8.24
 signing_key: 
 specification_version: 3
 summary: Tiny everything-agnostic migrations framework
-test_files: []
+test_files:
+- spec/examples/spec_helper.rb
+- spec/examples/migrator_spec.rb
+- spec/examples/cassandra_cql_backend_spec.rb
 has_rdoc: false