在 go 应用程序中,与 grpc 服务交互的步骤如下:创建一个客户端连接:使用 grpc.dial 函数连接到目标 grpc 服务。创建一个 grpc 客户端:使用 newxxxclient 函数创建服务特定的客户端。调用 grpc 服务方法:使用客户端调用 grpc 服务方法,传入请求消息并接收响应消息。处理响应:处理服务方法调用的响应,检查错误并打印响应消息。
用 Go 函数与其他 gRPC 服务交互
在 Go 应用程序中与 gRPC 服务进行交互是一种常见的场景。要做到这一点,我们需要建立一个客户端连接到目标 gRPC 服务,然后调用服务中的方法。有了 Go 函数,这个过程变得更加简单。
创建 gRPC 客户连接
要创建到 gRPC 服务的连接,我们需要使用 grpc.Dial 函数:
conn, err := grpc.Dial("localhost:50051", grpc.WithInsecure())
这个函数接受一个地址和一组选项。在这种情况下,我们正在连接到一个本地侦听在端口 50051 上的 gRPC 服务,并且我们指定了 grpc.WithInsecure 选项,因为我们使用的是未加密的 HTTP/2 传输。
创建 gRPC 客户端
有了连接,我们就可以创建服务特定的客户端:
client := pb.NewGreeterClient(conn)
NewGreeterClient 函数会创建一个实现 pb.GreeterClient 接口的客户端,其中包含与 gRPC 服务关联的方法。
调用 gRPC 服务方法
现在我们可以使用客户端调用 gRPC 服务方法:
resp, err := client.SayHello(ctx, &pb.HelloRequest{Name: "John Doe"})
SayHello 方法接受一个上下文和一个请求消息,并返回一个响应消息。
处理响应
调用服务方法后,我们需要处理响应:
if err != nil {
log.Fatalf("could not greet: %v", err)
}
fmt.Printf("Greeting: %sn", resp.Message)
如果发生错误,我们将打印错误信息并退出程序。否则,我们将打印响应消息。
实战案例:gRPC 聊天室
为了展示如何使用 Go 函数与 gRPC 服务交互,我们创建一个简单的聊天室示例。
gRPC 服务:聊天服务器
package main
import (
"context"
"io"
"time"
"grpc_chat/pb"
"google.golang.org/grpc"
)
type server struct{}
func main() {
lis, err := net.Listen("tcp", ":50051")
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
grpcServer := grpc.NewServer()
pb.RegisterChatServiceServer(grpcServer, &server{})
if err := grpcServer.Serve(lis); err != nil {
log.Fatalf("failed to serve: %v", err)
}
}
func (s *server) Join(stream pb.ChatService_JoinServer) error {
ctx := stream.Context()
for {
select {
case <-ctx.Done():
return ctx.Err()
default:
msg, err := stream.Recv()
if err == io.EOF {
return nil
}
if err != nil {
return err
}
fmt.Printf("Received message from %s: %sn", msg.Name, msg.Message)
if err := stream.Send(&pb.Message{Name: "Server", Message: "Hello " + msg.Message}); err != nil {
return err
}
}
}
}
Go 应用程序:聊天客户端
package main
import (
"context"
"time"
"grpc_chat/pb"
"google.golang.org/grpc"
)
func main() {
conn, err := grpc.Dial("localhost:50051", grpc.WithInsecure())
if err != nil {
log.Fatalf("failed to connect: %v", err)
}
defer conn.Close()
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
client := pb.NewChatServiceClient(conn)
stream, err := client.Join(ctx)
if err != nil {
log.Fatalf("failed to start chat: %v", err)
}
if err := stream.Send(&pb.Message{Name: "John", Message: "Hello!"}); err != nil {
log.Fatalf("failed to send message: %v", err)
}
go func() {
for {
msg, err := stream.Recv()
if err == io.EOF {
log.Println("Chat ended.")
return
}
if err != nil {
log.Fatalf("failed to receive message: %v", err)
}
fmt.Printf("Received message from server: %sn", msg.Message)
}
}()
for {
time.Sleep(time.Second)
if err := stream.Send(&pb.Message{Name: "John", Message: time.Now().Format(time.RFC3339)}); err != nil {
log.Fatalf("failed to send message: %v", err)
}
}
}
在聊天室示例中,服务是一个聊天服务器,允许多个客户端加入并发送消息。客户端应用程序连接到服务器,发送消息并接收服务器的响应消息。交