Question

我有一个相对较大的枚举，其中每个成员代表一种消息类型。客户端将收到一条消息，其中包含与枚举中的msg类型关联的整数值。对于每个msg类型，将有一个单独的函数回调来处理msg。

我想通过使用稀疏数组（或向量）来尽可能快地查找和调度回调，其中枚举值映射到回调的索引。如果数组不能保存函数类型，这在Python中是否可行？

#pseudo code for 'enum'
class MsgType(object):
    LOGIN, LOGOUT, HEARTBEAT, ... = range(n)

#handler class
class Handler(object):
    def handleMsg(self, msg):
        #dispatch msg to specific handler

    def __onLogin(self, msg):
        #handle login

    def __onLogout(self, msg):
        #handle logout

更新我的术语不清楚。我现在理解Python字典查找具有复杂性O（1），这使它们成为完美的候选者。感谢。

Answer 1

class MsgID(int):
    pass

LOGIN = MsgID(0)
LOGOUT = MsgID(1)
HEARTBEAT = MsgID(2)
... # add all other message identifier numbers

class MsgType(object):
    def __init__(self, id, data):
        self.id = id
        self.data = data


def login_handler(msg):
    ...  # do something here

def logout_handler(msg):
    ...  # do something here

def heartbeat_handler(msg):
    ...  # do something here


msg_func = {
    LOGIN  : login_handler,
    LOGOUT : logout_handler,
    HEARTBEAT : heartbeat_handler,
    ...
}


class Handler(object):
    def handleMsg(self, msg):
        try:
            msg_func[msg.id](msg)  # lookup function reference in dict, call function
        except KeyError:
            log_error_mesg('message without a handler function: %d' % msg.id)

并非严格需要，但我为消息ID添加了int的子类。这样你就可以检查ID值是否真的是一个ID值，而不仅仅是一些随机整数。

我假设每条消息都有一个ID值，标识它是什么类型的消息，加上一些数据。 msg_func字典使用MsgID值作为键，映射到函数引用。

你可以把所有的功能都放在一个班级里，但我在这里没有这样做;它们只是功能。

使用Python在稀疏数组中存储函数

1 个答案: