micro-server-grpc/stream.go

41 lines
693 B
Go
Raw Normal View History

2019-06-03 18:44:43 +01:00
package grpc
import (
"context"
"github.com/unistack-org/micro/v3/server"
2019-06-03 18:44:43 +01:00
"google.golang.org/grpc"
)
// rpcStream implements a server side Stream.
type rpcStream struct {
// embed the grpc stream so we can access it
grpc.ServerStream
2019-06-03 18:44:43 +01:00
request server.Request
}
func (r *rpcStream) Close() error {
return nil
}
func (r *rpcStream) Error() error {
return nil
}
func (r *rpcStream) Request() server.Request {
return r.request
}
func (r *rpcStream) Context() context.Context {
return r.ServerStream.Context()
2019-06-03 18:44:43 +01:00
}
func (r *rpcStream) Send(m interface{}) error {
return r.ServerStream.SendMsg(m)
2019-06-03 18:44:43 +01:00
}
func (r *rpcStream) Recv(m interface{}) error {
return r.ServerStream.RecvMsg(m)
2019-06-03 18:44:43 +01:00
}