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

最新推荐文章于 2023-03-05 13:59:33 发布

ball32109

最新推荐文章于 2023-03-05 13:59:33 发布

阅读量2.1k

点赞数

文章标签： python 服务器游戏 merge import

本文链接：https://blog.csdn.net/ball32109/article/details/7901929

版权

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