阅读量:0
在Golang中实现流量转发功能通常使用net包中的TCP或者UDP相关功能来实现。以下是一个简单的示例代码，实现了一个简单的TCP流量转发功能：
package main import ( "fmt" "net" "os" ) func main() { // 监听端口 ln, err := net.Listen("tcp", "localhost:8080") if err != nil { fmt.Println("Error listening:", err) os.Exit(1) } defer ln.Close() fmt.Println("Listening on localhost:8080") // 接受客户端连接 for { conn, err := ln.Accept() if err != nil { fmt.Println("Error accepting connection:", err) os.Exit(1) } fmt.Println("Accepted connection from", conn.RemoteAddr()) // 将数据转发到目标地址 go func(conn net.Conn) { targetAddr := "localhost:9090" targetConn, err := net.Dial("tcp", targetAddr) if err != nil { fmt.Println("Error connecting to target:", err) conn.Close() return } defer targetConn.Close() // 进行数据转发 go func() { _, err := io.Copy(targetConn, conn) if err != nil { fmt.Println("Error copying data to target:", err) } }() _, err = io.Copy(conn, targetConn) if err != nil { fmt.Println("Error copying data to client:", err) } conn.Close() }(conn) } } 以上代码中，我们首先监听localhost的8080端口，然后接受客户端连接。每当有新的客户端连接时，我们会创建一个新的goroutine来处理这个连接，将客户端的数据转发到目标地址localhost:9090上，同时也将目标地址的数据转发回客户端。最后，我们关闭连接。
当然，这只是一个简单的示例，实际应用中可能需要考虑更多的错误处理，以及一些优化和安全性方面的考虑。
