This changes the RPC methods to return/consume a stream of chunks, instead of a very big message containing the whole blob, to keep message sizes in manageable sizes (less than 4MiB). Change-Id: I2a3a50f07b059d8a2f5196860254adff98c8a352 Reviewed-on: https://cl.tvl.fyi/c/depot/+/7651 Reviewed-by: tazjin <tazjin@tvl.su> Tested-by: BuildkiteCI
202 lines
5.8 KiB
Go
202 lines
5.8 KiB
Go
// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
|
|
// versions:
|
|
// - protoc-gen-go-grpc v1.2.0
|
|
// - protoc (unknown)
|
|
// source: tvix/store/protos/rpc_blobstore.proto
|
|
|
|
package storev1
|
|
|
|
import (
|
|
context "context"
|
|
grpc "google.golang.org/grpc"
|
|
codes "google.golang.org/grpc/codes"
|
|
status "google.golang.org/grpc/status"
|
|
)
|
|
|
|
// This is a compile-time assertion to ensure that this generated file
|
|
// is compatible with the grpc package it is being compiled against.
|
|
// Requires gRPC-Go v1.32.0 or later.
|
|
const _ = grpc.SupportPackageIsVersion7
|
|
|
|
// BlobServiceClient is the client API for BlobService service.
|
|
//
|
|
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
|
|
type BlobServiceClient interface {
|
|
Get(ctx context.Context, in *GetBlobRequest, opts ...grpc.CallOption) (BlobService_GetClient, error)
|
|
Put(ctx context.Context, opts ...grpc.CallOption) (BlobService_PutClient, error)
|
|
}
|
|
|
|
type blobServiceClient struct {
|
|
cc grpc.ClientConnInterface
|
|
}
|
|
|
|
func NewBlobServiceClient(cc grpc.ClientConnInterface) BlobServiceClient {
|
|
return &blobServiceClient{cc}
|
|
}
|
|
|
|
func (c *blobServiceClient) Get(ctx context.Context, in *GetBlobRequest, opts ...grpc.CallOption) (BlobService_GetClient, error) {
|
|
stream, err := c.cc.NewStream(ctx, &BlobService_ServiceDesc.Streams[0], "/tvix.store.v1.BlobService/Get", opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
x := &blobServiceGetClient{stream}
|
|
if err := x.ClientStream.SendMsg(in); err != nil {
|
|
return nil, err
|
|
}
|
|
if err := x.ClientStream.CloseSend(); err != nil {
|
|
return nil, err
|
|
}
|
|
return x, nil
|
|
}
|
|
|
|
type BlobService_GetClient interface {
|
|
Recv() (*BlobChunk, error)
|
|
grpc.ClientStream
|
|
}
|
|
|
|
type blobServiceGetClient struct {
|
|
grpc.ClientStream
|
|
}
|
|
|
|
func (x *blobServiceGetClient) Recv() (*BlobChunk, error) {
|
|
m := new(BlobChunk)
|
|
if err := x.ClientStream.RecvMsg(m); err != nil {
|
|
return nil, err
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
func (c *blobServiceClient) Put(ctx context.Context, opts ...grpc.CallOption) (BlobService_PutClient, error) {
|
|
stream, err := c.cc.NewStream(ctx, &BlobService_ServiceDesc.Streams[1], "/tvix.store.v1.BlobService/Put", opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
x := &blobServicePutClient{stream}
|
|
return x, nil
|
|
}
|
|
|
|
type BlobService_PutClient interface {
|
|
Send(*BlobChunk) error
|
|
CloseAndRecv() (*PutBlobResponse, error)
|
|
grpc.ClientStream
|
|
}
|
|
|
|
type blobServicePutClient struct {
|
|
grpc.ClientStream
|
|
}
|
|
|
|
func (x *blobServicePutClient) Send(m *BlobChunk) error {
|
|
return x.ClientStream.SendMsg(m)
|
|
}
|
|
|
|
func (x *blobServicePutClient) CloseAndRecv() (*PutBlobResponse, error) {
|
|
if err := x.ClientStream.CloseSend(); err != nil {
|
|
return nil, err
|
|
}
|
|
m := new(PutBlobResponse)
|
|
if err := x.ClientStream.RecvMsg(m); err != nil {
|
|
return nil, err
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
// BlobServiceServer is the server API for BlobService service.
|
|
// All implementations must embed UnimplementedBlobServiceServer
|
|
// for forward compatibility
|
|
type BlobServiceServer interface {
|
|
Get(*GetBlobRequest, BlobService_GetServer) error
|
|
Put(BlobService_PutServer) error
|
|
mustEmbedUnimplementedBlobServiceServer()
|
|
}
|
|
|
|
// UnimplementedBlobServiceServer must be embedded to have forward compatible implementations.
|
|
type UnimplementedBlobServiceServer struct {
|
|
}
|
|
|
|
func (UnimplementedBlobServiceServer) Get(*GetBlobRequest, BlobService_GetServer) error {
|
|
return status.Errorf(codes.Unimplemented, "method Get not implemented")
|
|
}
|
|
func (UnimplementedBlobServiceServer) Put(BlobService_PutServer) error {
|
|
return status.Errorf(codes.Unimplemented, "method Put not implemented")
|
|
}
|
|
func (UnimplementedBlobServiceServer) mustEmbedUnimplementedBlobServiceServer() {}
|
|
|
|
// UnsafeBlobServiceServer may be embedded to opt out of forward compatibility for this service.
|
|
// Use of this interface is not recommended, as added methods to BlobServiceServer will
|
|
// result in compilation errors.
|
|
type UnsafeBlobServiceServer interface {
|
|
mustEmbedUnimplementedBlobServiceServer()
|
|
}
|
|
|
|
func RegisterBlobServiceServer(s grpc.ServiceRegistrar, srv BlobServiceServer) {
|
|
s.RegisterService(&BlobService_ServiceDesc, srv)
|
|
}
|
|
|
|
func _BlobService_Get_Handler(srv interface{}, stream grpc.ServerStream) error {
|
|
m := new(GetBlobRequest)
|
|
if err := stream.RecvMsg(m); err != nil {
|
|
return err
|
|
}
|
|
return srv.(BlobServiceServer).Get(m, &blobServiceGetServer{stream})
|
|
}
|
|
|
|
type BlobService_GetServer interface {
|
|
Send(*BlobChunk) error
|
|
grpc.ServerStream
|
|
}
|
|
|
|
type blobServiceGetServer struct {
|
|
grpc.ServerStream
|
|
}
|
|
|
|
func (x *blobServiceGetServer) Send(m *BlobChunk) error {
|
|
return x.ServerStream.SendMsg(m)
|
|
}
|
|
|
|
func _BlobService_Put_Handler(srv interface{}, stream grpc.ServerStream) error {
|
|
return srv.(BlobServiceServer).Put(&blobServicePutServer{stream})
|
|
}
|
|
|
|
type BlobService_PutServer interface {
|
|
SendAndClose(*PutBlobResponse) error
|
|
Recv() (*BlobChunk, error)
|
|
grpc.ServerStream
|
|
}
|
|
|
|
type blobServicePutServer struct {
|
|
grpc.ServerStream
|
|
}
|
|
|
|
func (x *blobServicePutServer) SendAndClose(m *PutBlobResponse) error {
|
|
return x.ServerStream.SendMsg(m)
|
|
}
|
|
|
|
func (x *blobServicePutServer) Recv() (*BlobChunk, error) {
|
|
m := new(BlobChunk)
|
|
if err := x.ServerStream.RecvMsg(m); err != nil {
|
|
return nil, err
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
// BlobService_ServiceDesc is the grpc.ServiceDesc for BlobService service.
|
|
// It's only intended for direct use with grpc.RegisterService,
|
|
// and not to be introspected or modified (even as a copy)
|
|
var BlobService_ServiceDesc = grpc.ServiceDesc{
|
|
ServiceName: "tvix.store.v1.BlobService",
|
|
HandlerType: (*BlobServiceServer)(nil),
|
|
Methods: []grpc.MethodDesc{},
|
|
Streams: []grpc.StreamDesc{
|
|
{
|
|
StreamName: "Get",
|
|
Handler: _BlobService_Get_Handler,
|
|
ServerStreams: true,
|
|
},
|
|
{
|
|
StreamName: "Put",
|
|
Handler: _BlobService_Put_Handler,
|
|
ClientStreams: true,
|
|
},
|
|
},
|
|
Metadata: "tvix/store/protos/rpc_blobstore.proto",
|
|
}
|