masscollaborationlabs/p2pfs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Only convert Enum to int if message type header is Enum.

Browse source
This commit is contained in:
Yuxin Wang 2018-10-09 21:04:23 -04:00
parent c35c2f2c61
commit 373adc7e30
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
p2pfs/core/server.py
View file

@ -77,7 +77,8 @@ class MessageServer:
assert isinstance(writer, asyncio.StreamWriter)
logger.debug('Writing {}'.format(self._message_log(message)))
# use value of enum since Enum is not JSON serializable
message['type'] = message['type'].value
if isinstance(message['type'], MessageType):
message['type'] = message['type'].value
# json string (str) -> encode to utf8 (bytes) -> compress (bytes) -> add length header (bytes)
raw_msg = json.dumps(message).encode('utf-8')
compressed = self._compressor.compress(raw_msg)