如果另一个事件处理程序进程未完成,则 Flask-soketio 应用程序不会处理来自客户端的新发出事件



我想使用flask,flask-socketio来实现一个在浏览器上接受"命令字符串"(如:ping 8.8.8.8)的应用程序,然后服务器执行它并将命令的输出实时返回给浏览器。

我的实现如下。

客户端向服务器发出"exec_event"事件,服务器运行命令,并在命令生成新输出时不断向客户端发出"exec_response"事件。

如果客户端想要终止命令,它也可以向服务器发送"ctrlc_event"。

我得到的问题是:

服务器在处理

上一个"exec_event"时无法处理"ctrlc_event"。如果我直接在服务器上杀死了长时间运行的命令,例如"ping 8.8.8.8",那么服务器就会开始处理"ctrlc_event"。

谁能指出我的原因或给我一些方向?

服务器端代码

from gevent import monkey
monkey.patch_all()
from flask import Flask, render_template, session, Response
from flask.ext.wtf import Form
from wtforms import StringField, SubmitField
from wtforms.validators import DataRequired
from flask.ext.socketio import SocketIO, emit
import os
import time 
from subprocess import Popen, PIPE

class ExecForm(Form):
    command = StringField('Command: ', validators=[DataRequired()])
    submit = SubmitField('Execute')
class CtrlCForm(Form):
    submit = SubmitField('Ctrl C')

app = Flask(__name__)
app.debug = True
app.config['SECRET_KEY'] = 'hard-to-guess'
socketio = SocketIO(app)
@app.route('/')
def index():
    exec_form = ExecForm()
    ctrlc_form = CtrlCForm()
    return render_template('index.html', 
                           exec_form=exec_form, 
                           ctrlc_form=ctrlc_form)

@socketio.on('exec_event', namespace='/test')
def execute(message):
    command = message['data']
    process = Popen(command, shell=True, stdout=PIPE, stderr=PIPE,
                    close_fds=True, preexec_fn=os.setsid)
    session['pid'] = process.pid
    for out in iter(process.stdout.readline, ''):
        out = '<pre>' + out + '</pre>'
        emit('exec_response', {'data': out})
        time.sleep(0.001)
    out = "Command finished." + '<br /><br />'
    emit('exec_terminate', {'data': out})

@socketio.on('ctrlc_event', namespace='/test')
def ctrlc(message):
    print("Received ctrlc")
    # do something to kill the process
    # finally emit an event.
    out = "Process killed." + '<br /><br/>'
    emit('ctrlc_response', {'data': out})

if __name__ == '__main__':
    socketio.run(app, host='0.0.0.0', port=8080)

客户端代码是

<html>
<head>
<title>Example</title>
<script type="text/javascript" src='//code.jquery.com/jquery-1.11.0.min.js'></script>
<script type="text/javascript" src="//cdnjs.cloudflare.com/ajax/libs/socket.io/0.9.16/socket.io.min.js"></script>
<script type="text/javascript">
$(document).ready(function(){
    var socket = io.connect('http://' + document.domain + ':' + location.port + '/test');
    $('#exec_form').submit(function(event) {
        socket.emit('exec_event', {data: $('#exec_input').val()});
        return false;
    }); 
    $('#ctrlc_form').submit(function(event) {
        socket.emit('ctrlc_event', {data: 'nothing'});
        return false;
    }); 
    socket.on('exec_response', function(msg) {
        $('#result').append(msg.data);
    }); 
    socket.on('exec_terminate', function(msg) {
        $('#result').append(msg.data);
    });
    socket.on('ctrlc_response', function(msg) {
        $('#result').append(msg.data);
    });
});
</script>
<style>
.result {
    overflow-y: scroll;
    width: 100%;
    height: 400px;
    border: 4px solid red;
}
.result pre {
    margin: 2px;
}
</style>
</head>
<body>
    <h3>Flask Example</h3>
    <form id="exec_form" method='POST' action='#' target='result'>        
        {{ exec_form.command.label }} 
        {{ exec_form.command(id="exec_input", placeholder="command", size=100) }} 
        {{ exec_form.submit(id="exec_btn") }}
    </form>
    <form id="ctrlc_form" method='POST' action='#' target='result'>
        {{ ctrlc_form.submit(id="ctrlc_btn") }}
    </form>
    <div id="result" class="result"></div>
</body>
</html>

该问题是由gevent-socketio服务器的结构方式引起的。当服务器从客户端(假设您的exec_event)接收事件时,它通过调用相应的处理程序来调度事件。然后,该线程将运行事件处理程序,并且只有在处理程序返回后,它才会返回以检查新的客户端发起的事件。

解决方案是将ctrlc_event发送到不同的通信通道上,而不是用于exec_event的通信通道。例如,在不同的命名空间上打开两个连接,然后每个连接将获得自己的服务器线程。

相关内容

  • 没有找到相关文章

最新更新