讲师中心 微信公众号
AI工具推荐 视频效率加速

Python 中实现 TCP 连接双向数据透传的等效 io.Copy 方法

风浩吖_5347

风浩吖_5347

发布时间:2026-09-14 09:34:18

|

871人浏览过

|

来源于php中文网

原创

Python 中实现 TCP 连接双向数据透传的等效 io.Copy 方法

在 Python(含 2.6)中,可通过自定义循环读写 + 多线程方式模拟 Go 的 io.Copy,实现两个 socket 间无缓冲、低开销的双向数据转发。

在 python(含 2.6)中,可通过自定义循环读写 + 多线程方式模拟 go 的 `io.copy`,实现两个 socket 间无缓冲、低开销的双向数据转发。

Go 语言中 io.Copy(dst, src) 提供了简洁、阻塞式的数据流复制能力,常用于代理、隧道等场景。Python 标准库虽无完全等价的单函数,但可通过轻量级循环 + socket.recv()/socket.send() 配合多线程,精准复现其行为。

以下是一个兼容 Python 2.6 的可靠实现:

import threading

BUFFER_SIZE = 8192

def copy_proxy(src, dst):
    """将 src socket 的数据持续读取并写入 dst socket,遇 EOF 或异常时安全关闭双方连接"""
    try:
        while True:
            data = src.recv(BUFFER_SIZE)
            if not data:  # 对端关闭连接(EOF)
                break
            dst.sendall(data)  # 使用 sendall 确保全部数据发出
    except (IOError, OSError, socket.error):
        pass  # 忽略网络中断、连接重置等常见错误
    finally:
        try:
            src.close()
        except:
            pass
        try:
            dst.close()
        except:
            pass

使用时需启动两个并发线程,分别处理正向与反向数据流:

Shadows Python Sensei
Shadows Python Sensei

Python 最佳实践助手——代码规范、设计模式、性能优化、测试与类型注解。适用于编写或审查 Python 代码。

下载
import socket

# 示例:建立两个已连接的 socket(如 client_conn 和 upstream_conn)
# conn1 = accepted client socket
# conn2 = connected upstream server socket

t1 = threading.Thread(target=copy_proxy, args=(conn1, conn2))
t2 = threading.Thread(target=copy_proxy, args=(conn2, conn1))
t1.daemon = True  # 设为守护线程,避免主程序退出时残留
t2.daemon = True
t1.start()
t2.start()

# 主线程可等待或继续处理其他逻辑(如日志、监控)
t1.join()
t2.join()

⚠️ 关键注意事项

立即学习Python免费学习笔记(深入)”;

  • 必须使用 sendall() 而非 send()send() 可能仅发送部分数据,而 sendall() 会重试直至全部写入或出错,更贴近 io.Copy 的语义;
  • 显式关闭逻辑置于 finally 块中:确保无论是否异常,连接都能被释放,防止文件描述符泄漏;
  • 线程设为 daemon=True:避免代理进程因后台线程未退出而挂起;
  • BUFFER_SIZE 建议设为 4096–8192 字节:兼顾吞吐与内存占用,在 Python 2.6 中尤其重要;
  • 不依赖第三方库:该方案纯标准库实现,完全兼容 Python 2.6 及以上版本。

此模式已广泛应用于轻量 TCP 代理、调试中间件和协议桥接工具中,兼具简洁性、可控性与可维护性。

热门AI工具

更多
讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

LibLibAI
LibLibAI Hot

一款AI视频创作工具,主要用于国内领先的AI创意平台,以海量模型、低门槛操作与“创作-分享-商业化”生态,让小白与专业创作者都能高效实现图文乃至视频创意表达,适合需要提升相关任务效率的用户。

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

1571

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

3724

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1569

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

21297

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2627

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2687

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

1083

2023.08.11

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

576

2023.08.10

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

20

2026.09.23

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn