grpc使用教程

教程"/>

grpc使用教程

准备

1，安装grpc
go get -u google.golang/grpc2, 安装go语言protobuf生成器protoc-gen-go
go get -u google.golang/protobuf/cmd/protoc-gen-go注意： 将GOPATH下的bin目录配置在环境变量中否则会找不到命令

编写proto文件

# server/proto/test.proto 文件
syntax = "proto3";package test;
option go_package = "../proto";service Greeter {rpc SayHello (HelloRequest) returns (HelloResponse) {}
}message HelloRequest {string name = 1;
}message HelloResponse {string message = 1;
}

生成protobuf

cd server/proto
protoc --proto_path=. --go_out=plugins=grpc:. *

编写服务端程序

# server/main.go
package mainimport ("context""google.golang/grpc""log""net"pb "server/proto"
)type server struct {
}func (s *server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloResponse, error) {return &pb.HelloResponse{Message: "hello " + in.Name,}, nil
}func main() {listen, err := net.Listen("tcp", ":8080")if err != nil {log.Fatal("net.Listen err", err)}s := grpc.NewServer()pb.RegisterGreeterServer(s, &server{})if err = s.Serve(listen); err != nil {log.Fatal("s.Serve err", err)}
}

编写客户端

# 先将server/proto 目录复制到 client/proto# client/main.go
package mainimport (pb "client/proto""context""google.golang/grpc""google.golang/grpc/credentials/insecure""log""net""net/http"
)type server struct {httpServer http.Server
}func (s *server) ServeHTTP(w http.ResponseWriter, r *http.Request) {if r.URL.Path == "/sayHello" {message := sayHello(r.Form.Get("name"))w.WriteHeader(http.StatusOK)w.Header().Set("Content-Type", "text/html")w.Write([]byte("<h1>" + message + "</h1>"))}
}func sayHello(name string) string {conn, err := grpc.Dial("127.0.0.1:8080", grpc.WithTransportCredentials(insecure.NewCredentials()))if err != nil {log.Fatal("连接失败:", err)}defer conn.Close()client := pb.NewGreeterClient(conn)resp, err := client.SayHello(context.Background(), &pb.HelloRequest{Name: name})if err != nil {log.Fatal("调用失败:", err)}return resp.GetMessage()
}func main() {listen, err := net.Listen("tcp", ":80")if err != nil {log.Fatal("net.Listen err", err)}s := &server{}s.httpServer.Handler = sif err = s.httpServer.Serve(listen); err != nil {log.Fatal("httpServer.Serve err", err)}
}

验证

通过postman请求 http://127.0.0.1/sayHello 接口, 参数 name 为jack, 看到响应了 <h1>hello </h1>