多线程

时间:2016-12-07 06:46:14

标签: python multithreading sockets

我的python脚本只有一个连接到java服务器的套接字对象。

我开始了一个线程,每隔5秒向服务器发送心跳消息。

另一个用于从服务器接收消息的线程。

顺便说一下,所有数据发送/接收都是protobuffer格式。

# socket_client.py

def recv_handler():
    global client_socket
    while True:
        try:
            # read 4 bytes first 
            pack_len = client_socket.recv(4)
            pack_len = struct.unpack('!i', pack_len)[0]
            # read the rest
            recv_data = client_socket.recv(pack_len)
            # decode
            decompressed_data = data_util.decompressMessage(recv_data)
            sc_pb_message = data_util.decodePBMessage(decompressed_data)
            sc_head = data_util.parseHead(sc_pb_message)
        except:
            print 'error'

def heart_handler():
    global client_socket
    while True:
        if client_socket:
            message = data_util.makeMessage('MSG_HEART_BEAT')
            compressed_data = data_util.compressMessage(message)
            send_data = data_util.makeSendData(compressed_data)
            try: 
                client_socket.send(send_data)
            except: 
                print 'except'
                pass
        time.sleep(5)

def connect(address, port):
    global client_socket
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    client_socket.connect((address, port))

    # thread sending heart beat message
    th = threading.Thread(target = heart_handler)
    th.start()

    # thread recving message
    tr = threading.Thread(target = recv_handler)
    tr.start()

上面的代码工作得很好。该脚本每5秒发送一次心跳消息,并从服务器接收消息,消息可以成功解码。

触发器部分比我不知道如何实现。

我的python脚本需要同时从浏览器接收输入,所以我启动了一个BaseHTTPServer来处理来自浏览器的POST请求。

当请求到来时,我想调用client_socket.send方法向服务器发送特定消息,当然我需要将数据从服务器返回给浏览器。

# http_server.py

def do_POST(self):
    # ...
    result = socket_client.request(message)
    self.send_response(200)
    self.end_headers()
    self.wfile.write(...)

以下是我在请求中尝试做的事情:

def request(message):
    global client_socket
    client_socket.send(message)
    pack_len = client_socket.recv(4)
    pack_len = struct.unpack('!i', pack_len)[0]
    recv_data = client_socket.recv(pack_len)
    return recv_data

我遇到的问题是在调用send方法后我在请求方法中收到的数据似乎受到线程中心跳数据的干扰。

如果我注释掉心跳线程和接收线程,那么请求方法就可以了。来自服务器的数据可以无错误地解码,并且可以成功发送回浏览器。

我的解决方案现在可能是错的,我真的不知道如何让这项工作。

任何建议都将不胜感激,谢谢:)

1 个答案:

答案 0 :(得分:1)

Python中的

socket对象不是线程安全的,您需要借助一些同步原语(例如{{1})来访问共享资源(在本例中为client_socket对象)在Python中2.检查此处是否存在类似问题:Python: Socket and threads?