一、使用os.kill()
os.kill()是一种向进程发送信号的方法，可以用来强制结束一个进程的运行。如果你的程序中包含有线程，用这种方式绝对没错！当使用os.kill()方法结束一个进程时，需要指定该进程的PID（进程号），同时需要指定信号类型。下面是使用os.kill()方法结束指定ID的程序运行的示例代码：
import os
pid = os.getpid() # 获取当前进程的PID
os.kill(pid, signal.SIGTERM) # 主动结束指定ID的程序运行二、使用os._exit()
os._exit()是一种基于系统调用的退出方式，它可以忽略所有的清理活动，包括所有的缓存和终止信号处理器等。当使用os.exit()方法退出时，Python解释器将立即终止程序的运行，不会执行任何清理操作并且不会返回状态码。下面是使用os._exit()方法退出程序的示例代码：
import os
os._exit(0) # 退出程序，不返回状态码三、使用raise SystemExit()
raise SystemExit()是一种简单的退出方式，它可以引发一个SystemExit异常，终止程序执行，并且可返回一个状态码。在使用raise SystemExit()方法退出时，如果指定了状态码，则会返回该状态码，否则将返回0。下面是使用raise SystemExit()方法退出程序的示例代码：
raise SystemExit(0) # 退出程序，返回状态码0四、使用exit()
exit()是一种基于终端指令的退出方式，它可以退出Python解释器并返回状态码。当使用exit()方法退出时，会先执行一些清理工作，如关闭文件等，然后退出Python解释器。下面是使用exit()方法退出程序的示例代码：
exit(0) # 退出程序，返回状态码0五、使用sys.exit()
sys.exit()是一种常用的退出当前进程的方法，它可以终止一个程序的运行，并可以返回一个状态码。当sys.exit()被调用时，程序将立即停止执行，所有的finally子句（如果有的话）将被忽略，并且返回一个标识状态的整数。下面是使用sys.exit()方法退出程序的示例代码：
import sys
sys.exit(0) # 退出程序，返回状态码0就像websocket客户端，当服务端主动关闭后，客户端直接退出：
服务端代码：
from flask import Flask
from flask_sock import Sock
app = Flask(__name__)
sock = Sock(app)
# 创建全局的ws对象数组
ws_list = []
ws_closed = []
@app.route('/')
def index():
return "你好，世界"
@app.route('/ws')
def websocket():
return """
<!doctype html>
<html>
<head>
<title>Flask-Sock Demo</title>
</head>
<body>
<h1>Flask-Sock Demo</h1>
<div id="log"></div>
<br>
<form id="form">
<label for="text">Input: </label>
<input type="text" id="text" autofocus>
</form>
<script>
const log = (text, color) => {
document.getElementById('log').innerHTML += `<span style="color: ${color}">${text}</span><br>`;
};
const socket = new WebSocket('ws://' + location.host + '/echo');
socket.addEventListener('message', ev => {
log('<<< ' + ev.data, 'blue');
});
document.getElementById('form').onsubmit = ev => {
ev.preventDefault();
const textField = document.getElementById('text');
log('>>> ' + textField.value, 'red');
socket.send(textField.value);
textField.value = '';
};
</script>
</body>
</html>
"""
def handle_sync(data, self):
# 同步各端ws消息
for ws in ws_list:
if not ws.connected:
ws_closed.append(ws)
elif data and ws != self:
ws.send(data)
# 删除已断开的链接
for cl in ws_closed:
ws_list.remove(cl)
ws_closed.clear()
print(f"live num: {len(ws_list)}, done num: {len(ws_closed)}")
@sock.route('/echo')
def echo(self):
global ws_list
ws_list.append(self)
# 同步消息和关闭ws clear
handle_sync(None, self)
while True:
print(f"global ws list: {len(ws_list)}")
# 判断是断开连接还是还在链接
data = self.receive()
print(f"receive client message: {data}")
self.send(data)
# 同步消息和关闭ws clear
handle_sync(data, self)
if __name__ == '__main__':
app.run(host="0.0.0.0", port=8989)
客户端代码：
import os
import signal
import threading
import time
import websocket
def on_open(ws):
print("连接已建立")
# 发送消息给服务器
ws.send("Hello, server!")
def on_message(ws, message):
print("收到消息：", message)
def on_close(ws, arg1, arg2):
print("连接已关闭")
pid = os.getpid() # 获取当前进程的PID
os.kill(pid, signal.SIGTERM)
def on_error(ws, error):
print("发生错误：", error)
# 创建WebSocket连接
ws = websocket.WebSocketApp("ws://192.168.110.196:8989/echo",
on_open=on_open,
on_message=on_message,
on_close=on_close,
on_error=on_error)
def sub_task():
while True:
print("子线程运行")
time.sleep(2)
def run():
t = threading.Thread(target=sub_task)
t.start()
# 启动WebSocket客户端
ws.run_forever()
if __name__ == '__main__':
run()
相关内容