You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.1 KiB
79 lines
2.1 KiB
import os
|
|
import signal
|
|
from time import sleep
|
|
|
|
import threading
|
|
|
|
import numpy as np
|
|
import requests
|
|
from flask import Flask, Response
|
|
|
|
import utils
|
|
|
|
app_flask = Flask(__name__)
|
|
|
|
# 全局变量，用于存储最新的帧
|
|
latest_frame:np.ndarray = None
|
|
lock = threading.Lock()
|
|
is_running = True
|
|
|
|
def update_latest_frame(n_bytes:np.ndarray):
|
|
global latest_frame
|
|
latest_frame=n_bytes
|
|
|
|
def generate_mjpeg():
|
|
"""生成MJPEG格式的视频流"""
|
|
while is_running:
|
|
with lock:
|
|
if latest_frame is None:
|
|
continue
|
|
_,latest_frame_bytes = utils.frame_to_img(latest_frame)
|
|
frame = latest_frame_bytes.tobytes()
|
|
pass
|
|
sleep(0.1)
|
|
yield (b'--frame\r\n'
|
|
b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
|
|
|
|
|
|
@app_flask.route('/video_flow')
|
|
def video_feed():
|
|
"""视频流路由"""
|
|
return Response(
|
|
generate_mjpeg(),
|
|
mimetype='multipart/x-mixed-replace; boundary=frame'
|
|
)
|
|
|
|
# 添加关闭路由
|
|
@app_flask.route('/shutdown', methods=['POST'])
|
|
def shutdown():
|
|
os.kill(os.getpid(), signal.SIGTERM)
|
|
return 'Server shutting down...'
|
|
def run():
|
|
port=2240
|
|
print(f"推流服务启动,访问端口 127.0.0.1:{port}/video_flow")
|
|
app_flask.run(host='0.0.0.0', port=port, threaded=True)
|
|
|
|
def video_server_run():
|
|
thread = threading.Thread(target=run)
|
|
thread.daemon = True # 设置为守护线程，主线程退出时自动终止
|
|
thread.start()
|
|
def stop():
|
|
global is_running
|
|
is_running = False
|
|
# 通过HTTP请求关闭Flask服务器
|
|
try:
|
|
response = requests.post('http://localhost:2240/shutdown', timeout=1)
|
|
print(f"服务器关闭请求发送成功，状态码: {response.status_code}")
|
|
except requests.exceptions.ConnectionError as e:
|
|
print(f"无法连接到服务器: {e}")
|
|
except requests.exceptions.Timeout as e:
|
|
print(f"关闭请求超时: {e}")
|
|
except requests.exceptions.RequestException as e:
|
|
print(f"发送关闭请求时发生错误: {e}")
|
|
except Exception as e:
|
|
print(f"关闭服务器时发生未知错误: {e}")
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
run()
|