以下是一个简单的 Go gRPC 例子,其中包括服务端和客户端:
首先,我们需要安装 grpc
和 protobuf
的 Go 插件:
go get -u google.golang.org/grpc
go get -u github.com/golang/protobuf/protoc-gen-go
然后,我们需要定义一个 .proto
文件来定义我们的服务和消息。这里是一个示例 .proto
文件:
protobuf
syntax = "proto3";
package example;
service Greeter {
rpc SayHello (HelloRequest) returns (HelloResponse);
}
message HelloRequest {
string name = 1;
}
message HelloResponse {
string message = 1;
}
接下来,我们使用 protoc
工具来生成 Go 代码:
protoc --go_out=plugins=grpc:. *.proto
这将在当前目录下生成 example.pb.go
文件,其中包含我们定义的服务和消息的 Go 代码。
接下来,我们可以编写服务端代码:
package main
import (
"context"
"google.golang.org/grpc"
"log"
"net"
pb "path/to/your/proto/package"
)
type server struct{}
func (s *server) SayHello(ctx context.Context, req *pb.HelloRequest) (*pb.HelloResponse, error) {
return &pb.HelloResponse{
Message: "Hello, " + req.Name + "!",
}, nil
}
func main() {
lis, err := net.Listen("tcp", ":50051")
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
s := grpc.NewServer()
pb.RegisterGreeterServer(s, &server{})
log.Println("Starting server on port 50051...")
if err := s.Serve(lis); err != nil {
log.Fatalf("failed to serve: %v", err)
}
}
在这个例子中,我们定义了一个名为 server
的结构体,并实现了 SayHello
方法来响应客户端的请求。我们还在 main
函数中创建了一个 grpc.Server
实例,并将我们的服务注册到该实例上。最后,我们使用 Serve
方法来启动服务端并监听端口 50051
。
现在,我们可以编写客户端代码:
package main
import (
"context"
"log"
"google.golang.org/grpc"
pb "path/to/your/proto/package"
)
func main() {
conn, err := grpc.Dial(":50051", grpc.WithInsecure())
if err != nil {
log.Fatalf("failed to connect: %v", err)
}
defer conn.Close()
c := pb.NewGreeterClient(conn)
resp, err := c.SayHello(context.Background(), &pb.HelloRequest{Name: "World"})
if err != nil {
log.Fatalf("failed to call SayHello: %v", err)
}
log.Printf("Response: %s", resp.Message)
}
在这个例子中,我们首先使用 grpc.Dial
方法来连接到服务端,然后创建一个 GreeterClient
实例来调用服务端的 SayHello
方法。最后,我们输出响应的消息。
现在,我们可以运行服务端和客户端来测试我们的 gRPC 服务了。