add streaming response to helloworld service

This commit is contained in:
Asim Aslam
2021-06-11 12:17:54 +01:00
parent 7033fcfdb7
commit 1ae03a6011
4 changed files with 274 additions and 14 deletions

View File

@@ -43,6 +43,7 @@ func NewHelloworldEndpoints() []*api.Endpoint {
type HelloworldService interface {
Call(ctx context.Context, in *Request, opts ...client.CallOption) (*Response, error)
Stream(ctx context.Context, in *StreamRequest, opts ...client.CallOption) (Helloworld_StreamService, error)
}
type helloworldService struct {
@@ -67,15 +68,66 @@ func (c *helloworldService) Call(ctx context.Context, in *Request, opts ...clien
return out, nil
}
func (c *helloworldService) Stream(ctx context.Context, in *StreamRequest, opts ...client.CallOption) (Helloworld_StreamService, error) {
req := c.c.NewRequest(c.name, "Helloworld.Stream", &StreamRequest{})
stream, err := c.c.Stream(ctx, req, opts...)
if err != nil {
return nil, err
}
if err := stream.Send(in); err != nil {
return nil, err
}
return &helloworldServiceStream{stream}, nil
}
type Helloworld_StreamService interface {
Context() context.Context
SendMsg(interface{}) error
RecvMsg(interface{}) error
Close() error
Recv() (*StreamResponse, error)
}
type helloworldServiceStream struct {
stream client.Stream
}
func (x *helloworldServiceStream) Close() error {
return x.stream.Close()
}
func (x *helloworldServiceStream) Context() context.Context {
return x.stream.Context()
}
func (x *helloworldServiceStream) SendMsg(m interface{}) error {
return x.stream.Send(m)
}
func (x *helloworldServiceStream) RecvMsg(m interface{}) error {
return x.stream.Recv(m)
}
func (x *helloworldServiceStream) Recv() (*StreamResponse, error) {
m := new(StreamResponse)
err := x.stream.Recv(m)
if err != nil {
return nil, err
}
return m, nil
}
// Server API for Helloworld service
type HelloworldHandler interface {
Call(context.Context, *Request, *Response) error
Stream(context.Context, *StreamRequest, Helloworld_StreamStream) error
}
func RegisterHelloworldHandler(s server.Server, hdlr HelloworldHandler, opts ...server.HandlerOption) error {
type helloworld interface {
Call(ctx context.Context, in *Request, out *Response) error
Stream(ctx context.Context, stream server.Stream) error
}
type Helloworld struct {
helloworld
@@ -91,3 +143,43 @@ type helloworldHandler struct {
func (h *helloworldHandler) Call(ctx context.Context, in *Request, out *Response) error {
return h.HelloworldHandler.Call(ctx, in, out)
}
func (h *helloworldHandler) Stream(ctx context.Context, stream server.Stream) error {
m := new(StreamRequest)
if err := stream.Recv(m); err != nil {
return err
}
return h.HelloworldHandler.Stream(ctx, m, &helloworldStreamStream{stream})
}
type Helloworld_StreamStream interface {
Context() context.Context
SendMsg(interface{}) error
RecvMsg(interface{}) error
Close() error
Send(*StreamResponse) error
}
type helloworldStreamStream struct {
stream server.Stream
}
func (x *helloworldStreamStream) Close() error {
return x.stream.Close()
}
func (x *helloworldStreamStream) Context() context.Context {
return x.stream.Context()
}
func (x *helloworldStreamStream) SendMsg(m interface{}) error {
return x.stream.Send(m)
}
func (x *helloworldStreamStream) RecvMsg(m interface{}) error {
return x.stream.Recv(m)
}
func (x *helloworldStreamStream) Send(m *StreamResponse) error {
return x.stream.Send(m)
}