从Go语言的*net.TCPConn中高效获取远程IP地址

tcpconn.RemoteAddr().(*net.TCPAddr).IP

package main

import (
    "fmt"
    "log"
    "net"
    "time"
)

func main() {
    // 1. 监听TCP端口
    addr, err := net.ResolveTCPAddr("tcp", "localhost:8080")
    if err != nil {
        log.Fatalf("无法解析TCP地址: %v", err)
    }

    listener, err := net.ListenTCP("tcp", addr)
    if err != nil {
        log.Fatalf("无法监听TCP端口: %v", err)
    }
    defer listener.Close()
    fmt.Printf("服务器正在监听 %s\n", listener.Addr().String())

    // 2. 启动一个goroutine模拟客户端连接
    go func() {
        time.Sleep(1 * time.Second) // 等待服务器启动
        conn, err := net.Dial("tcp", "localhost:8080")
        if err != nil {
            log.Printf("客户端连接失败: %v", err)
            return
        }
        defer conn.Close()
        fmt.Println("客户端已连接到服务器")
        _, _ = conn.Write([]byte("Hello from client!"))
    }()

    // 3. 接受传入连接
    conn, err := listener.AcceptTCP()
    if err != nil {
        log.Fatalf("接受连接失败: %v", err)
    }
    defer conn.Close()

    fmt.Printf("已接受来自 %s 的连接\n", conn.RemoteAddr().String())

    // 4. 从 *net.TCPConn 中提取IP地址
    remoteAddr := conn.RemoteAddr()
    tcpAddr, ok := remoteAddr.(*net.TCPAddr) // 进行类型断言，并检查是否成功
    if !ok {
        log.Fatalf("RemoteAddr() 返回的不是 *net.TCPAddr 类型")
    }

    ipAddress := tcpAddr.IP
    fmt.Printf("提取到的远程IP地址 (net.IP类型): %s\n", ipAddress.String())

    // 如果需要将 net.IP 转换为字符串
    ipString := ipAddress.String()
    fmt.Printf("IP地址的字符串表示: %s\n", ipString)

    // 读取客户端发送的数据 (可选)
    buffer := make([]byte, 1024)
    n, err := conn.Read(buffer)
    if err != nil {
        log.Printf("读取数据失败: %v", err)
        return
    }
    fmt.Printf("收到客户端数据: %s\n", string(buffer[:n]))
}