golang gRPC示例

gRPC是一个高性能、通用的开源RPC框架，其由Google主要面向移动应用开发并基于HTTP/2协议标准而设计，基于ProtoBuf(Protocol Buffers)序列化协议开发，且支持众多开发语言。gRPC提供了一种简单的方法来精确地定义服务和为iOS、Android和后台支持服务自动生成可靠性很强的客户端功能库。客户端充分利用高级流和链接功能，从而有助于节省带宽、降低的TCP链接次数、节省CPU使用、和电池寿命。

1、普通帐号安装protobuf

unzip protobuf-cpp-3.0.0-alpha-3.zip
cd protobuf-3.0.0-alpha-3/
./configure
make && sudo make install
go get -u github.com/golang/protobuf/protoc-gen-go #golang 插件

2、定义grpc.proto文件

syntax = "proto3";  //protobuf3协议
package inf;

//请求
message UserRq {
int32 id = 1;
}

//响应
message UserRp {
string name = 1;
}

//服务
service Data {
rpc GetUser(UserRq) returns (UserRp);
}

然后编译

cd ~/src/inf
protoc --go_out=plugins=grpc:. grpc.proto

3、get 所引用的包

go get -u google.golang.org/grpc

由于墙的原因，我们一些依赖的包文件可以通过下面方式下载到：

在 github 可以找到源码，下载后复制到对应目录即可的：

google.golang.org/grpc 对应的代码地址在： https://github.com/grpc/grpc-go

google.golang.org/cloud/compute/metadata 对应的代码地址在： https://github.com/GoogleCloudPlatform/gcloud-golang 不再需要

golang.org/x/oauth2 对应的代码地址在： https://github.com/golang/oauth2

golang.org/x/net/context 对应的代码地址在： https://github.com/golang/net

这些包的源码也可以通过 http://gopm.io/ 或者 http://golangtc.com/download/package 进行下载.

4、服务端代码

// grpc project main.go
package main

import (
"inf"
"log"
"net"
"runtime"
"strconv"

"golang.org/x/net/context"
"google.golang.org/grpc"
)

const (
port = "41005"
)

type Data struct{}

func main() {
runtime.GOMAXPROCS(runtime.NumCPU())

//起服务
lis, err := net.Listen("tcp", ":"+port)
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
s := grpc.NewServer()
inf.RegisterDataServer(s, &Data{})
s.Serve(lis)

log.Println("grpc server in: %s", port)
}

// 定义方法
func (t *Data) GetUser(ctx context.Context, request *inf.UserRq) (response *inf.UserRp, err error) {
response = &inf.UserRp{
Name: strconv.Itoa(int(request.Id)) + ":test",
}
return response, err
}

5、客户端代码

package main

import (
"inf"
"log"
"runtime"
"strconv"
"strings"
"sync"
"time"

"math/rand"

"golang.org/x/net/context"
"google.golang.org/grpc"
)

var (
wg sync.WaitGroup
)

const (
networkType = "tcp"
server      = "127.0.0.1"
port        = "41005"
parallel    = 50        //连接并行度
times       = 100000    //每连接请求次数
)

func main() {
runtime.GOMAXPROCS(runtime.NumCPU())
currTime := time.Now()

//并行请求
for i := 0; i < int(parallel); i++ {
wg.Add(1)
go func() {
defer wg.Done()
exe()
}()
}
wg.Wait()

log.Printf("time taken: %.2f ", time.Now().Sub(currTime).Seconds())
}

func exe() {
//建立连接
conn, _ := grpc.Dial(server + ":" + port)
defer conn.Close()
client := inf.NewDataClient(conn)

for i := 0; i < int(times); i++ {
getUser(client)
}
}

func getUser(client inf.DataClient) {
var request inf.UserRq
r := rand.Intn(parallel)
request.Id = int32(r)

response, _ := client.GetUser(context.Background(), &request) //调用远程方法

//判断返回结果是否正确
if id, _ := strconv.Atoi(strings.Split(response.Name, ":")[0]); id != r {
log.Printf("response error  %#v", response)
}

}

引用

http://www.infoq.com/cn/news/2015/03/grpc-google-http2-protobuf

https://github.com/google/protobuf

https://github.com/golang/protobuf

http://studygolang.com/articles/3192