Golang TCP 代理io.Copy在使用 io.Multiwriter 时中断

如何解决Golang TCP 代理io.Copy在使用 io.Multiwriter 时中断

我目前正在尝试 Go,但无法弄清楚以下内容:

我试图通过我的应用程序代理 MySQL 连接,打算对来自客户端的请求和来自服务器的响应进行一些内省。我对 Redis 连接做了完全相同的事情,效果很好。

现在的问题是,如果我只是这样做

go io.Copy(serverConnection,clientConnection)
go io.Copy(clientConnection,serverConnection)

这很好用。我可以使用 mysql 客户端连接到我的代理并向 MySQL 服务器发出查询和接收响应。

但是,如果我尝试通过写入 io.MultiWriter 来“监视”连接,则连接会挂起,等待初始响应。

实现(简化)如下所示(并以这种方式工作):

package main

import (
    "fmt"
    "io"
    "log"
    "net"
)

func main () {
    port := 21001

    listenAddr,err := net.ResolveTCPAddr("tcp",fmt.Sprintf("127.0.0.1:%d",port))

    if err != nil {
        log.Fatalln(err)
    }

    targetAddr,"127.0.0.1:11001")

    if err != nil {
        log.Fatalln(err)
    }

    listener,err := net.ListenTCP("tcp",listenAddr)

    if err != nil {
        log.Fatalln(err)
    }

    for {
        clientConnection,err := listener.AcceptTCP()

        if err != nil {
            log.Fatalln(err)
        }

        go func () {
            serverConnection,err := net.DialTCP("tcp",nil,targetAddr)

            if err != nil {
                return
            }

            connectionDone := make(chan bool)

            go monitorClientConnection(clientConnection,serverConnection,connectionDone)
            go monitorServerConnection(serverConnection,clientConnection,connectionDone)

            _ = <-connectionDone
            _ = <-connectionDone

            fmt.Println("MySQL connection closed")
        }()
    }
}

func monitorClientConnection(clientConnection *net.TCPConn,serverConnection *net.TCPConn,connectionDone chan bool) {
    go func() {
        _,err := io.Copy(serverConnection,clientConnection)

        fmt.Println(err)

        _ = serverConnection.Close()

        connectionDone <- true
    }()
}

func monitorServerConnection(serverConnection *net.TCPConn,clientConnection *net.TCPConn,err := io.Copy(clientConnection,serverConnection)

        fmt.Println(err)

        _ = clientConnection.Close()

        connectionDone <- true
    }()
}

它开始失败的地方

如果我将监控功能的实现从(客户端和服务器相同,只是颠倒过来):

func monitorServerConnection(serverConnection *net.TCPConn,serverConnection)

        fmt.Println(err)

        _ = clientConnection.Close()

        connectionDone <- true
    }()
}

func monitorServerConnection(serverConnection *net.TCPConn,connectionDone chan bool) {
    _,w := io.Pipe() // The reader here would be used to consume the stream,left out for simplicity

    go func() {
        mw := io.MultiWriter(clientConnection,w)

        _,err := io.Copy(mw,serverConnection)

        fmt.Println(err)

        _ = clientConnection.Close()

        connectionDone <- true
    }()
}

它只是挂起。这与适用于 Redis 代理的方法完全相同(Redis 序列化协议是纯文本、换行符分隔的,可能是一个因素?)。

我想知道的

  1. 据我所知,为什么使用 io.MultiWriter 会导致流挂起?
  2. 有没有更好的方法来“窥探”这个流,或者将它放到一个单独的 goroutine 中,该 goroutine 可以在将流转发到上游服务器的同时读取/解析流?

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐


依赖报错 idea导入项目后依赖报错,解决方案:https://blog.csdn.net/weixin_42420249/article/details/81191861 依赖版本报错:更换其他版本 无法下载依赖可参考:https://blog.csdn.net/weixin_42628809/a
错误1:代码生成器依赖和mybatis依赖冲突 启动项目时报错如下 2021-12-03 13:33:33.927 ERROR 7228 [ main] o.s.b.d.LoggingFailureAnalysisReporter : *************************** APPL
错误1:gradle项目控制台输出为乱码 # 解决方案:https://blog.csdn.net/weixin_43501566/article/details/112482302 # 在gradle-wrapper.properties 添加以下内容 org.gradle.jvmargs=-Df
错误还原:在查询的过程中,传入的workType为0时,该条件不起作用 &lt;select id=&quot;xxx&quot;&gt; SELECT di.id, di.name, di.work_type, di.updated... &lt;where&gt; &lt;if test=&qu
报错如下,gcc版本太低 ^ server.c:5346:31: 错误:‘struct redisServer’没有名为‘server_cpulist’的成员 redisSetCpuAffinity(server.server_cpulist); ^ server.c: 在函数‘hasActiveC
解决方案1 1、改项目中.idea/workspace.xml配置文件,增加dynamic.classpath参数 2、搜索PropertiesComponent,添加如下 &lt;property name=&quot;dynamic.classpath&quot; value=&quot;tru
删除根组件app.vue中的默认代码后报错:Module Error (from ./node_modules/eslint-loader/index.js): 解决方案:关闭ESlint代码检测,在项目根目录创建vue.config.js,在文件中添加 module.exports = { lin
查看spark默认的python版本 [root@master day27]# pyspark /home/software/spark-2.3.4-bin-hadoop2.7/conf/spark-env.sh: line 2: /usr/local/hadoop/bin/hadoop: No s
使用本地python环境可以成功执行 import pandas as pd import matplotlib.pyplot as plt # 设置字体 plt.rcParams[&#39;font.sans-serif&#39;] = [&#39;SimHei&#39;] # 能正确显示负号 p
错误1:Request method ‘DELETE‘ not supported 错误还原:controller层有一个接口,访问该接口时报错:Request method ‘DELETE‘ not supported 错误原因:没有接收到前端传入的参数,修改为如下 参考 错误2:cannot r
错误1:启动docker镜像时报错:Error response from daemon: driver failed programming external connectivity on endpoint quirky_allen 解决方法:重启docker -&gt; systemctl r
错误1:private field ‘xxx‘ is never assigned 按Altʾnter快捷键,选择第2项 参考:https://blog.csdn.net/shi_hong_fei_hei/article/details/88814070 错误2:启动时报错,不能找到主启动类 #
报错如下,通过源不能下载,最后警告pip需升级版本 Requirement already satisfied: pip in c:\users\ychen\appdata\local\programs\python\python310\lib\site-packages (22.0.4) Coll
错误1:maven打包报错 错误还原:使用maven打包项目时报错如下 [ERROR] Failed to execute goal org.apache.maven.plugins:maven-resources-plugin:3.2.0:resources (default-resources)
错误1:服务调用时报错 服务消费者模块assess通过openFeign调用服务提供者模块hires 如下为服务提供者模块hires的控制层接口 @RestController @RequestMapping(&quot;/hires&quot;) public class FeignControl
错误1:运行项目后报如下错误 解决方案 报错2:Failed to execute goal org.apache.maven.plugins:maven-compiler-plugin:3.8.1:compile (default-compile) on project sb 解决方案:在pom.
参考 错误原因 过滤器或拦截器在生效时,redisTemplate还没有注入 解决方案:在注入容器时就生效 @Component //项目运行时就注入Spring容器 public class RedisBean { @Resource private RedisTemplate&lt;String
使用vite构建项目报错 C:\Users\ychen\work&gt;npm init @vitejs/app @vitejs/create-app is deprecated, use npm init vite instead C:\Users\ychen\AppData\Local\npm-