golang 转发tcp

时间：2023-05-17 01:40

Go语言中可以非常轻松地创建和管理TCP连接。本文将介绍如何使用Go语言创建TCP服务并转发TCP连接。

前置知识

在学习本文之前，需要掌握以下基本知识点：

Go语言基础知识
TCP协议的基本概念和使用

创建TCP服务

在Go语言中创建TCP服务非常简单。首先，我们需要导入net包和bufio包，然后使用net.Listen方法监听一个端口号：

package mainimport (    "bufio"    "fmt"    "net")func main() {    ln, err := net.Listen("tcp", ":8080")    if err != nil {        fmt.Println("Error listening:", err.Error())        return    }    defer ln.Close()    fmt.Println("Listening on port 8080...")}

在上面的例子中，我们监听了8080端口号，并在控制台输出Listening on port 8080...。

接受TCP连接

接下来，我们需要接受TCP连接。使用ln.Accept()方法可以在一个连接准备好之前阻塞程序。一旦有一个连接请求，我们就可以创建一个新协程，处理连接：

for {    conn, err := ln.Accept()        if err != nil {        fmt.Println("Error accepting connection:", err.Error())        return    }    fmt.Println("Connection accepted.")    // 处理连接    go handleConnection(conn)}func handleConnection(conn net.Conn) {    // 处理连接}

在上面的代码中，我们使用了一个无限循环来不断接受连接请求。当有连接请求时，我们会创建一个新的协程go handleConnection(conn)来处理连接。

处理TCP连接

接下来，我们需要对连接进行处理。对于每个客户端连接，我们需要将客户端发送的所有数据转发到另一个服务器上。可以使用bufio包来进行读取和写入。

func handleConnection(conn net.Conn) {    defer conn.Close()        remoteConn, err := net.Dial("tcp", "127.0.0.1:9090")    if err != nil {        fmt.Println("Error connecting to remote server:", err.Error())        return    }    defer remoteConn.Close()    go func() {        defer conn.Close()        defer remoteConn.Close()        for {            data, err := bufio.NewReader(conn).ReadString(')            if err != nil {                fmt.Println("Error reading from conn:", err.Error())                return            }            _, err = remoteConn.Write([]byte(data))            if err != nil {                fmt.Println("Error writing to remoteConn:", err.Error())                return            }        }    }()    // 从remoteConn读取数据并转发到conn    for {        data, err := bufio.NewReader(remoteConn).ReadString(')        if err != nil {            fmt.Println("Error reading from remoteConn:", err.Error())            return        }        _, err = conn.Write([]byte(data))        if err != nil {            fmt.Println("Error writing to conn:", err.Error())            return        }    }}

上面的代码中，我们首先使用net.Dial方法连接到另一个服务器，然后开启两个协程：一个协程读取客户端发来的数据并转发给另一个服务器，另一个协程从服务器读取数据并转发给客户端。

需要注意的是，在每个协程中都要使用defer conn.Close()和defer remoteConn.Close()来保证连接不会被意外关闭。

完整代码

以下是完整的TCP转发服务代码：

package mainimport (    "bufio"    "fmt"    "net")func main() {    ln, err := net.Listen("tcp", ":8080")    if err != nil {        fmt.Println("Error listening:", err.Error())        return    }    defer ln.Close()    fmt.Println("Listening on port 8080...")    for {        conn, err := ln.Accept()                if err != nil {            fmt.Println("Error accepting connection:", err.Error())            return        }        fmt.Println("Connection accepted.")        go handleConnection(conn)    }}func handleConnection(conn net.Conn) {    defer conn.Close()        remoteConn, err := net.Dial("tcp", "127.0.0.1:9090")    if err != nil {        fmt.Println("Error connecting to remote server:", err.Error())        return    }    defer remoteConn.Close()    go func() {        defer conn.Close()        defer remoteConn.Close()        for {            data, err := bufio.NewReader(conn).ReadString(')            if err != nil {                fmt.Println("Error reading from conn:", err.Error())                return            }            _, err = remoteConn.Write([]byte(data))            if err != nil {                fmt.Println("Error writing to remoteConn:", err.Error())                return            }        }    }()    for {        data, err := bufio.NewReader(remoteConn).ReadString(')        if err != nil {            fmt.Println("Error reading from remoteConn:", err.Error())            return        }        _, err = conn.Write([]byte(data))        if err != nil {            fmt.Println("Error writing to conn:", err.Error())            return        }    }}