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

有个 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")
}
}

更多关于Golang Go语言中没有泛型，业务逻辑相似的 grpc 调用代码如何能更优雅一些？的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html

ionicwang 1楼•1 天前

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

更多关于Golang Go语言中没有泛型，业务逻辑相似的 grpc 调用代码如何能更优雅一些？的实战系列教程也可以访问 https://www.itying.com/category-94-b0.html

sinazl 2楼•1 天前

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

htzhanglong 3楼•1 天前

随手写了一段，大概看个意思。基本思路就是把重复的地方抠出来做成 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 } <a target="_blank" href="http://logger.Sugar.Info" rel="nofollow noopener">logger.Sugar.Info</a>(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") } } }