websocket实现群聊

server



# @File: 群聊
from flask import Flask, render_template, request
from geventwebsocket.handler import WebSocketHandler
from gevent.pywsgi import WSGIServer
from geventwebsocket.websocket import WebSocket

app = Flask(__name__)
user_socket_list = []


@app.route('/ws')
def ws():
    # 一个用户的长链接wsgi.websocket
    user_socket = request.environ.get("wsgi.websocket")  # type:WebSocket
    if user_socket:
        # 将每个链接存入列表中
        user_socket_list.append(user_socket)
    print(len(user_socket_list), user_socket_list)
    while 1:
        msg = user_socket.receive()
        print(msg)
        for usocket in user_socket_list:
            if user_socket == usocket:
                continue
            try:
                usocket.send(msg)
            except:
                continue


@app.route('/')
def index():
    return render_template("many_person.html")


if __name__ == '__main__':
    http_server = WSGIServer(("0.0.0.0", 7000), app, handler_class=WebSocketHandler)
    http_server.serve_forever()

many_person.html



<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Title</title>
</head>
<body>

<p>发送内容：<input type="text" id="message"><button onclick="send_msg()">发送</button></p>
<div id="msg_list" style=" 500px">



</div>

</body>
<script type="application/javascript">

    var ws = new WebSocket("ws://192.168.12.41:7000/ws");
    ws.onmessage = function (ws_status) {
        console.log(ws_status.data);
        var ptag = document.createElement("p");
        ptag.innerText = ws_status.data;
        document.getElementById("msg_list").appendChild(ptag)
    };
    function send_msg() {
        var msg = document.getElementById("message").value;
        var ptag = document.createElement("p");
        ptag.style.cssText = "text-align: right";
        ptag.innerText = msg;
        document.getElementById("msg_list").appendChild(ptag);
        ws.send(msg);
    }
</script>
</html>