以下是关于如何使用rk-grpc库实现gRPC日志监控认证追踪的中间件插件的完整示例和说明：

rk-grpc简介

rk-grpc是一个Golang库，可以从YAML文件注入gRPC和grpc-gateway的中间件和服务器配置。它属于rk-boot家族，建议与rk-boot一起使用。

架构图

快速入门示例

1. 安装

go get github.com/rookie-ninja/rk-grpc/v2

2. 准备.proto文件

syntax = "proto3";

package api.v1;

option go_package = "api/v1/greeter";

service Greeter {
  rpc Greeter (GreeterRequest) returns (GreeterResponse) {}
}

message GreeterRequest {
  bytes msg = 1;
}

message GreeterResponse {}

3. 创建boot.yaml配置文件

grpc:
  - name: greeter                     # 必填
    port: 8080                        # 必填
    enabled: true                     # 必填
    enableReflection: true            # 可选，默认: false
    enableRkGwOption: true            # 可选，默认: false
    commonService:
      enabled: true                   # 可选，默认: false
    docs:
      enabled: true                   # 可选，默认: false
    sw:
      enabled: true                   # 可选，默认: false
    prom:
      enabled: true                   # 可选，默认: false
    middleware:
      logging:
        enabled: true                 # 可选，默认: false
      prom:
        enabled: true                 # 可选，默认: false
      meta:
        enabled: true                 # 可选，默认: false

4. 创建main.go

package main

import (
  "context"
  "embed"
  _ "embed"
  "github.com/rookie-ninja/rk-entry/v2/entry"
  "github.com/rookie-ninja/rk-grpc/v2/boot"
  proto "github.com/rookie-ninja/rk-grpc/v2/example/boot/simple/api/gen/v1"
  "google.golang.org/grpc"
)

//go:embed boot.yaml
var boot []byte

//go:embed api/gen/v1
var docsFS embed.FS

//go:embed api/gen/v1
var staticFS embed.FS

func init() {
  rkentry.GlobalAppCtx.AddEmbedFS(rkentry.DocsEntryType, "greeter", &docsFS)
  rkentry.GlobalAppCtx.AddEmbedFS(rkentry.SWEntryType, "greeter", &docsFS)
  rkentry.GlobalAppCtx.AddEmbedFS(rkentry.StaticFileHandlerEntryType, "greeter", &staticFS)
}

func main() {
  // 从boot配置引导基本入口
  rkentry.BootstrapPreloadEntryYAML(boot)

  // 从boot配置引导grpc入口
  res := rkgrpc.RegisterGrpcEntryYAML(boot)

  // 获取GrpcEntry
  grpcEntry := res["greeter"].(*rkgrpc.GrpcEntry)
  
  // 注册gRPC服务器
  grpcEntry.AddRegFuncGrpc(func(server *grpc.Server) {
    proto.RegisterGreeterServer(server, &GreeterServer{})
  })
  
  // 注册grpc-gateway函数
  grpcEntry.AddRegFuncGw(proto.RegisterGreeterHandlerFromEndpoint)

  // 引导grpc入口
  grpcEntry.Bootstrap(context.Background())

  // 等待关机信号
  rkentry.GlobalAppCtx.WaitForShutdownSig()

  // 中断gin入口
  grpcEntry.Interrupt(context.Background())
}

// GreeterServer实现GreeterServer接口
type GreeterServer struct{}

// Greeter处理Greeter方法
func (server *GreeterServer) Greeter(context.Context, *proto.GreeterRequest) (*proto.GreeterResponse, error) {
  return &proto.GreeterResponse{}, nil
}

5. 启动服务器

go run main.go

验证功能

1. gRPC & grpc-gateway服务

# 访问通用服务
curl localhost:8080/rk/v1/ready

2. Swagger UI

默认访问地址：http://localhost:8080/sw

3. 文档UI

默认访问地址：http://localhost:8080/docs

4. Prometheus指标

默认访问地址：http://localhost:8080/metrics

支持的中间件

rk-grpc支持多种中间件，包括：

• 日志(Logging)
• 监控(Metrics)
• 追踪(Tracing)
• 认证(Auth)
• 限流(RateLimit)
• 超时(Timeout)
• CORS
• JWT
• 安全(Secure)
• CSRF等

所有中间件都可以通过YAML或代码配置，用户可以根据需要启用。

完整YAML配置选项

grpc:
  - name: greeter                     # 必填
    port: 8080                        # 必填
    enabled: true                     # 必填
    # 其他可选配置...
    middleware:
      logging:
        enabled: true
      prom:
        enabled: true
      meta:
        enabled: true
      # 其他中间件配置...

这个示例展示了如何使用rk-grpc快速搭建一个具有日志、监控和认证追踪功能的gRPC服务。通过YAML配置可以轻松启用或禁用各种中间件功能。