想把流量发个多个后端,每个后端都可以收到内容,但我写的,每次都只有一个后端收到内容,随机的
```
package main
import (
"flag"
"fmt"
"io"
"net"
"strings"
)
var tcp_conn []net.Conn
func main() {
var address *string
var dst *string
address = flag.String("listen_address", "127.0.0.1:8080", "listen address")
dst = flag.String("target_address", "172.16.5.9:18061,172.16.5.9:18060", "目标地址,多个地址逗号分割")
flag.Parse()
listener, err := net.Listen("tcp", *address)
if err != nil {
fmt.Println(err)
}
ips := strings.Split(*dst, ",")
if len(ips) == 0 {
fmt.Println("目标地址不能为空")
return
}
for _, ip := range ips {
conn, err := net.Dial("tcp", ip)
if err != nil {
fmt.Println(err)
continue
}
tcp_conn = append(tcp_conn, conn)
}
for {
conn, err := listener.Accept()
if err != nil {
continue
}
go handler(conn)
}
}
func handler(conn net.Conn) {
// defer conn.Close()
for _, dst_conn := range tcp_conn {
go io.Copy(dst_conn, conn)
go io.Copy(conn, dst_conn)
}
}
```
该如何改呢,哪位大佬帮忙看看
楼主这个场景有点像流量镜像,就是把流量拷贝N多份,一份照常给业务服务器。其他的发送给流量审计服务器或者测试环境服务器。
需要有一些注意:
1. 如果只读取第一个连接的数据回复,其他的不管,`有可能`会造其他连接的消息积压。
虽然你并不关心其他连接回复的消息,但是应该把回复的消息读出来,比如读出来丢弃掉:`io.Copy(io.Discard, otherConn)`
2. 最好不要用 `io.MultiWriter()`,因为有一个写入出错就会导致 `io.MultiWriter()` 报错, 因为后端某个连接可以能会异常或断开,这样一个异常,会影响所有的 Writer 写入
#6
更多评论
某个协程io.Copy之后相当于读完了流,另外的协程当然读不到数据了,你需要修改代码,每次读取[]byte,然后再每个连接Write([]byte)
#1