Skip to content

Commit fe640ea

Browse files
committed
Use ServiceUnavailable in BufferingSocket
1 parent 13b34fc commit fe640ea

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

neo4j/v1/bolt.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,7 @@
8484
class BufferingSocket(object):
8585

8686
def __init__(self, socket):
87+
self.address = socket.getpeername()
8788
self.socket = socket
8889
self.buffer = bytearray()
8990

@@ -96,7 +97,7 @@ def fill(self):
9697
self.buffer[len(self.buffer):] = received
9798
else:
9899
if ready_to_read is not None:
99-
raise ProtocolError("Server closed connection")
100+
raise ServiceUnavailable("Failed to read from connection %r" % (self.address,))
100101

101102
def read_message(self):
102103
message_data = bytearray()

0 commit comments

Comments
 (0)