我编写了一个多线程Web套接字客户端类,以便用户的(主)线程不会阻塞run_forever()
方法调用。代码似乎工作正常,除了最后,当我停止线程时,它不会干净地关闭Web套接字,我的进程不会退出。每次我必须做kill -9
才能摆脱它。我尝试调用线程的join()
方法来确保主线程等待子进程完成执行,但这没有帮助。
代码如下所示。你能帮我把线程退出/停止优雅吗?
import thread
import threading
import time
import websocket
class WebSocketClient(threading.Thread):
def __init__(self, url):
self.url = url
threading.Thread.__init__(self)
def run(self):
# Running the run_forever() in a seperate thread.
#websocket.enableTrace(True)
self.ws = websocket.WebSocketApp(self.url,
on_message = self.on_message,
on_error = self.on_error,
on_close = self.on_close)
self.ws.on_open = self.on_open
self.ws.run_forever()
def send(self, data):
# Wait till websocket is connected.
while not self.ws.sock.connected:
time.sleep(0.25)
print 'Sending data...', data
self.ws.send("Hello %s" % data)
def stop(self):
print 'Stopping the websocket...'
self.ws.keep_running = False
def on_message(self, ws, message):
print 'Received data...', message
def on_error(self, ws, error):
print 'Received error...'
print error
def on_close(self, ws):
print 'Closed the connection...'
def on_open(self, ws):
print 'Opened the connection...'
if __name__ == "__main__":
wsCli = WebSocketClient("ws://localhost:8888/ws")
wsCli.start()
wsCli.send('Hello')
time.sleep(.1)
wsCli.send('World')
time.sleep(1)
wsCli.stop()
#wsCli.join()
print 'After closing client...'
答案 0 :(得分:5)
总结通过我们在评论中进行的扩展讨论所发现的解决方案(并让您有机会接受并推荐我的答案。; - )
要解决此问题,您需要在self.ws.close()
课程WebSocketClient
方法而非中仅调用stop()
来调用self.ws.keep_running = False
。这样,网络套接字就会干净利落。
我建议您还将线程的daemon
属性设置为True
,这样当主线程因任何原因终止时线程将自动停止 - 以防万一完全意外发生