Go 没有泛型，业务逻辑相似的 grpc 调用代码如何能更优雅一些？

V2EX = way to explore

V2EX 是一个关于分享和探索的地方

现在注册

已注册用户请登录

The Go Programming Language

http://golang.org/

Go Playground

Go Projects

Revel Web Framework

这是一个创建于 1846 天前的主题，其中的信息可能已经有所发展或是发生改变。

有个 gate 网关，包装一层 grpc 调用 logic 上的接口，协议使用的 protobuf 。大致业务流程都差不多： 1.反序列化成对象 2.调用 grpc 接口 3.根据结果，发送到客户端。

具体代码如下：

// 创建群 func (tcp *TcpConn) onHandleCreateGroup(header *cim.ImHeader, buff []byte) { req := &cim.CIMGroupCreateReq{} err := proto.Unmarshal(buff, req) if err != nil { logger.Sugar.Warnf("onHandleCreateGroup error") return } logger.Sugar.Info("onHandleCreateGroup") conn := GetMessageConn() ctx, cancelFun := context.WithTimeout(context.Background(), time.Second*3) defer cancelFun() rsp, err := conn.CreateGroup(ctx, req) if err != nil { logger.Sugar.Warnf("CreateGroup(gRPC) err:", err.Error()) } else { _, err = tcp.Send(header.SeqNum, uint16(GROUP_CREATE_DEFAULT_REQ), rsp) logger.Sugar.Infof("onHandleCreateGroup CreateGroup(gRPC) res") } } // 解散群 func (tcp *TcpConn) onHandleDisbandingGroup(header *cim.ImHeader, buff []byte) { req := &cim.CIMGroupDisbandingReq{} err := proto.Unmarshal(buff, req) if err != nil { logger.Sugar.Warnf("onHandleDisbandingGroup error") return } logger.Sugar.Info("onHandleDisbandingGroup") conn := GetMessageConn() ctx, cancelFun := context.WithTimeout(context.Background(), time.Second*3) defer cancelFun() rsp, err := conn.DisbandingGroup(ctx, req) if err != nil { logger.Sugar.Warnf("DisbandingGroup(gRPC) err:") } else { _, err = tcp.Send(header.SeqNum, uint16(GROUP_DISBINGDING_RSP), rsp) logger.Sugar.Infof("onHandleDisbandingGroup res") } }

17 条回复 2020-11-10 14:24:26 +08:00

BoarBoar

2020-10-22 20:40:04 +08:00

一般都是 interface{}代替泛型，不过会写很多用反射或者断言去判断 interface 持有类型的代码，用得少的话没必要搞

lance6716

2020-10-22 23:00:18 +08:00 via Android

我当时用的运行时和反射。当时也考虑过 go generate template…

fuis

2020-10-22 23:39:40 +08:00

@xmcy0011 随手写了一段，大概看个意思。基本思路就是把重复的地方抠出来做成 func，然后用另一个 helper func 去调用它们。参数尽量看准原来就使用 interface{} 的地方，这样不用写很多 type assert

```go
func (tcp *TcpConn) onHandleCreateGroup(header *cim.ImHeader, buff []byte) {
req := &cim.CIMGroupCreateReq{}
call(header, buff, req, func() {
return conn.CreateGroup(ctx, req)
}, func(rsp Response) {
_, err = tcp.Send(...)
})
}

func (tcp *TcpConn) onHandleDisbandingGroup(header *cim.ImHeader, buff []byte) {
req := &cim.CIMGroupDisbandingReq{}
call(header, buff, req, func () {
return conn.DisbandingGroup(ctx, req)
}, func(rsp Response) {
_, err = tcp.Send(...)
})
}

func (t *TcpConn) call(
header *cim.ImHeader,
buff []byte,
req interface{},
method string,
rpcFn func(context.Context, interface{}) (Response, error),
callback func(Response) error) func() {
return func() {
err := proto.Unmarshal(buff, req)
if err != nil {
logger.Sugar.Warnf("%s error", method)
return
}

logger.Sugar.Info(method)

conn := GetMessageConn()
ctx, cancelFun := context.WithTimeout(context.Background(), time.Second*3)
defer cancelFun()

rsp, err := rpcFn(ctx, req)
if err != nil {
logger.Sugar.Warnf("DisbandingGroup(gRPC) err:")
} else {
_, err := callback(rsp)
logger.Sugar.Infof("onHandleDisbandingGroup res")
}
}
}

```