diff options
Diffstat (limited to 'weed/pb/mq_pb/mq_grpc.pb.go')
| -rw-r--r-- | weed/pb/mq_pb/mq_grpc.pb.go | 35 |
1 files changed, 15 insertions, 20 deletions
diff --git a/weed/pb/mq_pb/mq_grpc.pb.go b/weed/pb/mq_pb/mq_grpc.pb.go index f655c4e5b..2b8d11cc8 100644 --- a/weed/pb/mq_pb/mq_grpc.pb.go +++ b/weed/pb/mq_pb/mq_grpc.pb.go @@ -35,7 +35,7 @@ type SeaweedMessagingClient interface { CheckTopicPartitionsStatus(ctx context.Context, in *CheckTopicPartitionsStatusRequest, opts ...grpc.CallOption) (*CheckTopicPartitionsStatusResponse, error) // data plane Publish(ctx context.Context, opts ...grpc.CallOption) (SeaweedMessaging_PublishClient, error) - Subscribe(ctx context.Context, opts ...grpc.CallOption) (SeaweedMessaging_SubscribeClient, error) + Subscribe(ctx context.Context, in *SubscribeRequest, opts ...grpc.CallOption) (SeaweedMessaging_SubscribeClient, error) } type seaweedMessagingClient struct { @@ -149,17 +149,22 @@ func (x *seaweedMessagingPublishClient) Recv() (*PublishResponse, error) { return m, nil } -func (c *seaweedMessagingClient) Subscribe(ctx context.Context, opts ...grpc.CallOption) (SeaweedMessaging_SubscribeClient, error) { +func (c *seaweedMessagingClient) Subscribe(ctx context.Context, in *SubscribeRequest, opts ...grpc.CallOption) (SeaweedMessaging_SubscribeClient, error) { stream, err := c.cc.NewStream(ctx, &SeaweedMessaging_ServiceDesc.Streams[1], "/messaging_pb.SeaweedMessaging/Subscribe", opts...) if err != nil { return nil, err } x := &seaweedMessagingSubscribeClient{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 SeaweedMessaging_SubscribeClient interface { - Send(*SubscribeRequest) error Recv() (*SubscribeResponse, error) grpc.ClientStream } @@ -168,10 +173,6 @@ type seaweedMessagingSubscribeClient struct { grpc.ClientStream } -func (x *seaweedMessagingSubscribeClient) Send(m *SubscribeRequest) error { - return x.ClientStream.SendMsg(m) -} - func (x *seaweedMessagingSubscribeClient) Recv() (*SubscribeResponse, error) { m := new(SubscribeResponse) if err := x.ClientStream.RecvMsg(m); err != nil { @@ -197,7 +198,7 @@ type SeaweedMessagingServer interface { CheckTopicPartitionsStatus(context.Context, *CheckTopicPartitionsStatusRequest) (*CheckTopicPartitionsStatusResponse, error) // data plane Publish(SeaweedMessaging_PublishServer) error - Subscribe(SeaweedMessaging_SubscribeServer) error + Subscribe(*SubscribeRequest, SeaweedMessaging_SubscribeServer) error mustEmbedUnimplementedSeaweedMessagingServer() } @@ -232,7 +233,7 @@ func (UnimplementedSeaweedMessagingServer) CheckTopicPartitionsStatus(context.Co func (UnimplementedSeaweedMessagingServer) Publish(SeaweedMessaging_PublishServer) error { return status.Errorf(codes.Unimplemented, "method Publish not implemented") } -func (UnimplementedSeaweedMessagingServer) Subscribe(SeaweedMessaging_SubscribeServer) error { +func (UnimplementedSeaweedMessagingServer) Subscribe(*SubscribeRequest, SeaweedMessaging_SubscribeServer) error { return status.Errorf(codes.Unimplemented, "method Subscribe not implemented") } func (UnimplementedSeaweedMessagingServer) mustEmbedUnimplementedSeaweedMessagingServer() {} @@ -419,12 +420,15 @@ func (x *seaweedMessagingPublishServer) Recv() (*PublishRequest, error) { } func _SeaweedMessaging_Subscribe_Handler(srv interface{}, stream grpc.ServerStream) error { - return srv.(SeaweedMessagingServer).Subscribe(&seaweedMessagingSubscribeServer{stream}) + m := new(SubscribeRequest) + if err := stream.RecvMsg(m); err != nil { + return err + } + return srv.(SeaweedMessagingServer).Subscribe(m, &seaweedMessagingSubscribeServer{stream}) } type SeaweedMessaging_SubscribeServer interface { Send(*SubscribeResponse) error - Recv() (*SubscribeRequest, error) grpc.ServerStream } @@ -436,14 +440,6 @@ func (x *seaweedMessagingSubscribeServer) Send(m *SubscribeResponse) error { return x.ServerStream.SendMsg(m) } -func (x *seaweedMessagingSubscribeServer) Recv() (*SubscribeRequest, error) { - m := new(SubscribeRequest) - if err := x.ServerStream.RecvMsg(m); err != nil { - return nil, err - } - return m, nil -} - // SeaweedMessaging_ServiceDesc is the grpc.ServiceDesc for SeaweedMessaging service. // It's only intended for direct use with grpc.RegisterService, // and not to be introspected or modified (even as a copy) @@ -495,7 +491,6 @@ var SeaweedMessaging_ServiceDesc = grpc.ServiceDesc{ StreamName: "Subscribe", Handler: _SeaweedMessaging_Subscribe_Handler, ServerStreams: true, - ClientStreams: true, }, }, Metadata: "mq.proto", |