46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package grpc
|
|
|
|
// Picker picks a Conn for RPC requests.
|
|
// This is EXPERIMENTAL and Please do not implement your own Picker for now.
|
|
type Picker interface {
|
|
// Pick returns the Conn to use for the upcoming RPC. It may return different
|
|
// Conn's up to the implementation.
|
|
Pick() (*Conn, error)
|
|
// Peek returns the Conn use use for the next upcoming RPC. It returns the same
|
|
// Conn until next time Pick gets invoked.
|
|
Peek() *Conn
|
|
// Close closes all the Conn's owned by this Picker.
|
|
Close() error
|
|
}
|
|
|
|
func newSimplePicker(target string, dopts dialOptions) (Picker, error) {
|
|
c, err := NewConn(target, dopts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &simplePicker{
|
|
conn: c,
|
|
}, nil
|
|
}
|
|
|
|
// simplePicker is default Picker which is used when there is no custom Picker
|
|
// specified by users. It always picks the same Conn.
|
|
type simplePicker struct {
|
|
conn *Conn
|
|
}
|
|
|
|
func (p *simplePicker) Pick() (*Conn, error) {
|
|
return p.conn, nil
|
|
}
|
|
|
|
func (p *simplePicker) Peek() *Conn {
|
|
return p.conn
|
|
}
|
|
|
|
func (p *simplePicker) Close() error {
|
|
if p.conn != nil {
|
|
return p.conn.Close()
|
|
}
|
|
return nil
|
|
}
|