1. 微服务架构上篇：2. grpc+protobuf+网关实战

syntax = "proto3";
package helloworld;
import "gogoproto/gogo.proto";
import "google/api/annotations.proto";//注解 用于生成反向代理 for gateway使用
option (gogoproto.marshaler_all) = true;
option (gogoproto.unmarshaler_all) = true;
// 服务定义 即我们要提供给外部的接口
service Greeter {
    rpc SayHello (HelloRequest) returns (HelloReply) {
        option (google.api.http) = { // google.api.http会把restful api转为grpc
            post: "/hello" // method：post 路由：/hello
            body: "*"
        };
    }
}
//请求参数结构体
message HelloRequest {
    string name = 1; //1 只是代表参数序号 不是值
}
// 返回响应结构体
message HelloReply {
    string message = 1; //解释同上
}

func main() {
flag.Parse()
  //首先定义一个基于tcp协议的ip端口Listener
 lis, err := net.Listen("tcp", "localhost:50051")
 if err != nil {
  panic(err)
 }
  //初始化一个grpc服务
  s := grpc.NewServer()
  //注册server服务到grpc 这个server结构体会实现SayHello接口
 pb.RegisterGreeterServer(s, &server{})
  //grpc服务利用Listener提供server能力
 s.Serve(lis)
}

// 定义server结构体实现helloworld.GreeterServer接口
type server struct{}
// SayHello实现helloworld.GreeterServer
// 这里就会用proto生成的文件中的请求和响应
func (s *server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloReply, error) {
 logrus.Infof("%v: Receive is %s\n", time.Now(), in.Name)
 return &pb.HelloReply{Message: "Hello " + in.Name + " from " + net.JoinHostPort(*host, *port)}, nil
}

func main() {
 flag.Parse()
 ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
 // 创建 grpc Channel 与 grpc Server 进行通信（需服务器地址和端口作为参数）
 // 并且设置 DialOptions 比如负载均衡，凭证等
 conn, err := grpc.DialContext(ctx, "localhost:50051", grpc.WithInsecure(), grpc.WithBalancerName(roundrobin.Name), grpc.WithBlock())
 cancel()
 if err != nil {
  panic(err)
 }
 //创建 Client Stub
 client := pb.NewGreeterClient(conn)
 //调用对应的服务方法 即调服务端定义的接口SayHello 参数和server参数对应 是proto的request
 resp, err := client.SayHello(context.Background(), &pb.HelloRequest{Name: "world " + strconv.Itoa(t.Second())})
 if err == nil {
  logrus.Infof("%v: Reply is %s\n", t, resp.Message)
 }
}

func main() {
 conn, err := grpc.DialContext(ctx, "localhost:50051", grpc.WithInsecure(), grpc.WithBalancerName(roundrobin.Name), grpc.WithBlock())
 cancel()
 if err != nil {
  panic(err)
 }
 // 初始化ServeMux
 mux := runtime.NewServeMux()
 // 将Greeter服务注册到mux
 err = pb.RegisterGreeterHandler(ctx, mux, conn)
 if err != nil {
  panic(err)
 }
 // Start HTTP server (and proxy calls to grpc server endpoint)
 logrus.Fatal(http.ListenAndServe("localhost:60001", mux))
}