网络编程入门, 100行实现Websocket转TCP代理

于 2024-05-06 20:13:07 发布
阅读量556 收藏 18
点赞数 25
分类专栏： 实战项目知识 文章标签： 网络 websocket tcp/ip golang
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/ruzhila/article/details/138506525
版权

本文发表于入职啦(公众号: ruzhila) 大家可以访问入职啦学习更多的编程实战。

🎉 用Go实现Websocket转TCP代理，浏览器可以访问TCP Socket，网络编程入门 实用项目 👏 🎉

项目地址

代码已经开源， websockify-go 👏 欢迎Star

代码运行效果：

Websockify

所有的项目都在github上开源：100-line-code 欢迎Star 👏

用100行代码的不同语言（Java、Python、Go、Javascript、Rust）实现项目，通过讲解项目的实现，帮助大家学习编程

会定期在群里分享最新的项目实战代码，包括不同语言的实现

老师还会详细讲解代码优化的思路，扫码加入实战群：
入群学习

为什么需要Websocket转TCP

Websocket是一种全双工通信协议，可以在浏览器和服务器之间建立持久连接，用于实时通信。但是Websocket只支持文本和二进制数据，无法直接访问TCP Socket。
如果要在网页上访问TCP Socket，就需要一个代理服务器，将Websocket请求转发到TCP Socket， 这就是Websocket转TCP代理:

Websocket <----> Websocket转TCP代理 <----> TCP Socket

典型的场景：

  • 在浏览器上访问VNC远程桌面
  • 在浏览器上访问SSH终端
  • 在浏览器上访问数据库管理工具
  • 在浏览器上访问自定义TCP服务, 比如游戏服务器或者IM服务器

novnc/websockify是一个非常流行的Websocket转TCP代理，但是它是用Python实现的，性能不够好,代码也很复杂

用Go实现一个Websocket转TCP代理，性能更好，更稳定

Go实现Websocket转TCP代理

Go特别适合网络编程，标准库提供了丰富的网络库，不仅性能好，而且代码简洁，只需要一个可执行文件就可以运行，不需要依赖其他库
这次要通过100行代码实现一个Websocket转TCP代理：

  • 要支持SSL
  • 要支持自定义URL路径
  • 要支持自定义目标地址
  • 完整的单元测试和标准的服务程序架构

代码

在这里插入图片描述

代码解析

代码结构

  • cmd/main.go 是程序的入口，解析命令行参数，启动服务
  • websockify.go 是Websocket转TCP代理的核心代码
  • websockify_test.go 是单元测试代码
  • Dockerfile 是构建Docker镜像的文件，方便部署可以直接上生产系统

启动一个HTTP服务器

在这里插入图片描述

  • 82行： 监听一个HTTP端口
  • 89-95行： 如果是443并且有SSL证书，就启动一个HTTPS服务器
  • 86行： 利用标准库创建一个能自定义处理的HTTP Server

处理Websocket请求

采用了gorilla/websocket库来处理Websocket请求，这个库是Go语言中最流行的Websocket库

所有的业务逻辑都在ServeHTTP函数：
在这里插入图片描述

  • 31行： 升级HTTP连接为Websocket连接, Websocket协议是基于GET这个HTTP请求升级为Websocket协议
  • 36行： 利用defer关键字，当链接断开的时候，关闭TCP连接，确保不会泄露资源
  • 44-59行： 读取Websocket数据，转发到TCP Socket
  • 60-73行： 读取TCP Socket数据，转发到Websocket
一个链接启动了两个协程

其实从ServerHTTP这个函数开始，系统库就为每个请求创建了一个协程，这个协程负责处理这个请求
在这个函数里面又启动了两个协程：

  • 一个负责读取Websocket数据
  • 一个负责读取TCP Socket数据

这样就实现了一个链接同时处理两个方向的数据流

  • 43和75行 创建了能接受err的channel，当处理数据从协程出现错误的时候，就会发送一个错误到这个channel，然后当前的主协程就会退出关闭链接

单元测试

golang的单元测试非常的方便，只需要在文件名后面加上_test.go，然后在函数名前面加上Test就可以了
在这里插入图片描述

单元测试的逻辑：

  • 2-28行 创建一个测试用的TCP服务器，写了一个Echo服务器，就是把接收到的数据原样返回
  • 31-41行 创建一个Websockify代理服务器
  • 43-63行 启动一个Websocket客户端测试链接是否能正常工作

标准的服务器程序架构

这个项目提供了Dockerfile，现在Docker是最主要的线上程序交付方式，强烈建议用Docker部署服务
基于flag这个库，来处理命令行参数，这是Go语言标准库提供的命令行参数解析库

程序能支持的命令行参数来控制服务的启动

总结

通过Websockify这个项目，学习了Go语言的网络编程，Websocket是一个非常重要的协议，可以用于实时通信，实现一个Websockify的转发代理，对于内部项目的开发非常有用

Docker也是非常重要的技术，我们提供了Dockerfile，可以直接部署到生产环境

过去的演示项目都没提供单元测试，这里提供了一个非常实用的单元测试流程，大家可以参考即便是复杂的流程也是可以通过单元测试来验证的

交流

构建一个100行代码项目的实战群，大家可以扫码加入，一起学习编程

入群学习

也可以访问入职啦学习更多的编程实战

所有的代码都在github上开源：100-line-code 欢迎Star 👏

入职啦
关注
  • 25
    点赞
  • 踩
  • 18
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
专栏目录
利用websockify将websocket通信转换成tcp
AlbertS Home of Technology
12-13 1464
目前遇到一个问题，原本的服务都是利用tcp通信的，但是某些平台只支持websocket，是不是要从头实现一套websocket网络收发逻辑呢？短时间内有没有替代方案呢？在经过一番寻找之后发现了websockify这个神器，可以将websocket协议转换成tcp协议，从而实现不修改应用服务就支持tcp协议的目的，接下来就说说它怎么用，以及探索的过程...
WebSocket转发TCP
idwtwt的专栏
08-27 2398
1 需求背景 某些特殊原因，不能访问某些内部服务器，或服务器只开放HTTP或HTTPS端口，为了连接ssh等服务（其实有安全的方法:跳板机，这里不讲安全） 2 方案 通过WebSocket转发TCP协议数据 [tcp 服务,如sshd] | | <= TCP | [zserver 服务端] || || <= WebSocket || [nginx等] || || <= WebSocket || [ztun客户端] | | <= TCP | [tcp
将websocket通信端口代理到TCP通信端口的方法记录
weixiao的博客
06-02 3064
１、代理服务基于go语言实现 ２、代理服务源码 package main import ( "flag" "fmt" "log" "net" "net/http" "os" "runtime" "github.com/gorilla/websocket" ) var g_tcpAddress string var upgrader = websocket.Upgrader{ CheckOrigin: func(r *http.Request) bool { ..
tcp支持浏览器websocket协议
qq_46118239的博客
03-09 4507
推荐一个零声学院免费公开课程，个人觉得老师讲得不错，分享给大家：Linux，Nginx，ZeroMQ，MySQL，Redis，fastdfs，MongoDB，ZK，流媒体，CDN，P2P，K8S，Docker，TCP/IP，协程，DPDK等技术内容，立即学习 tcp支持浏览器websocket协议 一个io它是怎么一种情况，一个客户端连接一个服务器，一个客户端一个连接，大家时刻在做服务器，都是时刻抓住这样一个点， 就是说一个客户端在服务端会有一个网络io，一个客户端在服务端会有一个网络io，之前用epoll
golang代理websocket请求
FlayHigherGT的博客
10-23 4409
基础： golang websocket github地址: https://github.com/gorilla/websocket/ github库官方文档: https://godoc.org/github.com/gorilla/websocket golang websocket运行机制以及原理： https://www.jianshu.com/p/65ef71ddb910 golang websocket在线测试地址： http://www.websocket-test.com/
编程知识+GO语言开发+GO语言基础课程+系列课程
04-04
网络编程：通过实例学习Go的HTTP协议编程、RESTful API开发、WebSocket通信和TCP/UDP编程。 测试与调试：教授如何编写单元测试，进行基准测试和覆盖率分析，以及使用Go的调试工具。 数据库编程：学习使用Go连接和...
python入门到高级全栈工程师培训 第3期 附课件代码
06-07
python入门到高级全栈工程师培训视频学习资料；本资料仅用于学习，请查看后24小时之内删除。 【课程内容】 第1章 01 计算机发展史 02 计算机系统 03 小结 04 数据的概念 05 进制转换 06 原码补码反码 07 物理层和...
node.js中的socket.io入门实例
12-12
关于websocket等反向ajax技术介绍 ...反向Ajax技术主要有两点内容：一是服务器端保持住TCP连接直到其有数据发送给客户端（可以使用循环和睡眠实现），二是客户端js编程技巧。 websocket是html5的规范，也属
java开源包1
06-28
WebSocket4J 是一个用 Java 实现的 WebSocket 协议的类库，可使用 Java 来构建交互式 Web 应用。WebSocket4J 并未实现客户端通讯协议，所以不能用它来连接 WebSocket 服务器。 Struts验证码插件 JCaptcha4Struts2 ...
netty权威指南第2版
12-27
网络编程分享 -> 最小资源分1分，以下是详细章节内容，欢迎评价和收藏 章节内容： 基础篇走进Java NIO 第1章 Java的I/O演进之路 第2章 NIO入门 第3章 Netty入门应用 第4章 TCP粘包/拆包问题解析之道 第5章 分隔符...
websocket-to-tcp:Websocket到TCP隧道
05-27
Web套接字到TCP 该实用程序创建一个接受WebSocket连接并将其转发到TCP套接字的服务器。 安装 npm i -g websocket-to-tcp 这会将实用程序安装在全局寄存器中，可以使用wstcp在命令行界面中调用该实用程序。 用法 wstcp -t tcpaddress -p tcpport -w wsport -n name tcpaddress是远程TCP连接的地址tcpport是远程TCP连接的端口号wsport websocket侦听本地端口号名称侦听websocket连接的名称 例子 wstcp -t bbs.sblendorio.eu -p 6510 -w 8080 -n bbs 创建一个本地服务器，该服务器在端口8080上接受WebSocket连接，并将其转发到bbs.sblendorio.eu:6510 。 WebSocket连接的名称是bbs 。
Python-Websockify是一个WebSocket至TCP的代理桥梁
08-10
Websockify是一个WebSocket至TCP的代理/桥梁。实现让让浏览器可以连接至任何应用程序/服务器/服务。采用Python, C, Node.js 和Ruby。
wspipe:桥接 websocket 到 tcp 套接字
06-26
WebSocket 到 TCP 管道 (wspipe) 连接 WebSocket 和 TCP（通过 WS 桥接 TCP 套接字）。 用法 首先创建一个WebSocket Server连接 var WebSocketServer = require('ws').Server; var server = new WebSocketServer({port:80, path:'/mysocket'}); 让 wspipe 将 WS 桥接到 TCP var WebSocketPipe = require('wspipe'); var debug = true; var tcpServerHost = 'localhost'; var tcpServerPort = 3474; var wspipe = new WebSocketPipe(server, debug, tcpServerHo
利用WebSocket技术实现实时通信
最新发布
weixin_53198972的博客
04-27 1731
WebSocket是一种在单个TCP连接上进行全双工通信的协议。为什么不用HTTP协议而是用WebSocket呢？因为HTTP 协议有一个缺陷：通信只能由客户端发起，而WebSocket使得客户端和服务器之间的数据交换变得更加简单，允许服务端主动向客户端推送数据。在WebSocket API中，浏览器和服务器只需要完成一次握手，两者之间就直接可以创建持久性的连接，并进行双向数据传输。其实一句话就可以做出概括：它就是一种网络通信协议，很多高级功能都需要它。
搞明白 WebSocket 与 TCP/IP
油墨香^-^的博客
06-22 2968
我们用websocket来研究一下TCP/IP协议的一些特性，在《https 连接的前几毫秒发生了什么》一文里我们已经研究了https建立的过程。 在《https 连接的前几毫秒发生了什么》一文里是用的wireshark的抓包工具，这一篇将用tcpdump命令行工具。 01、tcpdump Linux系的系统有一个很好用的抓包工具，叫tcpdump，可以用来抓取网络上的tcp包，例如我要抓取8080端口的包，可以执行以下命令： sudo tcpdump port 8080 –n
WebSocket主动通过线程推动TCP请求
bobo_supper的博客
10-11 792
业务需求备注：建立websocket连接时，开启定时任务，线程通过tcp请求获取返回内容，将返回内容定时推送，断开websocket连接时，终止定时任务，同时停止推送，释放资源。
基于Golang和WebSocket打造自已的反向代理
奋斗、
08-23 1609
目前已经有许多可用的代理服务了，比如ngrok和localtunnel,但ngrok有个缺点就是提供的域名只能用几个小时，然后需要新生成新的域名，如果想要固定域名就要花钱，但我们自已实现的代理可以用一个固定的域名，如果给前端同学来调试的话，不用改来改去，很方便。当我们在开发的时候，有时想要把自已开发的接口暴露给其他开发者或者第三方的服务，方便我们调试和排查问题，那就需要某种机制把我们本地的服务接口暴露到互联网上，本文将要介绍如何通过Golang和WebSocket来实现这一功能。## ggrok简介。
Nginx 解决WebSocket TCP 转发问题
Leon_Jinhai_Sun的博客
07-14 725
#user nobody; worker_processes 1; #error_log logs/error.log; #error_log logs/error.log notice; #error_log logs/error.log info; #pid logs/nginx.pid; events { worker_connections 1024; } http { include mime.types; defau.
springboot 实现websocket反向代理
09-09
要实现WebSocket的反向代理，可以使用Spring Boot提供的WebSocketStompClient来实现。首先，需要添加以下依赖到你的项目中： ```xml <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-websocket</artifactId> </dependency> ``` 然后，你需要配置你的WebSocket拦截器，可以通过添加`@ServerEndpoint`注解和设置configurator来获取Filter中的数据。这里需要注意的是，要加上断点注解：`@ServerEndpoint(value = "/websocket", configurator = WebSocketConfigurator.class)`。这样，你就可以通过session来获取Filter中的数据了。 接下来是分布式部署和共享的问题。在分布式部署中，WebSocket的共享问题可以通过使用消息代理来解决。你可以使用Spring的消息中间件，如RabbitMQ或ActiveMQ，来实现WebSocket的消息代理。通过配置消息代理，你可以确保WebSocket的消息在多个节点之间共享。 最后，如果你遇到了无法注入对象的问题，可以检查你的WebSocket配置是否正确，并确保你的对象被正确注入。你可以使用`@Autowired`注解来注入对象。如果仍然无法注入对象，可能需要检查你的配置和代码逻辑是否正确。 总结起来，要实现Spring Boot的WebSocket反向代理，你需要添加必要的依赖和配置WebSocket拦截器。在分布式部署中，你可以使用消息代理来解决WebSocket的共享问题。如果遇到无法注入对象的问题，可以检查配置和代码逻辑是否正确。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值