Use Enum class to represent message types.

This commit is contained in:
Yuxin Wang 2018-10-05 17:15:00 -04:00
parent fe646894b4
commit 49cd3f5a35

View file

@ -1,4 +1,17 @@
class MessageType:
REQUEST_REGISTER, REQUEST_PUBLISH, REQUEST_FILE_LIST, REQUEST_FILE_LOCATION, REQUEST_CHUNK_REGISTER, REQUEST_LEAVE, \
REPLY_REGISTER, REPLY_FILE_LIST, REPLY_PUBLISH, REPLY_FILE_LOCATION, REPLY_LEAVE, \
PEER_REQUEST_CHUNK, PEER_REPLY_CHUNK = range(13)
from enum import Enum, auto
class MessageType(Enum):
REQUEST_REGISTER = auto()
REQUEST_PUBLISH = auto()
REQUEST_FILE_LIST = auto()
REQUEST_FILE_LOCATION = auto()
REQUEST_CHUNK_REGISTER = auto()
REQUEST_LEAVE = auto()
REPLY_REGISTER = auto()
REPLY_FILE_LIST = auto()
REPLY_PUBLISH = auto()
REPLY_FILE_LOCATION = auto()
REPLY_LEAVE = auto()
PEER_REQUEST_CHUNK = auto()
PEER_REPLY_CHUNK = auto()