[UBP/baseserver] Allow gRPC server to receive much larger messages

components/common-go/baseserver/server.go

@@ -298,6 +298,7 @@ func (s *Server) initializeGRPC() error {
 		opts = append(opts, grpc.Creds(credentials.NewTLS(tlsConfig)))
 	}
 
+	opts = append(opts, grpc.MaxRecvMsgSize(100*1024*1024))
 	s.grpc = grpc.NewServer(opts...)
 
 	reflection.Register(s.grpc)

components/usage/pkg/apiv1/size_test.go

@@ -0,0 +1,59 @@
+// Copyright (c) 2022 Gitpod GmbH. All rights reserved.
+// Licensed under the GNU Affero General Public License (AGPL).
+// See License-AGPL.txt in the project root for license information.
+
+package apiv1
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/gitpod-io/gitpod/common-go/baseserver"
+	v1 "github.com/gitpod-io/gitpod/usage-api/v1"
+	"github.com/gitpod-io/gitpod/usage/pkg/stripe"
+	"github.com/stretchr/testify/require"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
+	timestamppb "google.golang.org/protobuf/types/known/timestamppb"
+)
+
+func TestServerCanReceiveLargeMessages(t *testing.T) {
+	srv := baseserver.NewForTests(t,
+		baseserver.WithGRPC(baseserver.MustUseRandomLocalAddress(t)),
+	)
+
+	v1.RegisterBillingServiceServer(srv.GRPC(), NewBillingService(&stripe.Client{}, time.Time{}))
+	baseserver.StartServerForTests(t, srv)
+
+	conn, err := grpc.Dial(srv.GRPCAddress(), grpc.WithTransportCredentials(insecure.NewCredentials()))
+	require.NoError(t, err)
+
+	client := v1.NewBillingServiceClient(conn)
+
+	_, err = client.UpdateInvoices(context.Background(), &v1.UpdateInvoicesRequest{
+		Sessions: getBilledSessions(),
+	})
+
+	require.NoError(t, err)
+}
+
+func getBilledSessions() (sessions []*v1.BilledSession) {
+	for i := 0; i < 900000; i++ {
+		sessions = append(sessions, &v1.BilledSession{
+			AttributionId:     "user:1234",
+			UserId:            "1234",
+			TeamId:            "",
+			WorkspaceId:       "",
+			WorkspaceType:     "",
+			ProjectId:         "",
+			InstanceId:        "",
+			WorkspaceClass:    "",
+			StartTime:         &timestamppb.Timestamp{},
+			EndTime:           &timestamppb.Timestamp{},
+			CreditsDeprecated: 0,
+			Credits:           0,
+		})
+	}
+	return
+}

components/usage/pkg/server/server.go

@@ -76,8 +76,8 @@ func Start(cfg Config) error {
 		grpc.WithUnaryInterceptor(grpcClientMetrics.UnaryClientInterceptor()),
 		grpc.WithStreamInterceptor(grpcClientMetrics.StreamClientInterceptor()),
 		grpc.WithDefaultCallOptions(
-			grpc.MaxCallRecvMsgSize(50*1024*1024),
-			grpc.MaxCallSendMsgSize(50*1024*1024),
+			grpc.MaxCallRecvMsgSize(100*1024*1024),
+			grpc.MaxCallSendMsgSize(100*1024*1024),
 		))
 	if err != nil {
 		return fmt.Errorf("failed to create self-connection to grpc server: %w", err)