2024年Python WebSocket服务器介绍_python websocket server

Python提供了一些库和框架来实现WebSocket服务器，其中比较常用的有websocket、websockets和Tornado等。下面我们将使用websockets库来实现一个简单的WebSocket服务器。

安装依赖库

首先，我们需要安装websockets库。打开终端，执行以下命令：

pip install websockets

编写WebSocket服务器代码

下面是一个简单的WebSocket服务器的示例代码：

import asyncio
import websockets

# 定义处理WebSocket连接的逻辑
async def handle\_websocket(websocket, path):
    print("WebSocket连接已建立")
    try:
        while True:
            # 接收客户端发送的消息
            message = await websocket.recv()
            print(f"接收到消息：{message}")

            # 发送消息给客户端
            response = f"服务器已收到消息：{message}"
            await websocket.send(response)
            print(f"已发送响应：{response}")
    except websockets.exceptions.ConnectionClosedOK:
        print("WebSocket连接已关闭")

# 启动WebSocket服务器
start_server = websockets.serve(handle_websocket, "localhost", 8000)

# 运行事件循环
asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()

在上述代码中，我们使用asyncio库来管理异步事件循环，使用websockets库来实现WebSocket服务器。handle_websocket函数定义了处理WebSocket连接的逻辑，当客户端和服务器建立连接后，会执行该函数。在函数中，我们可以通过await websocket.recv()来接收客户端发送的消息，通过await websocket.send(response)来发送消息给客户端。

启动WebSocket服务器

保存上述代码为websocket_server.py，在终端中执行以下命令启动WebSocket服务器：

python websocket_server.py

此时，WebSocket服务器会在本地的8000端口上监听连接。您可以使用任何支持WebSocket协议的客户端工具来连接服务器，例如浏览器的JavaScript代码或者WebSocket客户端库。

案例

案例一：实时聊天应用

在这个案例中，我们将使用Python和WebSocket服务器来构建一个简单的实时聊天应用。多个客户端可以通过WebSocket与服务器建立连接，并实时发送和接收消息。

代码如下：

import asyncio
import websockets

# 存储所有已连接的客户端
connected_clients = set()

# 定义处理WebSocket连接的逻辑
async def handle\_websocket(websocket, path):
    print("WebSocket连接已建立")
    # 将新的客户端添加到已连接的客户端集合中
    connected_clients.add(websocket)

    try:
        while True:
            # 接收客户端发送的消息
            message = await websocket.recv()
            print(f"接收到消息：{message}")



### 最后

> **🍅 硬核资料**：关注即可领取PPT模板、简历模板、行业经典书籍PDF。  
> **🍅 技术互助**：技术群大佬指点迷津，你的问题可能不是问题，求资源在群里喊一声。  
> **🍅 面试题库**：由技术群里的小伙伴们共同投稿，热乎的大厂面试真题，持续更新中。  
> **🍅 知识体系**：含编程语言、算法、大数据生态圈组件（Mysql、Hive、Spark、Flink）、数据仓库、Python、前端等等。




**网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。**

**[需要这份系统化学习资料的朋友，可以戳这里无偿获取](https://bbs.csdn.net/topics/618317507)**

**一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！**