Go语言如何在编译时条件定义connection结构体？

我有一个配置常量：

const connectionIsSerial = false // 也可能设为true...

还有一个类型定义：

type connection struct {
    connected bool
    mutex     sync.Mutex
    handler *mb.RTUOverTCPClientHandler // 选项A - TCP handler
    //handler *mb.RTUClientHandler     // 选项B - 串口 handler
}

以及全局连接变量：

var conn connection

两种handler类型拥有相似的方法。我希望能在编译时条件定义上述type connection，类似C语言中#ifdef预处理器的效果。

目前我找到的方案是把handler定义为any类型：

type connection struct {
    connected bool
    mutex     sync.Mutex
    handler   any
}

但每次调用handler都要做条件类型断言，既麻烦又不美观。求优雅的解决方案？

方案1：使用Go构建标签（Build Tags）拆分文件 #

这是最贴近C语言#ifdef编译时分支的方案，通过构建标签控制不同编译场景下的代码文件。

1. 创建两个结构体定义文件：

   • conn_tcp.go，添加构建标签//go:build !serial

     //go:build !serial
     package yourpackage
     
     import "sync"
     import "path/to/mb"
     
     type connection struct {
         connected bool
         mutex     sync.Mutex
         handler   *mb.RTUOverTCPClientHandler
     }
     

   • conn_serial.go，添加构建标签//go:build serial

     //go:build serial
     package yourpackage
     
     import "sync"
     import "path/to/mb"
     
     type connection struct {
         connected bool
         mutex     sync.Mutex
         handler   *mb.RTUClientHandler
     }
     

2. 全局变量conn放在通用文件（比如conn.go）中：

   package yourpackage
   
   var conn connection
   

3. 编译时通过标签切换：

   • 编译TCP版本：go build
   • 编译串口版本：go build -tags serial

这种方式完全在编译时确定结构体类型，运行时不需要任何类型断言，类型安全且高效。

方案2：用接口抽象统一handler调用 #

如果不想拆分文件，可以通过定义接口屏蔽两种handler的差异，避免使用any。

1. 先定义包含两种handler共同方法的接口：

   type ConnectionHandler interface {
       // 列出两种handler都有的方法，示例：
       Connect() error
       Disconnect() error
       Send([]byte) (int, error)
       // 其他共同方法...
   }
   

2. 修改connection结构体：

   type connection struct {
       connected bool
       mutex     sync.Mutex
       handler   ConnectionHandler
   }
   

3. 初始化时根据常量创建对应handler实例：

   func init() {
       const connectionIsSerial = false
       if connectionIsSerial {
           conn.handler = &mb.RTUClientHandler{/* 初始化参数 */}
       } else {
           conn.handler = &mb.RTUOverTCPClientHandler{/* 初始化参数 */}
       }
   }
   

后续调用handler方法时，直接通过接口调用即可，无需类型断言，代码更简洁。如果connectionIsSerial是通过go build -ldflags "-X main.connectionIsSerial=true"注入的编译时常量，编译器会自动优化掉无用分支。

方案3：类型别名+构建标签（单文件版本） #

如果不想拆分文件，也可以在单个文件里通过构建标签控制类型别名：

package yourpackage

import "sync"
import "path/to/mb"

//go:build !serial
// +build !serial

type HandlerType = *mb.RTUOverTCPClientHandler

//go:build serial
// +build serial

type HandlerType = *mb.RTUClientHandler

// 通用结构体定义
type connection struct {
    connected bool
    mutex     sync.Mutex
    handler   HandlerType
}

var conn connection

注：Go 1.17+支持新的//go:build语法，旧版本用// +build，同时保留可兼容两种写法。

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