'''
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类去找到每个连接类去执行实际逻辑