golang 轉送tcp

Go語言中可以非常輕鬆地建立和管理TCP連線。本文將介紹如何使用Go語言建立TCP服務並轉送TCP連線。

1. 前置知識

在學習本文之前，需要掌握以下基本知識點：

• Go語言基礎知識
• #TCP協定的基本概念和使用

2. 建立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...。

3. 接受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)來處理連線。

4. 處理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()來確保連線不會被意外關閉。

5. 完整程式碼

以下是完整的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
        }
    }
}

6. 總結

Go語言使得創建和管理TCP連接變得輕而易舉。本文介紹如何使用Go語言建立TCP服務並進行轉發，希望對你有幫助。

以上是golang 轉送tcp的詳細內容。更多資訊請關注PHP中文網其他相關文章！