openstack/deb-python-cassandra-driver
Code Issues Proposed changes

getting noop counter column updates working properly

Browse Source
This commit is contained in:
Blake Eggleston
2013-11-03 16:31:26 -08:00
parent 7a4e2be45e
commit b41a1d6d63
2 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cqlengine/statements.py
View File

@@ -337,11 +337,10 @@ class CounterUpdateClause(ContainerUpdateClause):
self.previous = self.previous or 0 self.previous = self.previous or 0
def get_context_size(self): def get_context_size(self):
return 1 if self.value != self.previous else 0 return 1
def update_context(self, ctx): def update_context(self, ctx):
if self.value != self.previous: ctx[str(self.context_id)] = self._to_database(abs(self.value - self.previous))
ctx[str(self.context_id)] = self._to_database(abs(self.value - self.previous))
def __unicode__(self): def __unicode__(self):
delta = self.value - self.previous delta = self.value - self.previous

7
cqlengine/tests/statements/test_assignment_clauses.py
View File

@@ -296,7 +296,12 @@ class CounterUpdateTests(TestCase):
c = CounterUpdateClause('a', 5, 5) c = CounterUpdateClause('a', 5, 5)
c.set_context_id(5) c.set_context_id(5)
self.assertEqual(c.get_context_size(), 0) self.assertEqual(c.get_context_size(), 1)
self.assertEqual(str(c), '"a" = "a" + :5')
ctx = {}
c.update_context(ctx)
self.assertEqual(ctx, {'5': 0})
class MapDeleteTests(TestCase): class MapDeleteTests(TestCase):