如何从net.UDPAddr或net.TCPAddr类型的net.Addr中获取IP和端口

嘿，我完全懂你为啥反感用反射来判断*net.UDPAddr和*net.TCPAddr——这种写法不仅啰嗦，还不够符合Go的惯用风格，其实咱们直接用类型断言的switch分支就能搞定，根本不需要依赖reflect包！

改进后的代码实现 #

// 直接用类型断言的switch，简洁又高效
switch remoteAddr := remoteAddr.(type) {
case *net.UDPAddr:
    // 断言本地地址为UDP类型（如果你的服务器逻辑保证类型匹配，这一步是安全的）
    localUDPAddr, ok := localAddr.(*net.UDPAddr)
    if !ok {
        // 可选：处理类型不匹配的异常情况，比如打日志
        return
    }
    p.SrcIP = remoteAddr.IP.String()
    p.SrcPort = uint(remoteAddr.Port)
    p.DstPort = uint(localUDPAddr.Port)
case *net.TCPAddr:
    localTCPAddr, ok := localAddr.(*net.TCPAddr)
    if !ok {
        return
    }
    p.SrcIP = remoteAddr.IP.String()
    p.SrcPort = uint(remoteAddr.Port)
    p.DstPort = uint(localTCPAddr.Port)
default:
    // 处理未知的地址类型，避免panic
    log.Printf("unsupported address type: %T", remoteAddr)
}

为啥这个写法更好？ #

• 更符合Go惯用风格：类型断言是Go处理接口类型分支的标准方式，比反射更直观，团队里的其他Go开发者一眼就能看懂
• 性能更优：反射会带来额外的性能开销，直接类型断言是编译期优化的操作，效率更高
• 鲁棒性更强：可以通过ok标志位避免类型断言失败导致的panic，还能在default分支处理意外的地址类型

如果你的服务器逻辑能保证localAddr和remoteAddr的类型始终一致（比如TCP连接的两端都是TCPAddr，UDP同理），甚至可以省略ok的检查，让代码更紧凑。

内容的提问来源于stack exchange，提问作者Jamie