-
Notifications
You must be signed in to change notification settings - Fork 4.1k
/
bufconn.go
38 lines (31 loc) · 1.09 KB
/
bufconn.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
36
37
38
package listenerutil
import (
"context"
"net"
"google.golang.org/grpc/test/bufconn"
)
const BufConnType = "bufconn"
// BufConnWrapper implements consul-template's TransportDialer using a
// bufconn listener, to provide a way to Dial the in-memory listener
type BufConnWrapper struct {
listener *bufconn.Listener
}
// NewBufConnWrapper returns a new BufConnWrapper using an
// existing bufconn.Listener
func NewBufConnWrapper(bcl *bufconn.Listener) *BufConnWrapper {
return &BufConnWrapper{
listener: bcl,
}
}
// Dial connects to the listening end of the bufconn (satisfies
// consul-template's TransportDialer interface). This is essentially the client
// side of the bufconn connection.
func (bcl *BufConnWrapper) Dial(_, _ string) (net.Conn, error) {
return bcl.listener.Dial()
}
// DialContext connects to the listening end of the bufconn (satisfies
// consul-template's TransportDialer interface). This is essentially the client
// side of the bufconn connection.
func (bcl *BufConnWrapper) DialContext(ctx context.Context, _, _ string) (net.Conn, error) {
return bcl.listener.DialContext(ctx)
}