Go语言中如何获取SQL连接对应的客户端端口号？

标准database/sql包没有直接暴露底层连接源端口的公开API，这个端口属于TCP传输层属性，被封装在数据库驱动的内部实现中，默认确实无法直接从sql.DB或sql.Conn对象中读取，但可以通过两种可靠方案获取。

常用的Postgres Go驱动（lib/pq、pgx）都支持自定义连接生命周期钩子，你可以在连接首次建立时拿到底层TCP连接，直接读取源端口。
以pgx驱动为例，使用AfterConnect钩子实现：

// 初始化pgx连接池配置
cfg, err := pgxpool.ParseConfig("postgres://your_dsn")
if err != nil {
    panic(err)
}

// 注册连接建立后钩子
cfg.AfterConnect = func(ctx context.Context, conn *pgx.Conn) error {
    rawConn := conn.PgConn().Conn()
    localAddr, ok := rawConn.LocalAddr().(*net.TCPAddr)
    if ok {
        // 拿到源端口，可自行和连接标识绑定存储或打印日志
        sourcePort := localAddr.Port
        println("new db conn, source port:", sourcePort)
    }
    return nil
}

// 初始化连接池
pool, err := pgxpool.NewWithConfig(context.Background(), cfg)

如果使用标准database/sql接口搭配lib/pq驱动，可通过sql.Register注册自定义包装后的驱动，在Open方法返回连接时提取端口。

无需修改驱动逻辑，只要你能拿到连接执行SQL，直接调用Postgres内置的inet_client_port()函数即可获取当前连接的源端口，返回值和Postgres日志中记录的端口完全一致：

var sourcePort int
// 用当前连接执行查询
err := db.QueryRow("SELECT inet_client_port()").Scan(&sourcePort)
if err != nil {
    // 处理错误逻辑
}
// sourcePort就是你要匹配的端口值

这个方案兼容性最高，不需要依赖特定驱动的特性，适合不想调整底层驱动配置的场景。

补充说明 #

database/sql作为通用的SQL抽象层，设计上要兼容所有类型的数据库（包括非TCP连接的内存数据库、Unix套接字连接的数据库等），因此不会暴露TCP专属的端口属性，确实属于内部封装的范畴，但上面两种方案都可以绕过这个限制拿到所需的端口。

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