如何获取 gRPC 响应限制

Question

我有返回一些项目数组的 Go gRPC 端点。我想限制响应的项目​​数，以适应 gRPC 允许发送的最大值。

我的想法是在处理函数 func 中获得最大值，然后将其除以项目大小，然后瞧。但是我怎样才能获得最大响应大小？

我不想设置最大响应大小以使我的类独立于 grpc 实例化。

---

ma​​in.go

package main

import (
    "log"
    "main/api"
    "main/router"
    "net"

    "google.golang.org/grpc"
)

func main() {
    l, err := net.Listen("tcp", "0.0.0.0:138080")
    if err != nil {
        log.Fatalf("failed to listen: %v", err)
    }

    grpcServer := grpc.NewServer()

    r := router.New()

    api.RegisterTestGenerateServiceServer(grpcServer, r)

    err = grpcServer.Serve(l)
    if err != nil {
        log.Fatalf("failed to run server: %v", err)
    }
}

---

router.go

package router

import (
    "context"
    "main/api"
    "main/model"
    "unsafe"
)

type router struct {
}

func New() *router {
    return &router{}
}

func (r *router) TestCall(context.Context, *api.TestCallRequest) (*api.TestCallResponse, error) {
    items := somewhere.GetItems()
    apiItems := transform.ToAPIItems(items)
    
    itemSize := unsafe.Sizeof(model.Item{})
    
    responseSize := someWonderFuncGetGrpcResponseMaxSize()
    
    NItems := responseSize / itemSize

    return &api.TestCallResponse{
        Items: apiItems[:NItems],
    }, nil
}

Answer 1

尝试在您的客户端上更新grpc.MaxCallSendMsgSize(s int) 以使您的客户端能够发送更大的消息。 这对我有用，例如：

func main() {
    // create connection
    opts := []grpc.DialOption{
        // grpc.WithInsecure(),
        grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(math.MaxInt32)), // uncomment for large amount of data returned
    }
    cc, err := grpc.Dial("localhost:9089", opts...)
    if err != nil {
        fmt.Printf("failed to dial: %v\n", err)
        return
    }
    defer cc.Close()

    client := pb.NewGrpcServiceClient(cc)
    data, err := client.GetDataFromGRPCService()