完善服务器
This commit is contained in:
@@ -23,7 +23,7 @@ const _ = grpc.SupportPackageIsVersion7
|
||||
//
|
||||
// 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 GatewayClient interface {
|
||||
SendMessage(ctx context.Context, in *common.Empty, opts ...grpc.CallOption) (*common.Empty, error)
|
||||
ToClient(ctx context.Context, opts ...grpc.CallOption) (Gateway_ToClientClient, error)
|
||||
}
|
||||
|
||||
type gatewayClient struct {
|
||||
@@ -34,20 +34,45 @@ func NewGatewayClient(cc grpc.ClientConnInterface) GatewayClient {
|
||||
return &gatewayClient{cc}
|
||||
}
|
||||
|
||||
func (c *gatewayClient) SendMessage(ctx context.Context, in *common.Empty, opts ...grpc.CallOption) (*common.Empty, error) {
|
||||
out := new(common.Empty)
|
||||
err := c.cc.Invoke(ctx, "/Gateway/SendMessage", in, out, opts...)
|
||||
func (c *gatewayClient) ToClient(ctx context.Context, opts ...grpc.CallOption) (Gateway_ToClientClient, error) {
|
||||
stream, err := c.cc.NewStream(ctx, &Gateway_ServiceDesc.Streams[0], "/Gateway/ToClient", opts...)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
x := &gatewayToClientClient{stream}
|
||||
return x, nil
|
||||
}
|
||||
|
||||
type Gateway_ToClientClient interface {
|
||||
Send(*ToClientReq) error
|
||||
CloseAndRecv() (*common.Empty, error)
|
||||
grpc.ClientStream
|
||||
}
|
||||
|
||||
type gatewayToClientClient struct {
|
||||
grpc.ClientStream
|
||||
}
|
||||
|
||||
func (x *gatewayToClientClient) Send(m *ToClientReq) error {
|
||||
return x.ClientStream.SendMsg(m)
|
||||
}
|
||||
|
||||
func (x *gatewayToClientClient) CloseAndRecv() (*common.Empty, error) {
|
||||
if err := x.ClientStream.CloseSend(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
m := new(common.Empty)
|
||||
if err := x.ClientStream.RecvMsg(m); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return m, nil
|
||||
}
|
||||
|
||||
// GatewayServer is the server API for Gateway service.
|
||||
// All implementations must embed UnimplementedGatewayServer
|
||||
// for forward compatibility
|
||||
type GatewayServer interface {
|
||||
SendMessage(context.Context, *common.Empty) (*common.Empty, error)
|
||||
ToClient(Gateway_ToClientServer) error
|
||||
mustEmbedUnimplementedGatewayServer()
|
||||
}
|
||||
|
||||
@@ -55,8 +80,8 @@ type GatewayServer interface {
|
||||
type UnimplementedGatewayServer struct {
|
||||
}
|
||||
|
||||
func (UnimplementedGatewayServer) SendMessage(context.Context, *common.Empty) (*common.Empty, error) {
|
||||
return nil, status.Errorf(codes.Unimplemented, "method SendMessage not implemented")
|
||||
func (UnimplementedGatewayServer) ToClient(Gateway_ToClientServer) error {
|
||||
return status.Errorf(codes.Unimplemented, "method ToClient not implemented")
|
||||
}
|
||||
func (UnimplementedGatewayServer) mustEmbedUnimplementedGatewayServer() {}
|
||||
|
||||
@@ -71,22 +96,30 @@ func RegisterGatewayServer(s grpc.ServiceRegistrar, srv GatewayServer) {
|
||||
s.RegisterService(&Gateway_ServiceDesc, srv)
|
||||
}
|
||||
|
||||
func _Gateway_SendMessage_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
|
||||
in := new(common.Empty)
|
||||
if err := dec(in); err != nil {
|
||||
func _Gateway_ToClient_Handler(srv interface{}, stream grpc.ServerStream) error {
|
||||
return srv.(GatewayServer).ToClient(&gatewayToClientServer{stream})
|
||||
}
|
||||
|
||||
type Gateway_ToClientServer interface {
|
||||
SendAndClose(*common.Empty) error
|
||||
Recv() (*ToClientReq, error)
|
||||
grpc.ServerStream
|
||||
}
|
||||
|
||||
type gatewayToClientServer struct {
|
||||
grpc.ServerStream
|
||||
}
|
||||
|
||||
func (x *gatewayToClientServer) SendAndClose(m *common.Empty) error {
|
||||
return x.ServerStream.SendMsg(m)
|
||||
}
|
||||
|
||||
func (x *gatewayToClientServer) Recv() (*ToClientReq, error) {
|
||||
m := new(ToClientReq)
|
||||
if err := x.ServerStream.RecvMsg(m); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if interceptor == nil {
|
||||
return srv.(GatewayServer).SendMessage(ctx, in)
|
||||
}
|
||||
info := &grpc.UnaryServerInfo{
|
||||
Server: srv,
|
||||
FullMethod: "/Gateway/SendMessage",
|
||||
}
|
||||
handler := func(ctx context.Context, req interface{}) (interface{}, error) {
|
||||
return srv.(GatewayServer).SendMessage(ctx, req.(*common.Empty))
|
||||
}
|
||||
return interceptor(ctx, in, info, handler)
|
||||
return m, nil
|
||||
}
|
||||
|
||||
// Gateway_ServiceDesc is the grpc.ServiceDesc for Gateway service.
|
||||
@@ -95,12 +128,13 @@ func _Gateway_SendMessage_Handler(srv interface{}, ctx context.Context, dec func
|
||||
var Gateway_ServiceDesc = grpc.ServiceDesc{
|
||||
ServiceName: "Gateway",
|
||||
HandlerType: (*GatewayServer)(nil),
|
||||
Methods: []grpc.MethodDesc{
|
||||
Methods: []grpc.MethodDesc{},
|
||||
Streams: []grpc.StreamDesc{
|
||||
{
|
||||
MethodName: "SendMessage",
|
||||
Handler: _Gateway_SendMessage_Handler,
|
||||
StreamName: "ToClient",
|
||||
Handler: _Gateway_ToClient_Handler,
|
||||
ClientStreams: true,
|
||||
},
|
||||
},
|
||||
Streams: []grpc.StreamDesc{},
|
||||
Metadata: "grpc/gateway.proto",
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user