mirror of
https://github.com/kevin-DL/services.git
synced 2026-01-17 05:14:52 +00:00
add streaming response to helloworld service
This commit is contained in:
@@ -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)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user