关于golang:grpcgateway让你同时拥有-grpc-和-http-接口一键生成-swagger-文档

编写 proto 文件

syntax = "proto3";

package pet.service.v1;
option go_package = ".;petpb";

import "google/protobuf/empty.proto";
import "google/api/annotations.proto";
import "google/protobuf/timestamp.proto";

service PetService {rpc ListPet (google.protobuf.Empty) returns (PetList) {option (google.api.http) = {get: "/v1/pets"};
  }

  rpc GetPet (Id) returns (Pet) {option (google.api.http) = {get: "/v1/pets/{id}"
    };
  }

  rpc CreatePet (Pet) returns (Pet) {option (google.api.http) = {post: "/v1/pets"};
  }

  rpc UpdatePet (Pet) returns (Pet) {option (google.api.http) = {put: "/v1/pets/{id}"
    };
  }

  rpc DeletePet (Id) returns (google.protobuf.Empty) {option (google.api.http) = {delete: "/v1/pets/{id}"
    };
  }
}

message Id {string id = 1;}

message PetList {repeated Pet items = 1;}

message Pet {
  string id = 1;
  google.protobuf.Timestamp createdAt = 2;
  google.protobuf.Timestamp updatedAt = 3;
  string name = 4;
  string type = 5;
  string  sex = 6;
  uint32 age = 7;
  bool owned = 8;
}

先决条件：

• protoc 已装置，参考：https://grpc.io/docs/protoc-i…

# MacOS
brew install protobuf

• grpc-gateway protoc 插件已装置

go install \
github.com/grpc-ecosystem/grpc-gateway/v2/protoc-gen-grpc-gateway \
github.com/grpc-ecosystem/grpc-gateway/v2/protoc-gen-openapiv2 \
google.golang.org/protobuf/cmd/protoc-gen-go \
google.golang.org/grpc/cmd/protoc-gen-go-grpc

生成桩代码及 swagger 文档：

protoc -I/usr/local/include -I. \
        -I${GOPATH}/protobuf \
        --go_out . --go_opt paths=source_relative \
        --go-grpc_out . --go-grpc_opt paths=source_relative \
        --grpc-gateway_out . --grpc-gateway_opt paths=source_relative \
        --grpc-gateway_opt logtostderr=true \
        --grpc-gateway_opt generate_unbound_methods=true \
        --grpc-gateway_opt register_func_suffix=GW \
        --grpc-gateway_opt allow_delete_body=true \
        --openapiv2_out . --openapiv2_opt logtostderr=true \
        pet.proto

生成文档

先决条件：

• docker 已装置并运行

有了 swagger.json 借助 swagger 相干的工具咱们就能够生成文档了。这里咱们应用 redoc，通过容器启动一个 server 渲染文档：

docker run -it --rm -p 80:80 \
      -v $$(pwd)/pet.swagger.json:/usr/share/nginx/html/swagger.yaml \
      -e SPEC_URL=swagger.yaml redocly/redoc

浏览器拜访 localhost，就能关上丑陋的文档页面了：

实现接口，启动服务

先启动 grpc server，这里同时通过 grpc-middleware 注入了一些罕用中间件：

s := grpc.NewServer(
    grpc.UnaryInterceptor(grpc_middleware.ChainUnaryServer(grpc_opentracing.UnaryServerInterceptor(),
        grpc_prometheus.UnaryServerInterceptor,
        grpc_logrus.UnaryServerInterceptor(logrusEntry),
        grpc_recovery.UnaryServerInterceptor(),)),
)
petpb.RegisterPetServiceServer(s, petsvc.NewPetService())

log.Infof("grpc server start: %s", addr)
if err := s.Serve(lis); err != nil {log.Fatalf("failed to serve: %v", err)
}

再启动 grpc-gateway，这里依赖 grpc server 的地址：

err := gw.RegisterPetServiceGWFromEndpoint(ctx, mux, grpcAddr, opts)
if err != nil {return err}

log.Infof("gateway server start: %s", gatewayAddr)
http.ListenAndServe(gatewayAddr, mux)

原理是在 grpc server 之前做了一层 http 反向代理，将 http 申请转为 protobuf 送给后端的 grpc server，对返回值再做一次转换。官网架构图 画的比较清楚了。

残缺代码

先决条件：

• make 命令已装置

proto 及 swagger：https://github.com/win5do/go-…

执行 make gen 生成 grpc 桩代码及 swagger.json 文档。

执行 make serve-docs 启动容器渲染文档。

grpc 及 gateway：https://github.com/win5do/go-…

Reference

https://github.com/grpc-ecosy…