使用Python创建websocket服务端并给出不同客户端的请求

下面是关于使用Python创建WebSocket服务端以及处理来自不同客户端请求的完整攻略。

简介

WebSocket是一种新兴的网络通信协议，基于HTTP协议，与HTTP协议一样可以在Web浏览器和服务器之间进行双向通信，使用websocket可以让客户端和服务端实时通信，达到更好的用户体验。

创建WebSocket服务端

要使用Python创建WebSocket服务端，需要安装一个websocket库，这里我们使用的是websocket库。

pip install websocket

创建WebSocket服务端十分简单，只需要使用WebSocketServer类中的websocket_server方法，示例代码如下：

import websocket


# 定义处理WebSocket请求的函数
def on_message(ws, message):
    # 处理收到的消息，这里简单的打印出来
    print(message)


# 创建WebSocket服务端，并加上处理函数
server = websocket.WebSocketServer('localhost', 8888, handle_message=on_message)
server.run_forever()

这样，我们就创建了一个WebSocket服务端，并打印了客户端发送的消息。

处理不同客户端请求

WebSocket服务端通过消息来识别不同客户端的请求，并根据消息内容进行处理。

下面给出两个处理不同客户端请求的示例。

示例1：实现简单的聊天室

聊天室的功能需要能够实现多个客户端之间相互发送消息，聊天室的服务端会把一个客户端发送的消息转发给其它客户端。

我们通过创建一个客户端列表来保存连接的客户端，当有客户端连接或发送消息时，遍历客户端列表进行消息转发。

import websocket


# 客户端列表
clients = []


# 处理WebSocket请求的函数
def on_message(ws, message):
    # 遍历客户端列表，将消息转发给其它客户端
    for client in clients:
        if client != ws:  # 不发送给消息的原始发送方
            client.send(message)


# 处理新建WebSocket连接的函数
def on_open(ws):
    # 将新客户端加入客户端列表
    clients.append(ws)


# 处理WebSocket关闭连接的函数
def on_close(ws):
    # 移除关闭的客户端
    clients.remove(ws)


# 创建WebSocket服务端，并加上处理函数
server = websocket.WebSocketServer('localhost', 8888, handle_message=on_message, handle_new_connection=on_open, handle_close_connection=on_close)
server.run_forever()

示例2：实现WebSocket服务端向客户端发送消息

有时WebSocket服务端需要主动向客户端发送消息，例如服务端推送新的消息或者更新到客户端。

在WebSocket服务端，可以通过客户端实例的send方法来向其它客户端发送消息。

import websocket


# 客户端列表
clients = []


# 处理WebSocket请求的函数
def on_message(ws, message):
    # 处理收到的消息，这里简单的打印出来
    print('received', message)


# 处理新建WebSocket连接的函数
def on_open(ws):
    # 将新客户端加入客户端列表
    clients.append(ws)

    # 向新客户端发送消息
    ws.send('Welcome to the server!')


# 处理WebSocket关闭连接的函数
def on_close(ws):
    # 移除关闭的客户端
    clients.remove(ws)


# 创建WebSocket服务端，并加上处理函数
server = websocket.WebSocketServer('localhost', 8888, handle_message=on_message, handle_new_connection=on_open, handle_close_connection=on_close)
server.run_forever()

这样，我们就完成了使用Python创建WebSocket服务端，并处理不同客户端请求的攻略。