def call(self, command, params=None):
"""
Sends the provided command to Serf for evaluation, with
any parameters as the message body.
"""
if self._socket is None:
raise SerfConnectionError('handshake must be made first')
header = msgpack.packb({"Seq": self._counter(), "Command": command})
if params is not None:
body = msgpack.packb(params)
self._socket.sendall(header + body)
else:
self._socket.sendall(header)
unpacker = msgpack.Unpacker()
unpacker.feed(self._socket.recv(4096))
response = SerfResult()
for item in unpacker:
if response.head is None:
response.head = item
else:
response.body = item
break
return response
def call(self, command, params=None, expect_body=True):
"""
Sends the provided command to Serf for evaluation, with
any parameters as the message body.
"""
if self._socket is None:
raise SerfConnectionError('handshake must be made first')
header = msgpack.packb({"Seq": self._counter(), "Command": command})
if params is not None:
body = msgpack.packb(params)
self._socket.sendall(header + body)
else:
self._socket.sendall(header)
# The number of msgpack messages that are expected
# in response to this command.
messages_expected = 2 if expect_body else 1
response = SerfResult()
unpacker = msgpack.Unpacker(object_hook=self._decode_addr_key)
# Continue reading from the network until the expected number of
# msgpack messages have been received.
while messages_expected > 0:
try:
buf = self._socket.recv(self._socket_recv_size)
if len(buf) == 0: # Connection was closed.
raise SerfConnectionError("Connection closed by peer")
unpacker.feed(buf)
except socket.timeout:
raise SerfTimeout(
"timeout while waiting for an RPC response. (Have %s so"
"far)", response)
# Might have received enough to deserialise one or more
# messages, try to fill out the response object.
for message in unpacker:
if response.head is None:
response.head = message
elif response.body is None:
response.body = message
else:
raise SerfProtocolError(
"protocol handler got more than 2 messages. "
"Unexpected message is: %s", message)
# Expecting one fewer message now.
messages_expected -= 1
return response
def call(self, command, params=None, expect_body=True):
"""
Sends the provided command to Serf for evaluation, with
any parameters as the message body.
"""
if self._socket is None:
raise SerfConnectionError('handshake must be made first')
header = msgpack.packb({"Seq": self._counter(), "Command": command})
if params is not None:
body = msgpack.packb(params)
self._socket.sendall(header + body)
else:
self._socket.sendall(header)
# The number of msgpack messages that are expected
# in response to this command.
messages_expected = 2 if expect_body else 1
response = SerfResult()
unpacker = msgpack.Unpacker(object_hook=self._decode_addr_key)
# Continue reading from the network until the expected number of
# msgpack messages have been received.
while messages_expected > 0:
try:
buf = self._socket.recv(self._socket_recv_size)
if len(buf) == 0: # Connection was closed.
raise SerfConnectionError("Connection closed by peer")
unpacker.feed(buf)
except socket.timeout:
raise SerfTimeout(
"timeout while waiting for an RPC response. (Have %s so"
"far)", response)
# Might have received enough to deserialise one or more
# messages, try to fill out the response object.
for message in unpacker:
if response.head is None:
response.head = message
elif response.body is None:
response.body = message
else:
raise SerfProtocolError(
"protocol handler got more than 2 messages. "
"Unexpected message is: %s", message)
# Expecting one fewer message now.
messages_expected -= 1
return response
def keep_reading_from_stream(init=[]):
sub_response = SerfResult()
while True:
if init is not None:
it = init
init = None
else:
if self._socket is None:
return
read_from_socket()
it = unpacker
for msg in it:
if sub_response.head is None:
sub_response.head = msg
elif sub_response.body is None:
sub_response.body = msg
yield sub_response
sub_response = SerfResult()
def keep_reading_from_stream(init=[]):
sub_response = SerfResult()
while True:
if init is not None:
it = init
init = None
else:
if self._socket is None:
return
read_from_socket()
it = unpacker
for msg in it:
if sub_response.head is None:
sub_response.head = msg
elif sub_response.body is None:
sub_response.body = msg
yield sub_response
sub_response = SerfResult()
