golang 转发tcp-Golang-PHP中文网

golang 转发tcp

WBOY

发布： 2023-05-16 14:45:37

原创

652人浏览过

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

前置知识

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

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

创建TCP服务

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

package main

import (
    "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()方法可以在一个连接准备好之前阻塞程序。一旦有一个连接请求，我们就可以创建一个新协程，处理连接：

立即学习“go语言免费学习笔记（深入）”；

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 main

import (
    "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
        }
    }
}

登录后复制