我有一个python脚本,只有一个连接到Java服务器的套接字对象。
我启动了一个线程,用于每5秒钟将心跳消息发送到服务器。
和另一个用于从服务器接收消息的线程。
顺便说一句,所有数据发送/recv均以原状格式。
# 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
我遇到的问题是我在调用发送方法后在请求方法中收到的数据似乎被线程中的心跳数据所困扰。
如果我评论心跳线和接收线程,请求方法可以正常工作。来自服务器的数据可以解码而没有错误,并且可以成功发送回浏览器。
我的解决方案现在可能是错误的,我真的不知道如何获得这项工作。
任何建议都将不胜感激,谢谢:)
python中的 socket
对象不是线程安全,您需要借助某些同步基原始人(例如python 2 in Python 2中的threading.Lock
)访问共享资源(在这种情况下为client_socket
对象)。在此处检查一个类似的问题:Python:套接字和线程?