gotime-cassandra_object 4.3.2 → 4.4.0

data/gotime-cassandra_object.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |s|
   s.name = 'gotime-cassandra_object'
-  s.version = '4.3.2'
+  s.version = '4.4.0'
   s.description = 'Cassandra ActiveModel'
   s.summary = 'Cassandra ActiveModel'
   s.authors = ["Michael Koziarski", "gotime"]

data/lib/cassandra_object/base.rb

@@ -29,6 +29,7 @@ module CassandraObject
     extend ActiveSupport::DescendantsTracker
     
     include Connection
+    include Consistency
     include Identity
     include Inspect
     include Persistence

data/lib/cassandra_object/consistency.rb

@@ -0,0 +1,18 @@
+module CassandraObject
+  module Consistency
+    extend ActiveSupport::Concern
+
+    module ClassMethods
+      def with_consistency(consistency)
+        previous, self.default_consistency = default_consistency, consistency
+        yield
+      ensure
+        self.default_consistency = previous
+      end
+    end
+
+    included do
+      class_attribute :default_consistency
+    end
+  end
+end

data/lib/cassandra_object/scope/batches.rb

@@ -24,7 +24,7 @@ module CassandraObject
           yield records
           break if next_record.nil?
 
-          records = scope.where("KEY >= #{next_record.id}").to_a
+          records = scope.where("KEY >= '#{next_record.id}'").to_a
         end
       end
     end

data/lib/cassandra_object/scope/finder_methods.rb

@@ -20,7 +20,8 @@ module CassandraObject
       end
 
       def first
-        limit(1).to_a.first
+        to_a.first
+        # limit(1).to_a.first
       end
 
       private

data/lib/cassandra_object/scope/query_methods.rb

@@ -36,7 +36,8 @@ module CassandraObject
         statement = [
           "SELECT #{select_string} FROM #{klass.column_family}",
           where_string,
-          limit_string
+          limit_string,
+          consistency_string
         ].delete_if(&:blank?) * ' '
 
         instantiate_from_cql(statement)
@@ -97,6 +98,12 @@ module CassandraObject
             ""
           end
         end
+
+        def consistency_string
+          if klass.default_consistency
+            "USING CONSISTENCY #{klass.default_consistency}"
+          end
+        end
     end
   end
 end

data/lib/gotime-cassandra_object.rb

@@ -7,6 +7,7 @@ module CassandraObject
 
   autoload :Base
   autoload :Connection
+  autoload :Consistency
   autoload :AttributeMethods
   autoload :BelongsTo
   autoload :Persistence

data/test/unit/consistency_test.rb

@@ -0,0 +1,13 @@
+require "test_helper"
+
+class CassandraObject::ConsistencyTest < CassandraObject::TestCase
+  test "with_consistency" do
+    assert_nil CassandraObject::Base.default_consistency
+
+    CassandraObject::Base.with_consistency 'QUORUM' do
+      assert_equal 'QUORUM', CassandraObject::Base.default_consistency
+    end
+
+    assert_nil CassandraObject::Base.default_consistency
+  end
+end

data/test/unit/scope/query_methods_test.rb

@@ -16,4 +16,10 @@ class CassandraObject::Scope::QueryMethodsTest < CassandraObject::TestCase
 
     assert_equal [foo_issue], Issue.select { |issue| issue.title == 'foo' }
   end
+
+  test "select with consistency" do
+    Issue.with_consistency 'ONE' do
+      Issue.all
+    end
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gotime-cassandra_object
 version: !ruby/object:Gem::Version
-  version: 4.3.2
+  version: 4.4.0
   prerelease: 
 platform: ruby
 authors:
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-14 00:00:00.000000000 Z
+date: 2012-09-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activemodel
@@ -104,6 +104,7 @@ files:
 - lib/cassandra_object/belongs_to/reflection.rb
 - lib/cassandra_object/callbacks.rb
 - lib/cassandra_object/connection.rb
+- lib/cassandra_object/consistency.rb
 - lib/cassandra_object/errors.rb
 - lib/cassandra_object/identity.rb
 - lib/cassandra_object/inspect.rb
@@ -150,6 +151,7 @@ files:
 - test/unit/belongs_to_test.rb
 - test/unit/callbacks_test.rb
 - test/unit/connection_test.rb
+- test/unit/consistency_test.rb
 - test/unit/identity_test.rb
 - test/unit/inspect_test.rb
 - test/unit/log_subscriber_test.rb