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

Merge branch 'pr-674'

Browse Source
This commit is contained in:
Adam Holmberg
2016-10-28 13:05:26 -05:00
parent 527ea3ed85 5827fe7ce6
commit 19c1603493
1 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
cassandra/protocol.py
View File

@@ -560,6 +560,9 @@ class QueryMessage(_MessageType):
if self.timestamp is not None: if self.timestamp is not None:
flags |= _PROTOCOL_TIMESTAMP flags |= _PROTOCOL_TIMESTAMP
if protocol_version >= 5:
write_int(f, flags)
else:
write_byte(f, flags) write_byte(f, flags)
if self._query_params is not None: if self._query_params is not None:
@@ -823,7 +826,12 @@ class ExecuteMessage(_MessageType):
"3 or higher. Consider setting Cluster.protocol_version to 3.") "3 or higher. Consider setting Cluster.protocol_version to 3.")
if self.skip_meta: if self.skip_meta:
flags |= _SKIP_METADATA_FLAG flags |= _SKIP_METADATA_FLAG
if protocol_version >= 5:
write_int(f, flags)
else:
write_byte(f, flags) write_byte(f, flags)
write_short(f, len(self.query_params)) write_short(f, len(self.query_params))
for param in self.query_params: for param in self.query_params:
write_value(f, param) write_value(f, param)
@@ -872,6 +880,10 @@ class BatchMessage(_MessageType):
flags |= _WITH_SERIAL_CONSISTENCY_FLAG flags |= _WITH_SERIAL_CONSISTENCY_FLAG
if self.timestamp is not None: if self.timestamp is not None:
flags |= _PROTOCOL_TIMESTAMP flags |= _PROTOCOL_TIMESTAMP
if protocol_version >= 5:
write_int(f, flags)
else:
write_byte(f, flags) write_byte(f, flags)
if self.serial_consistency_level: if self.serial_consistency_level: