python游戏服务器消息处理_用python的twisted做个简单游戏服务器原形--客户端消息处理和重组类MessageQueue...

'''

Created on 2012-8-14

@author: qs

'''

import threading

import thread

import time

import struct

class MessageQueue(threading.Thread):

def __init__(self, monitor):

threading.Thread.__init__(self)

self.lock = thread.allocate_lock()

self.msg_queue = []

self.msg_map = {}

self.monitor = monitor

def run(self):

while(True):

while(True):

msg = self.pop()

if (msg == None):

break

self.process_message(msg)

time.sleep(0.001)

def process_message(self, message):

assert len(message) >= 4

index, bytes = struct.unpack('i%ds' % (len(message) - 4), message)

if (self.msg_map.has_key(index) == False):

if (len(bytes) < 4):

self.msg_map[index] = message

return 0

length, bytes = struct.unpack('i%ds' % (len(bytes) - 4), bytes)

if ((len(message) - 4) > length):

self.message_merge(message)

elif ((len(message) - 4) == length):

self.monitor.process_message(message)

else:

self.msg_map[index] = message

else:

pre_message = self.msg_map[index]

del(self.msg_map[index])

message = pre_message + bytes

self.message_merge(message)

return 0

def message_merge(self, message):

index, bytes = struct.unpack('i%ds' % (len(message) - 4), message)

while(len(bytes) >= 4):

length, bytes = struct.unpack('i%ds' % (len(bytes) - 4), bytes)

if (len(bytes) >= (length - 4)):

process_msg, bytes = struct.struct.unpack('%ds%ds' % (length, len(message) - length), message)

self.monitor.process_message(process_msg)

else:

bytes = struct.pack('i%ds' % len(bytes), length, bytes)

break

if (len(bytes) != 0):

self.msg_map[index] = bytes

def push(self, msg):

self.lock.acquire()

self.msg_queue.append(msg)

self.lock.release()

def pop(self):

self.lock.acquire()

item = None

if (len(self.msg_queue) != 0):

item = self.msg_queue.pop()

self.lock.release()

return item以上一个用一个线程启动的消息处理类，当monitor的连接有消息来的时候，立即把消息放到这个MessageQueue线程做合并和其它逻辑处理.当一个消息是完整后，就调用self.monitor.process_message转到monitor类去找到每个连接类去执行实际逻辑