6. Python使用Asyncio开发TCP服务器简单案例

1. 说明

在Python中开发TCP/IP服务器有两种方式,一种使用Socket,需要在py文件中引入对应的socket包,这种方式只能执行单项任务;另一种方式使用Asyncio异步编程,可以一次创建多个服务器执行不同的任务。

2. 接口说明

python 复制代码
1. asyncio.start_server(callback,ip,port) # 生成一个异步服务器,调用后会自动传给回调函数两个参数writer和reader
	callback: 回调函数,异步操作均在此函数中执行
	ip: ip地址,可以为空,表明接受任意ip的连接
	port: 端口号
2. serve_forever() # 循环执行,接受连接,直到协程被取消
3. asyncio.wait_for(func,time) # 等待执行
	func: 等待执行结束的函数
	time: 等待时间,可以为None表示无限等待
4. writer.write() # 向客户端发送数据
5. writer.drain() # 发送数据后用于清空套接字,需要和writer.write()一起使用
6. reader.read() # 从客户端读取数据
7. writer.close() # 关闭套接字(同时会关闭协程和当前的服务器)
8. writer.wait_closed() # 等待套接字完全关闭,需要和writer.close()一起使用
9. writer.is_closing() # 判断连接是否是断开状态

3. 简单案例

创建一个tcp服务器,并实现数据的接受和发送

python 复制代码
from datetime import datetime
import asyncio

# 服务器的回调函数
async def script_handle(reader, writer): # reader和writer参数是asyncio.start_server生成异步服务器后自动传入进来的
    while True: # 循环接受数据,直到套接字关闭
        # wait_for等待读取数据,第二个参数为等待时间(None表示无限等待)
        data = await asyncio.wait_for(reader.read(2**10), None)
        if not data:
            print('script client disconnected')
            writer.close() # 关闭套接字
            await writer.wait_closed() # 等待套接字完全关闭
            return
        print("received data: ", data.decode())
		writer.write(data.encode()) # 发送数据
		await writer.drain() # 发送数据后,清空套接字

# 主函数
async def main():
    # 生成一个服务器
    server = await asyncio.start_server(script_handle,
                                        host='',
                                        port=8888)
    # 获取请求连接的客户端信息
    addr = server.sockets[0].getsockname()
    print(f'Serving on {addr}')
    # 处理多个请求,永远执行着调用
    async with server:
        await server.serve_forever()
   
if __name__ == '__main__':
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        print(datetime.now().strftime('%Y/%m/%d %H:%M:%S.%f'),
              'script server exit by key')

运行此脚本即可接受客户端的正常连接和信息传输。

相关推荐
Doris_LMS7 小时前
Git的强软硬回退(三)
运维·服务器·数据库·git·idea
(Charon)7 小时前
基于 epoll 的高并发服务器原理与实现(对比 select 和 poll)
运维·服务器
THMAIL8 小时前
机器学习从入门到精通 - 机器学习调参终极手册:网格搜索、贝叶斯优化实战
人工智能·python·算法·机器学习·支持向量机·数据挖掘·逻辑回归
ERP老兵-冷溪虎山8 小时前
Python/JS/Go/Java同步学习(第三篇)四语言“切片“对照表: 财务“小南“纸切片术切凭证到崩溃(附源码/截图/参数表/避坑指南/老板沉默术)
java·javascript·python·golang·中医编程·四语言同步学习·职场生存指南
科技树支点8 小时前
无GC的Java创新设计思路:作用域引用式自动内存管理
java·python·go·web·编程语言·编译器
THMAIL9 小时前
机器学习从入门到精通 - 集成学习核武器:随机森林与XGBoost工业级应用
人工智能·python·算法·随机森林·机器学习·集成学习·sklearn
程序员miki10 小时前
Pytorch的CUDA版本安装使用教程
人工智能·pytorch·python
小冷coding10 小时前
随时随地写代码:Jupyter Notebook+cpolar让远程开发像在本地一样流畅
ide·python·jupyter
偷心伊普西隆11 小时前
Pandas DataFrame 指南
python·数据分析·pandas
Yokon_D11 小时前
Pycharm终端pip install的包都在C:\Users\\AppData\Roaming\Python\解决办法
python·pycharm·pip