cequel-migrations-rails 0.0.1 → 0.1.0

data/README.md

@@ -63,6 +63,21 @@ column family in it called `schema_migrations` run the following command:
 rake cequel:create
 ```
 
+#### Strategy & Strategy Options
+
+Before `cequel:create` will work the `strategy_class` and any
+`strategy_options` need to be set per environment in the `config/cequel.yml`
+config as seen in the example below.
+
+```
+development:
+  host: '127.0.0.1:9160'
+  keyspace: capture_api_dev
+  strategy_class: SimpleStrategy
+  strategy_options:
+    replication_factor: 1
+```
+
 ### Drop Database
 
 If you have already created the database and you want to drop it to start from

data/lib/cequel-migrations-rails/keyspace_manager.rb

@@ -0,0 +1,45 @@
+module Cequel
+  module Migrations
+    module Rails
+      class KeyspaceManager
+        attr_reader :db
+
+        def initialize
+          @db = CassandraCQL::Database.new(self.class.cequel_env_conf['host'])
+        end
+
+        def self.cequel_env_conf
+          YAML::load(File.open(File.join(::Rails.root,"config", "cequel.yml")))[::Rails.env]
+        end
+
+        def create_keyspace
+          db.execute(build_create_keyspace_cmd(self.class.cequel_env_conf['keyspace'], self.class.cequel_env_conf['strategy_class'], self.class.cequel_env_conf['strategy_options']))
+        end
+
+        def use_keyspace
+          db.execute("USE #{self.class.cequel_env_conf['keyspace']}")
+        end
+
+        def drop_keyspace
+          db.execute("DROP KEYSPACE #{self.class.cequel_env_conf['keyspace']}")
+        end
+
+        private
+
+        def build_create_keyspace_cmd(keyspace_name, strategy_class_name, strategy_options)
+          strat_opts_array = []
+          if strategy_options
+            strategy_options.each_pair do |k,v|
+              strat_opts_array << "strategy_options:#{k.to_s} = #{v}"
+            end
+          end
+          cql_cmd = ["CREATE KEYSPACE #{keyspace_name} WITH strategy_class = '#{strategy_class_name}'"]
+          if !strat_opts_array.empty?
+            cql_cmd << strat_opts_array.join(" AND ")
+          end
+          cql_cmd.join(" AND ")
+        end
+      end
+    end
+  end
+end

data/lib/cequel-migrations-rails/tasks/migrations.rake

@@ -4,37 +4,25 @@ require 'shearwater/cassandra_cql_backend'
 namespace :cequel do
   desc "Create the cequel specified cassandra keystore for the current environment"
   task :create => :environment do
-    # Read in the cequel config for the current Rails environment
-    cequel_env_conf = YAML::load(File.open(File.join(::Rails.root,"config", "cequel.yml")))[Rails.env]
-
-    # Create a CQL connection to use as the migrations backend.
-    db = CassandraCQL::Database.new(cequel_env_conf['host'])
-    db.execute("CREATE KEYSPACE #{cequel_env_conf['keyspace']} WITH strategy_class = 'SimpleStrategy' AND strategy_options:replication_factor = 1")
-    db.execute("USE #{cequel_env_conf['keyspace']}")
-    db.execute("CREATE COLUMNFAMILY schema_migrations (version bigint PRIMARY KEY, migrated_at timestamp)")
+    keyspace_manager = Cequel::Migrations::Rails::KeyspaceManager.new
+    keyspace_manager.create_keyspace
+    keyspace_manager.use_keyspace
+    keyspace_manager.db.execute("CREATE COLUMNFAMILY schema_migrations (version bigint PRIMARY KEY, migrated_at timestamp)")
   end
 
   desc "Drop the cequel specified cassandra keystore for the current environment"
   task :drop => :environment do
-    # Read in the cequel config for the current Rails environment
-    cequel_env_conf = YAML::load(File.open(File.join(::Rails.root,"config", "cequel.yml")))[Rails.env]
-
-    # Create a CQL connection to use as the migrations backend.
-    db = CassandraCQL::Database.new(cequel_env_conf['host'])
-    db.execute("DROP KEYSPACE #{cequel_env_conf['keyspace']}")
+    keyspace_manager = Cequel::Migrations::Rails::KeyspaceManager.new
+    keyspace_manager.drop_keyspace
   end
 
   desc "Migrate the cassandra store"
   task :migrate => :environment do
-    # Read in the cequel config for the current Rails environment
-    cequel_env_conf = YAML::load(File.open(File.join(::Rails.root,"config", "cequel.yml")))[Rails.env]
-
-    # Create a CQL connection to use as the migrations backend.
-    db = CassandraCQL::Database.new(cequel_env_conf['host'])
-    db.execute("USE #{cequel_env_conf['keyspace']}")
+    keyspace_manager = Cequel::Migrations::Rails::KeyspaceManager.new
+    keyspace_manager.use_keyspace
 
     # Create the migrator
-    backend = Shearwater::CassandraCqlBackend.new(db)
+    backend = Shearwater::CassandraCqlBackend.new(keyspace_manager.db)
     migrations_directory = ::Rails.root.join('cequel', 'migrate')
     migrator = Shearwater::Migrator.new(migrations_directory, backend)
 
@@ -44,15 +32,11 @@ namespace :cequel do
 
   desc "Rollback to the previous migration version by 1 step"
   task :rollback => :environment do
-    # Read in the cequel config for the current Rails environment
-    cequel_env_conf = YAML::load(File.open(File.join(::Rails.root,"config", "cequel.yml")))[Rails.env]
-
-    # Create a CQL connection to use as the migrations backend.
-    db = CassandraCQL::Database.new(cequel_env_conf['host'])
-    db.execute("USE #{cequel_env_conf['keyspace']}")
+    keyspace_manager = Cequel::Migrations::Rails::KeyspaceManager.new
+    keyspace_manager.use_keyspace
 
     # Create the migrator
-    backend = Shearwater::CassandraCqlBackend.new(db)
+    backend = Shearwater::CassandraCqlBackend.new(keyspace_manager.db)
     migrations_directory = ::Rails.root.join('cequel', 'migrate')
     migrator = Shearwater::Migrator.new(migrations_directory, backend)
 
@@ -68,5 +52,4 @@ namespace :cequel do
     Rake::Task["cequel:create"].invoke
     Rake::Task["cequel:migrate"].invoke
   end
-  
 end

data/lib/cequel-migrations-rails/version.rb

@@ -1,7 +1,7 @@
 module Cequel
   module Migrations
     module Rails
-      VERSION = "0.0.1"
+      VERSION = "0.1.0"
     end
   end
 end

data/lib/cequel-migrations-rails.rb

@@ -1,6 +1,7 @@
 require "cassandra-cql"
 
 require "cequel-migrations-rails/version"
+require "cequel-migrations-rails/keyspace_manager"
 require "cequel/migration"
 
 module Cequel

data/spec/lib/cequel/migration_spec.rb

@@ -1,6 +1,7 @@
 require 'spec_helper'
 
 describe Cequel::Migration do
+
   let(:migration_class) { class FooMigration < Cequel::Migration; end; FooMigration }
   let(:migration) { migration_class.new }
 

data/spec/lib/cequel-migrations-rails/cql_manager_spec.rb

@@ -0,0 +1,62 @@
+require 'spec_helper'
+
+# - creation of keyspace
+# - drop keyspace
+# - migrate
+# - rollback
+#
+
+describe Cequel::Migrations::Rails::KeyspaceManager do
+  describe "#new" do
+    it "creates a CQL connection" do
+      Cequel::Migrations::Rails::KeyspaceManager.stub(:cequel_env_conf).and_return({ 'host' => 'some host' })
+      CassandraCQL::Database.should_receive(:new).with('some host')
+      Cequel::Migrations::Rails::KeyspaceManager.new
+    end
+  end
+
+  describe "#create_keyspace" do
+    it "send cql to create the specified keyspace" do
+      Cequel::Migrations::Rails::KeyspaceManager.stub(:cequel_env_conf).and_return({ 'host' => 'some host', 'keyspace' => 'aoeuoae', 'strategy_class' => 'stratclass' })
+      db = mock('db')
+      CassandraCQL::Database.stub(:new).and_return(db)
+      subject.stub(:build_create_keyspace_cmd).and_return('xxx')
+      db.should_receive(:execute).with('xxx')
+      subject.create_keyspace
+    end
+  end
+
+  describe "#build_create_keyspace_cmd" do
+    it "build the cql command string to create a keyspace" do
+      Cequel::Migrations::Rails::KeyspaceManager.stub(:cequel_env_conf).and_return({ 'host' => 'some host' })
+      CassandraCQL::Database.stub(:new)
+      subject.send(:build_create_keyspace_cmd, 'somename_keyspace_name', 'StrategyClass', { :replication_factor => 1 }).should == "CREATE KEYSPACE somename_keyspace_name WITH strategy_class = 'StrategyClass' AND strategy_options:replication_factor = 1"
+    end
+
+    it "builds the cql command string to create a keyspace without strategy_options" do
+      Cequel::Migrations::Rails::KeyspaceManager.stub(:cequel_env_conf).and_return({ 'host' => 'some host' })
+      CassandraCQL::Database.stub(:new)
+      subject.send(:build_create_keyspace_cmd, 'somename_keyspace_name', 'StrategyClass', nil).should == "CREATE KEYSPACE somename_keyspace_name WITH strategy_class = 'StrategyClass'"
+    end
+  end
+
+  describe "#use_keyspace" do
+    it "send the cql command to use the keyspace from the cequel.yml" do
+      Cequel::Migrations::Rails::KeyspaceManager.stub(:cequel_env_conf).and_return({ 'host' => 'some host', 'keyspace' => 'my_keyspace' })
+      db = mock('db')
+      CassandraCQL::Database.stub(:new).and_return(db)
+      db.should_receive(:execute).with('USE my_keyspace')
+      subject.use_keyspace
+    end
+  end
+
+  describe "#drop_keyspace" do
+    it "sends the cql comand to drop the keyspace from the cequel.yml" do
+      Cequel::Migrations::Rails::KeyspaceManager.stub(:cequel_env_conf).and_return({ 'host' => 'some host', 'keyspace' => 'my_keyspace' })
+      db = mock('db')
+      CassandraCQL::Database.stub(:new).and_return(db)
+      db.should_receive(:execute).with('DROP KEYSPACE my_keyspace')
+      subject.drop_keyspace
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cequel-migrations-rails
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-12-03 00:00:00.000000000 Z
+date: 2012-12-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: shearwater
@@ -90,6 +90,7 @@ files:
 - Rakefile
 - cequel-migrations-rails.gemspec
 - lib/cequel-migrations-rails.rb
+- lib/cequel-migrations-rails/keyspace_manager.rb
 - lib/cequel-migrations-rails/tasks/migrations.rake
 - lib/cequel-migrations-rails/version.rb
 - lib/cequel/migration.rb
@@ -97,6 +98,7 @@ files:
 - lib/generators/cequel/migration/templates/migration.rb
 - lib/generators/cequel/migrations/install/install_generator.rb
 - lib/generators/cequel/migrations/install/templates/cequel/migrate/.gitkeep
+- spec/lib/cequel-migrations-rails/cql_manager_spec.rb
 - spec/lib/cequel/migration_spec.rb
 - spec/spec_helper.rb
 homepage: http://github.com/cyphactor/cequel-migrations-rails
@@ -124,5 +126,6 @@ signing_key:
 specification_version: 3
 summary: A Rails plugin that provides migration support for Cequel.
 test_files:
+- spec/lib/cequel-migrations-rails/cql_manager_spec.rb
 - spec/lib/cequel/migration_spec.rb
 - spec/spec_helper.rb