active_column 0.0.2 → 0.1

data/lib/active_column/tasks/column_family.rb

@@ -0,0 +1,64 @@
+module ActiveColumn
+
+  module Tasks
+
+    class ColumnFamily
+
+      COMPARATOR_TYPES = { :time      => 'TimeUUIDType',
+                           :timestamp => 'TimeUUIDType',
+                           :long      => 'LongType',
+                           :string    => 'BytesType' }
+
+      def initialize
+        @cassandra = ActiveColumn.connection
+      end
+
+      def exists?(name)
+        @cassandra.schema.cf_defs.find { |cf_def| cf_def.name == name.to_s }
+      end
+
+      def create(name, options = {})
+        opts = { :name => name.to_s,
+                 :keyspace => @cassandra.keyspace,
+                 :comparator_type => 'TimeUUIDType' }.merge(options)
+
+        cf = Cassandra::ColumnFamily.new.with_fields(opts)
+        @cassandra.add_column_family(cf)
+      end
+
+      def drop(name)
+        @cassandra.drop_column_family(name.to_s)
+      end
+
+      def clear(name)
+        @cassandra.truncate!(name.to_s)
+      end
+
+      private
+
+      def post_process_options(options)
+        type = options[:comparator_type]
+        if type && COMPARATOR_TYPES.has_key?(type)
+          options[:comparator_type] = COMPARATOR_TYPES[type]
+        end
+        options
+      end
+
+    end
+
+  end
+
+end
+
+class Cassandra
+  class ColumnFamily
+    def with_fields(options)
+      struct_fields.collect { |f| f[1][:name] }.each do |f|
+        send("#{f}=", options[f.to_sym])
+      end
+      self
+    end
+  end
+end
+
+

data/lib/active_column/tasks/keyspace.rb

@@ -0,0 +1,59 @@
+module ActiveColumn
+
+  module Tasks
+
+    class Keyspace
+
+      def initialize
+        c = ActiveColumn.connection
+        @cassandra = Cassandra.new('system', c.servers, c.thrift_client_options)
+      end
+
+      def exists?(name)
+        @cassandra.keyspaces.include? name.to_s
+      end
+
+      def create(name, options = {})
+        opts = { :name => name.to_s,
+                 :strategy_class => 'org.apache.cassandra.locator.LocalStrategy',
+                 :replication_factor => 1,
+                 :cf_defs => [] }.merge(options)
+
+        ks = Cassandra::Keyspace.new.with_fields(opts)
+        @cassandra.add_keyspace ks
+      end
+
+      def drop(name)
+        @cassandra.drop_keyspace name.to_s
+      end
+
+      def set(name)
+        @cassandra.keyspace = name.to_s
+      end
+
+      def get
+        @cassandra.keyspace
+      end
+
+      def clear
+        return puts 'Cannot clear system keyspace' if @cassandra.keyspace == 'system'
+
+        @cassandra.clear_keyspace!
+      end
+
+    end
+
+  end
+
+end
+
+class Cassandra
+  class Keyspace
+    def with_fields(options)
+      struct_fields.collect { |f| f[1][:name] }.each do |f|
+        send("#{f}=", options[f.to_sym])
+      end
+      self
+    end
+  end
+end

data/lib/active_column/tasks/ks.rb

@@ -0,0 +1,76 @@
+require 'rake'
+
+namespace :ks do
+
+  desc 'Create the keyspace in config/cassandra.yml for the current environment'
+  task :create => :environment do
+    config = load_config[Rails.env || 'development']
+    ks = config['keyspace']
+    ActiveColumn::Tasks::Keyspace.new.create ks, config
+    puts "Created keyspace: #{ks}"
+  end
+
+  desc 'Create keyspaces in config/cassandra.yml for all environments'
+  task 'create:all' => :environment do
+    config = load_config
+    kss = []
+    config.keys.each do |env|
+      kss << config[env]['keyspace']
+      ActiveColumn::Tasks::Keyspace.new.create kss.last, config
+    end
+    puts "Created keyspaces: #{kss.join(', ')}"
+  end
+
+  desc 'Drop keyspace in config/cassandra.yml for the current environment'
+  task :drop => :environment do
+    config = load_config[Rails.env || 'development']
+    ks = config['keyspace']
+    ActiveColumn::Tasks::Keyspace.new.drop ks
+    puts "Dropped keyspace: #{ks}"
+  end
+
+  desc 'Drop keyspaces in config/cassandra.yml for all environments'
+  task 'drop:all' => :environment do
+    config = load_config
+    kss = []
+    config.keys.each do |env|
+      kss << config[env]['keyspace']
+      ActiveColumn::Tasks::Keyspace.new.drop kss.last
+    end
+    puts "Dropped keyspaces: #{kss.join(', ')}"
+  end
+
+  desc 'Migrate the keyspace (options: VERSION=x)'
+  task :migrate => :environment do
+    set_keyspace
+    version = ( ENV['VERSION'] ? ENV['VERSION'].to_i : nil )
+    ActiveColumn::Migrator.migrate ActiveColumn::Migrator.migrations_path, version
+  end
+
+  desc 'Rolls the schema back to the previous version (specify steps w/ STEP=n)'
+  task :rollback => :environment do
+    set_keyspace
+    step = ENV['STEP'] ? ENV['STEP'].to_i : 1
+    ActiveColumn::Migrator.rollback ActiveColumn::Migrator.migrations_path, step
+  end
+
+  desc 'Pushes the schema to the next version (specify steps w/ STEP=n)'
+  task :forward => :environment do
+    set_keyspace
+    step = ENV['STEP'] ? ENV['STEP'].to_i : 1
+    ActiveColumn::Migrator.forward ActiveColumn::Migrator.migrations_path, step
+  end
+
+  private
+
+  def load_config
+    YAML.load_file(Rails.root.join("config", "cassandra.yml"))
+  end
+
+  def set_keyspace
+    config = load_config[Rails.env || 'development']
+    ActiveColumn::Tasks::Keyspace.new.set config['keyspace']
+  end
+
+end
+

data/lib/active_column/version.rb

@@ -1,3 +1,3 @@
 module ActiveColumn
-  VERSION = "0.0.2"
+  VERSION = "0.1"
 end

data/spec/active_column/base_crud_spec.rb

@@ -104,4 +104,4 @@ describe ActiveColumn::Base do
 
   end
 
-end
+end

data/spec/active_column/base_finders_spec.rb

@@ -20,7 +20,7 @@ describe ActiveColumn::Base do
         it 'find all of the models' do
           @found.size.should == 1
           @found['user1'].size.should == 3
-          @found['user1'].collect { |t| t.attributes['message'] }.should == [ 'Now im full', 'Now im hungry', 'Going running' ]
+          @found['user1'].collect { |t| t.message }.should == [ 'Now im full', 'Now im hungry', 'Going running' ]
         end
       end
 
@@ -31,8 +31,8 @@ describe ActiveColumn::Base do
 
         it 'finds all of the models' do
           @found.size.should == 2
-          @found['user1'].collect { |t| t.attributes['message'] }.should == [ 'Now im full' ]
-          @found['user2'].collect { |t| t.attributes['message'] }.should == [ 'Watching TV' ]
+          @found['user1'].collect { |t| t.message }.should == [ 'Now im full' ]
+          @found['user2'].collect { |t| t.message }.should == [ 'Watching TV' ]
         end
       end
     end
@@ -50,9 +50,9 @@ describe ActiveColumn::Base do
 
         it 'finds all of the models' do
           @found.size.should == 6
-          @found['user1:friend1'].collect { |t| t.attributes['message'] }.should == [ 'Need to do laundry' ]
-          @found['user1:friend2'].collect { |t| t.attributes['message'] }.should == [ 'My leg itches' ]
-          @found['user1:all'].collect { |t| t.attributes['message'] }.should == [ 'My leg itches' ]
+          @found['user1:friend1'].collect { |t| t.message }.should == [ 'Need to do laundry' ]
+          @found['user1:friend2'].collect { |t| t.message }.should == [ 'My leg itches' ]
+          @found['user1:all'].collect { |t| t.message }.should == [ 'My leg itches' ]
           @found['user2:friend1'].should == []
           @found['user2:friend2'].should == []
           @found['user2:all'].should == []

data/spec/active_column/migrator_spec.rb

@@ -0,0 +1,150 @@
+require 'spec_helper'
+
+$migrator_spec_data = {}
+
+def get_migrations
+  $cassandra.get(:schema_migrations, 'all').map {|name, _value| name.to_i}
+end
+
+def get_data
+  $migrator_spec_data.keys.sort
+end
+
+migrations_path = File.expand_path("../../support/migrate/migrator_spec", __FILE__)
+
+describe ActiveColumn::Migrator do
+
+  after do
+    $cassandra.truncate!("schema_migrations")
+    $migrator_spec_data.clear
+  end
+
+  describe '.migrate' do
+    context 'given no previous migrations and some pending migrations and no target version' do
+      before do
+        ActiveColumn::Migrator.migrate(migrations_path)
+      end
+
+      it 'runs the migrations' do
+        assert { get_migrations == [1, 2, 3, 4] }
+        assert { get_data       == [1, 2, 3, 4] }
+      end
+    end
+
+    context 'given no previous migrations and some pending migrations and a target version' do
+      before do
+        ActiveColumn::Migrator.migrate(migrations_path, 2)
+      end
+
+      it 'runs the migrations' do
+        assert { get_migrations == [1, 2] }
+        assert { get_data       == [1, 2] }
+      end
+    end
+
+    context 'given no previous migrations and no pending migrations' do
+      before do
+        ActiveColumn::Migrator.migrate(File.expand_path("./fake", migrations_path))
+      end
+
+      it 'runs no migrations' do
+        assert { get_migrations == [] }
+        assert { get_data       == [] }
+      end
+    end
+
+    context 'given some previous migrations and no target version' do
+      before do
+        ActiveColumn::Migrator.migrate(migrations_path, 2)
+        ActiveColumn::Migrator.migrate(migrations_path)
+      end
+
+      it 'runs the migrations' do
+        assert { get_migrations == [1, 2, 3, 4]}
+        assert { get_data       == [1, 2, 3, 4]}
+      end
+    end
+
+    context 'given some previous migrations and a target version up' do
+
+      before do
+        ActiveColumn::Migrator.migrate(migrations_path, 2)
+        ActiveColumn::Migrator.migrate(migrations_path, 3)
+      end
+
+      it 'runs the migrations' do
+        assert { get_migrations == [1, 2, 3] }
+        assert { get_data       == [1, 2, 3] }
+      end
+    end
+
+    context 'given some previous migrations and a target version down' do
+      before do
+        ActiveColumn::Migrator.migrate(migrations_path, 2)
+        ActiveColumn::Migrator.migrate(migrations_path, 1)
+      end
+
+      it 'rolls back the migrations' do
+        assert { get_migrations == [1] }
+        assert { get_data       == [1] }
+      end
+    end
+  end
+
+  describe '.rollback' do
+    before do
+      ActiveColumn::Migrator.migrate migrations_path, 3
+    end
+
+    context 'given no steps' do
+      before do
+        ActiveColumn::Migrator.rollback migrations_path
+      end
+
+      it 'rolls back one step' do
+        assert { get_migrations == [1, 2] }
+        assert { get_data       == [1, 2] }
+      end
+    end
+
+    context 'given steps = 2' do
+      before do
+        ActiveColumn::Migrator.rollback migrations_path, 2
+      end
+
+      it 'rolls back two steps' do
+        assert { get_migrations == [1] }
+        assert { get_data       == [1] }
+      end
+    end
+  end
+
+  describe '.forward' do
+    before do
+      ActiveColumn::Migrator.migrate migrations_path, 1
+    end
+
+    context 'given no steps' do
+      before do
+        ActiveColumn::Migrator.forward migrations_path
+      end
+
+      it 'migrates one step' do
+        assert { get_migrations == [1, 2] }
+        assert { get_data       == [1, 2] }
+      end
+    end
+
+    context 'given steps = 2' do
+      before do
+        ActiveColumn::Migrator.forward migrations_path, 2
+      end
+
+      it 'migrates two steps' do
+        assert { get_migrations == [1, 2, 3] }
+        assert { get_data       == [1, 2, 3] }
+      end
+    end
+  end
+
+end

data/spec/active_column/tasks/column_family_spec.rb

@@ -0,0 +1,34 @@
+require 'spec_helper'
+
+def translated_comparator(given)
+  cf = ActiveColumn::Tasks::ColumnFamily.new
+  cf.send(:post_process_options, { :comparator_type => given })[:comparator_type]
+end
+
+describe ActiveColumn::Tasks::ColumnFamily do
+
+  describe '.post_process_options' do
+    context 'given a time-based comparator_type' do
+      it 'sets TimeUUIDType' do
+        assert { translated_comparator(:time)          == 'TimeUUIDType' }
+        assert { translated_comparator(:timestamp)     == 'TimeUUIDType' }
+        assert { translated_comparator('TimeUUIDType') == 'TimeUUIDType' }
+      end
+    end
+
+    context 'given a long-based comparator_type' do
+      it 'sets LongType' do
+        assert { translated_comparator(:long)      == 'LongType' }
+        assert { translated_comparator('LongType') == 'LongType' }
+      end
+    end
+
+    context 'given a string-based comparator_type' do
+      it 'sets BytesType' do
+        assert { translated_comparator(:string)     == 'BytesType' }
+        assert { translated_comparator('BytesType') == 'BytesType' }
+      end
+    end
+  end
+
+end

data/spec/active_column/tasks/keyspace_spec.rb

@@ -0,0 +1,38 @@
+require 'spec_helper'
+
+describe ActiveColumn::Tasks::Keyspace do
+
+  before do
+    @ks = ActiveColumn::Tasks::Keyspace.new
+  end
+
+  describe ".create" do
+    context "given a keyspace" do
+      before do
+        @ks.drop :ks_create_test if @ks.exists?(:ks_create_test)
+        @ks.create :ks_create_test
+      end
+
+      it "creates the keyspace" do
+        @ks.exists?(:ks_create_test).should be
+      end
+
+      after do
+        @ks.drop :ks_create_test
+      end
+    end
+  end
+
+  describe '.drop' do
+    context 'given a keyspace' do
+      before do
+        @ks.create :ks_drop_test unless @ks.exists?(:ks_drop_test)
+        @ks.drop :ks_drop_test
+      end
+
+      it 'drops the keyspace' do
+        @ks.exists?(:ks_drop_test).should_not be
+      end
+    end
+  end
+end