grpc example

example/codegen.sh

@@ -0,0 +1,8 @@
+find proto -type f -name "*.pb.go" -delete
+
+# https://developers.google.com/protocol-buffers/docs/reference/go-generated#package
+# 默认就是import 模式
+protoc --go_out=.  --go_opt=paths=import  \
+    --go-grpc_out=.  --go-grpc_opt=paths=import \
+    proto/helloworld.proto
+

example/grpc_client_test.go

@@ -0,0 +1,36 @@
+/**
+User: cr-mao
+Date: 2024/2/19 12:52
+Email: crmao@qq.com
+Desc: grpc_server_client.go
+*/
+package example
+
+import (
+	"context"
+	"github.com/cr-mao/lori/example/proto"
+	"github.com/cr-mao/lori/log"
+	"testing"
+	"time"
+
+	"github.com/cr-mao/lori/transport/grpc"
+)
+
+func TestGrpcClient(t *testing.T) {
+	baseCtx := context.Background()
+	conn, err := grpc.DialInsecure(baseCtx,
+		grpc.WithClientEndpoint("127.0.0.1:8081"),
+		grpc.WithClientTimeout(time.Second*5))
+	if err != nil {
+		log.Fatalf("err:%v", err)
+	}
+	client := proto.NewGreeterClient(conn)
+	resp, err := client.SayHello(context.Background(), &proto.HelloRequest{
+		Name: "cr-mao",
+	})
+	if err != nil {
+		log.Errorf("SayHello err:%v", err)
+		return
+	}
+	log.Info(resp.Message)
+}

example/grpc_server_test.go

@@ -0,0 +1,40 @@
+/**
+User: cr-mao
+Date: 2024/2/19 12:48
+Email: crmao@qq.com
+Desc: test_grpc_server.go
+*/
+package example
+
+import (
+	"context"
+	"github.com/cr-mao/lori/example/proto"
+	"github.com/cr-mao/lori/log"
+	"testing"
+
+	"github.com/cr-mao/lori/transport/grpc"
+)
+
+type HelloWorldServer struct {
+	proto.UnsafeGreeterServer
+}
+
+func (s *HelloWorldServer) SayHello(ctx context.Context, r *proto.HelloRequest) (*proto.HelloResponse, error) {
+	return &proto.HelloResponse{
+		Message: "hello " + r.Name,
+	}, nil
+}
+
+func registerServer(server *grpc.Server) {
+	proto.RegisterGreeterServer(server, &HelloWorldServer{})
+}
+
+func TestGrpcServer(t *testing.T) {
+	baseCtx := context.Background()
+	grpcServer := grpc.NewServer(grpc.WithAddress("0.0.0.0:8081"))
+	registerServer(grpcServer)
+	err := grpcServer.Start(baseCtx)
+	if err != nil {
+		log.Errorf("err %v", err)
+	}
+}

example/proto/helloworld.proto

@@ -0,0 +1,17 @@
+syntax = "proto3";
+
+package lori.example.proto;
+
+option go_package = "./proto";
+
+service Greeter {
+  rpc SayHello (HelloRequest) returns (HelloResponse);
+}
+
+message HelloRequest {
+  string name = 1;
+}
+
+message HelloResponse{
+  string message = 1;
+}