-
Notifications
You must be signed in to change notification settings - Fork 10
/
dial_client.go
35 lines (29 loc) · 928 Bytes
/
dial_client.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
package stream_api
import stream_api_rpc "github.com/aperturerobotics/bifrost/stream/api/rpc"
// DialStreamClientRPC fulfills stream RPC on the client side.
type DialStreamClientRPC struct {
SRPCStreamService_DialStreamClient
}
// NewDialStreamClientRPC builds a new DialStreamClientRPC.
func NewDialStreamClientRPC(
client SRPCStreamService_DialStreamClient,
) stream_api_rpc.RPC {
return &DialStreamClientRPC{
SRPCStreamService_DialStreamClient: client,
}
}
// Send sends a packet.
func (r *DialStreamClientRPC) Send(resp *stream_api_rpc.Data) error {
return r.SRPCStreamService_DialStreamClient.Send(
&DialStreamRequest{
Data: resp,
},
)
}
// Recv receives a packet.
func (r *DialStreamClientRPC) Recv() (*stream_api_rpc.Data, error) {
msg, err := r.SRPCStreamService_DialStreamClient.Recv()
return msg.GetData(), err
}
// _ is a type assertion
var _ stream_api_rpc.RPC = ((*DialStreamClientRPC)(nil))