def send_body(self, f, protocol_version):
write_longstring(f, self.query)
flags = 0x00
if self.keyspace is not None:
if ProtocolVersion.uses_keyspace_flag(protocol_version):
flags |= _PREPARED_WITH_KEYSPACE_FLAG
else:
raise UnsupportedOperation(
"Keyspaces may only be set on queries with protocol version "
"5 or higher. Consider setting Cluster.protocol_version to 5.")
if ProtocolVersion.uses_prepare_flags(protocol_version):
write_uint(f, flags)
else:
# checks above should prevent this, but just to be safe...
if flags:
raise UnsupportedOperation(
"Attempted to set flags with value {flags:0=#8x} on"
"protocol version {pv}, which doesn't support flags"
"in prepared statements."
"Consider setting Cluster.protocol_version to 5."
"".format(flags=flags, pv=protocol_version))
if ProtocolVersion.uses_keyspace_flag(protocol_version):
if self.keyspace:
write_string(f, self.keyspace)
def send_body(self, f, protocol_version):
write_longstring(f, self.query)
flags = 0x00
if self.keyspace is not None:
if ProtocolVersion.uses_keyspace_flag(protocol_version):
flags |= _PREPARED_WITH_KEYSPACE_FLAG
else:
raise UnsupportedOperation(
"Keyspaces may only be set on queries with protocol version "
"5 or higher. Consider setting Cluster.protocol_version to 5."
)
if ProtocolVersion.uses_prepare_flags(protocol_version):
write_uint(f, flags)
else:
# checks above should prevent this, but just to be safe...
if flags:
raise UnsupportedOperation(
"Attempted to set flags with value {flags:0=#8x} on"
"protocol version {pv}, which doesn't support flags"
"in prepared statements."
"Consider setting Cluster.protocol_version to 5."
"".format(flags=flags, pv=protocol_version))
if ProtocolVersion.uses_keyspace_flag(protocol_version):
if self.keyspace:
write_string(f, self.keyspace)
def send_body(self, f, protocol_version):
write_byte(f, self.batch_type.value)
write_short(f, len(self.queries))
for prepared, string_or_query_id, params in self.queries:
if not prepared:
write_byte(f, 0)
write_longstring(f, string_or_query_id)
else:
write_byte(f, 1)
write_short(f, len(string_or_query_id))
f.write(string_or_query_id)
write_short(f, len(params))
for param in params:
write_value(f, param)
write_consistency_level(f, self.consistency_level)
if protocol_version >= 3:
flags = 0
if self.serial_consistency_level:
flags |= _WITH_SERIAL_CONSISTENCY_FLAG
if self.timestamp is not None:
flags |= _PROTOCOL_TIMESTAMP
if self.keyspace:
if ProtocolVersion.uses_keyspace_flag(protocol_version):
flags |= _WITH_KEYSPACE_FLAG
else:
raise UnsupportedOperation(
"Keyspaces may only be set on queries with protocol version "
"5 or higher. Consider setting Cluster.protocol_version to 5."
)
if ProtocolVersion.uses_int_query_flags(protocol_version):
write_int(f, flags)
else:
write_byte(f, flags)
if self.serial_consistency_level:
write_consistency_level(f, self.serial_consistency_level)
if self.timestamp is not None:
write_long(f, self.timestamp)
if ProtocolVersion.uses_keyspace_flag(protocol_version):
if self.keyspace is not None:
write_string(f, self.keyspace)
def send_body(self, f, protocol_version):
write_byte(f, self.batch_type.value)
write_short(f, len(self.queries))
for prepared, string_or_query_id, params in self.queries:
if not prepared:
write_byte(f, 0)
write_longstring(f, string_or_query_id)
else:
write_byte(f, 1)
write_short(f, len(string_or_query_id))
f.write(string_or_query_id)
write_short(f, len(params))
for param in params:
write_value(f, param)
write_consistency_level(f, self.consistency_level)
if protocol_version >= 3:
flags = 0
if self.serial_consistency_level:
flags |= _WITH_SERIAL_CONSISTENCY_FLAG
if self.timestamp is not None:
flags |= _PROTOCOL_TIMESTAMP
if self.keyspace:
if ProtocolVersion.uses_keyspace_flag(protocol_version):
flags |= _WITH_KEYSPACE_FLAG
else:
raise UnsupportedOperation(
"Keyspaces may only be set on queries with protocol version "
"5 or higher. Consider setting Cluster.protocol_version to 5.")
if ProtocolVersion.uses_int_query_flags(protocol_version):
write_int(f, flags)
else:
write_byte(f, flags)
if self.serial_consistency_level:
write_consistency_level(f, self.serial_consistency_level)
if self.timestamp is not None:
write_long(f, self.timestamp)
if ProtocolVersion.uses_keyspace_flag(protocol_version):
if self.keyspace is not None:
write_string(f, self.keyspace)
def test_prepare_flag_with_keyspace(self):
message = PrepareMessage("a", keyspace='ks')
io = Mock()
for version in ProtocolVersion.SUPPORTED_VERSIONS:
if ProtocolVersion.uses_keyspace_flag(version):
message.send_body(io, version)
self._check_calls(io, [
(b'\x00\x00\x00\x01',),
(b'a',),
(b'\x00\x00\x00\x01',),
(b'\x00\x02',),
(b'ks',),
])
else:
with self.assertRaises(UnsupportedOperation):
message.send_body(io, version)
io.reset_mock()
def test_prepare_flag_with_keyspace(self):
message = PrepareMessage("a", keyspace='ks')
io = Mock()
for version in ProtocolVersion.SUPPORTED_VERSIONS:
if ProtocolVersion.uses_keyspace_flag(version):
message.send_body(io, version)
self._check_calls(io, [
(b'\x00\x00\x00\x01', ),
(b'a', ),
(b'\x00\x00\x00\x01', ),
(b'\x00\x02', ),
(b'ks', ),
])
else:
with self.assertRaises(UnsupportedOperation):
message.send_body(io, version)
io.reset_mock()
def send_body(self, f, protocol_version):
write_longstring(f, self.query)
write_consistency_level(f, self.consistency_level)
flags = 0x00
if self._query_params is not None:
flags |= _VALUES_FLAG # also v2+, but we're only setting params internally right now
if self.serial_consistency_level:
if protocol_version >= 2:
flags |= _WITH_SERIAL_CONSISTENCY_FLAG
else:
raise UnsupportedOperation(
"Serial consistency levels require the use of protocol version "
"2 or higher. Consider setting Cluster.protocol_version to 2 "
"to support serial consistency levels.")
if self.fetch_size:
if protocol_version >= 2:
flags |= _PAGE_SIZE_FLAG
else:
raise UnsupportedOperation(
"Automatic query paging may only be used with protocol version "
"2 or higher. Consider setting Cluster.protocol_version to 2."
)
if self.paging_state:
if protocol_version >= 2:
flags |= _WITH_PAGING_STATE_FLAG
else:
raise UnsupportedOperation(
"Automatic query paging may only be used with protocol version "
"2 or higher. Consider setting Cluster.protocol_version to 2."
)
if self.timestamp is not None:
flags |= _PROTOCOL_TIMESTAMP
if self.keyspace is not None:
if ProtocolVersion.uses_keyspace_flag(protocol_version):
flags |= _WITH_KEYSPACE_FLAG
else:
raise UnsupportedOperation(
"Keyspaces may only be set on queries with protocol version "
"5 or higher. Consider setting Cluster.protocol_version to 5."
)
if ProtocolVersion.uses_int_query_flags(protocol_version):
write_uint(f, flags)
else:
write_byte(f, flags)
if self._query_params is not None:
write_short(f, len(self._query_params))
for param in self._query_params:
write_value(f, param)
if self.fetch_size:
write_int(f, self.fetch_size)
if self.paging_state:
write_longstring(f, self.paging_state)
if self.serial_consistency_level:
write_consistency_level(f, self.serial_consistency_level)
if self.timestamp is not None:
write_long(f, self.timestamp)
if self.keyspace is not None:
write_string(f, self.keyspace)
def send_body(self, f, protocol_version):
write_longstring(f, self.query)
write_consistency_level(f, self.consistency_level)
flags = 0x00
if self._query_params is not None:
flags |= _VALUES_FLAG # also v2+, but we're only setting params internally right now
if self.serial_consistency_level:
if protocol_version >= 2:
flags |= _WITH_SERIAL_CONSISTENCY_FLAG
else:
raise UnsupportedOperation(
"Serial consistency levels require the use of protocol version "
"2 or higher. Consider setting Cluster.protocol_version to 2 "
"to support serial consistency levels.")
if self.fetch_size:
if protocol_version >= 2:
flags |= _PAGE_SIZE_FLAG
else:
raise UnsupportedOperation(
"Automatic query paging may only be used with protocol version "
"2 or higher. Consider setting Cluster.protocol_version to 2.")
if self.paging_state:
if protocol_version >= 2:
flags |= _WITH_PAGING_STATE_FLAG
else:
raise UnsupportedOperation(
"Automatic query paging may only be used with protocol version "
"2 or higher. Consider setting Cluster.protocol_version to 2.")
if self.timestamp is not None:
flags |= _PROTOCOL_TIMESTAMP
if self.keyspace is not None:
if ProtocolVersion.uses_keyspace_flag(protocol_version):
flags |= _WITH_KEYSPACE_FLAG
else:
raise UnsupportedOperation(
"Keyspaces may only be set on queries with protocol version "
"5 or higher. Consider setting Cluster.protocol_version to 5.")
if ProtocolVersion.uses_int_query_flags(protocol_version):
write_uint(f, flags)
else:
write_byte(f, flags)
if self._query_params is not None:
write_short(f, len(self._query_params))
for param in self._query_params:
write_value(f, param)
if self.fetch_size:
write_int(f, self.fetch_size)
if self.paging_state:
write_longstring(f, self.paging_state)
if self.serial_consistency_level:
write_consistency_level(f, self.serial_consistency_level)
if self.timestamp is not None:
write_long(f, self.timestamp)
if self.keyspace is not None:
write_string(f, self.keyspace)