micro/server/grpc/stream.go

41 lines
689 B
Go
Raw Normal View History

2019-06-03 20:44:43 +03:00
package grpc
import (
"context"
"github.com/micro/go-micro/v3/server"
2019-06-03 20:44:43 +03: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 20:44:43 +03: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 20:44:43 +03:00
}
func (r *rpcStream) Send(m interface{}) error {
return r.ServerStream.SendMsg(m)
2019-06-03 20:44:43 +03:00
}
func (r *rpcStream) Recv(m interface{}) error {
return r.ServerStream.RecvMsg(m)
2019-06-03 20:44:43 +03:00
}