一个go http和grpc客户端库

大家好，我是peachesTao，今天是五一假期的第4天，首先祝大家劳动节快乐。今天给大家推荐一个统一http和grpc客户端调用的库，名为prpc，github地址：prpc，该库是我公司根据最佳实践总结开发出来的，它可以提升开发效率，让开发者聚焦于业务层代码。

下面我从三个方面来介绍

自动生成http客户端代码

1、编写http客户端代码痛点

首选我们来看一段调用http接口获取用户列表的代码

type GetUserListReq struct {Age  int32  `json:"age"`
}type GetUserListReply struct {Code int32       `json:"code"`Msg  string      `json:"msg"`List []*UserInfo `json:"list"`
}type UserInfo struct {Name string `json:"name"`Age  int32  `json:"age"`
}func GetUserList(req *protocol.GetUserListReq) (*GetUserListReply, error) {user := config.GetServerConfig().Userrequest, err := httplib.Post(user.URL+user.GetUserList).Debug(true).JSONBody(req)if err != nil {return nil, err}resp := new(protocol.GetUserListReply)err = request.ToJSON(resp)if err != nil {return nil, err}return resp, nil
}

分别定义了请求和响应数据结构体，从配置文件中获取http服务的url和用户列表接口的path，将请求结构体编码，然后通过http client库发起http请求，将返回的数据解码为响应结构体。

我们发现这样的代码有下面三个痛点

每次调用都要对req对象编码、res对象进行解码
需要在代码中硬编码读取配置服务端的url
如果服务端将Post改成Get方式,要改客户端的代码

这非常影响开发效率且容易出错，这些代码能不能自动生成？答案是可以的，凡是有规律的代码都可以自动生成。我们的实现方案是基于protobuf实现的。

2、基于protobuf的http客户端代码生成

我们使用protobuf定义服务调用方法、请求和响应结构体，再借助http Method Option自定义服务url和http method,最后用prpc内置的插件protoc-gen-go-prpc生成跟grpc相似的函数签名的客户端代码。

GetUserList protobuf文件定义：

syntax = "proto3";option go_package = "/api/";
import "google/api/annotations.proto";service User {//获取用户列表rpc GetUserList(GetUserListReq) returns (GetUserListReply) {option (google.api.http) = {get: "/user/list"body: "*"additional_bindings:[{custom:{kind:"Content-Type", path:"application/json"}}]};}
}message GetUserListReq {int32 age = 1;
}message GetUserListReply {int32 code = 1;string msg = 2 ;repeated UserInfo list = 3;
}message UserInfo {string name = 1;int32 age = 2;
}

自动生成的调用入口函数如下（此处省略了其他部分代码，想看完整的代码可以前往prpc-examples）：

func (c *userClient) GetUserList(ctx context.Context, in *GetUserListReq, opts ...http.CallOption) (*GetUserListReply, error) {out := new(GetUserListReply)header := map[string]string{"Content-Type": "application/json",}opts = http.CallOptions(opts).CombineHeader(header)err := c.cc.HttpInvoke(ctx, "GET", "/user/list", in, out, opts...)if err != nil {return nil, err}return out, nil
}

开发者侧代码：

 conn := prpc.NewClientConn()err := conn.NewHttpClientConn(context.Background(), "127.0.0.1:8000")if err != nil {log.Error(err)return}userClient := api.NewUserClient(conn)listReply, err := userClient.GetUserList(context.Background(), &api.GetUserListReq{Age: 18})if err != nil {log.Error(err)return}log.Info(listReply)

可以看出，开发者就像调用grpc方法一样简单的调用http接口，实现了对协议的无感，开发者只需要在protobuf中定义相关信息，其他的交给插件就行。

http和grpc调用随意切换

因为http和grpc客户端调用代码都是根据protobuf生成的，生成的是http还是grpc取决于rpc方法是否定义了http method option,如果定义了则生成http代码，否则生成grpc调用代码。

所以http切换成grpc protobuf只需要删除http method option 部分即可，同理，grpc切换成http则加上它就行。

上面GetUserList改成grpc调用的话protobuf rpc方法定义如下：

service User {//获取用户列表rpc GetUserList(GetUserListReq) returns (GetUserListReply) {}
}

服务发现和负载均衡

prpc还包含集群模式下的服务发现和负载均衡功能，一套服务发现和负载均衡机制同时兼容http和grpc，且都支持扩展。

开发者只需要实现resolver和balancer接口就可以自定义服务发现和负载均衡，目前prpc内置了基于consul的服务发现和轮训的负载均衡功能。

后续我们将增加基于zookeeper,etcd的服务发现和实现权重的负载均衡方式，也欢迎小伙伴积极参与进来，给我们贡献代码，共同完善prpc。

更多的细节请前往github prpc了解