Add package functions to define and set the rendezvous method

Add exported functions to the snowflake client library to allow calling
programs to define and set their own custom broker rendezvous methods.
This commit is contained in:
Cecylia Bocovich 2021-09-09 12:54:31 -04:00
parent 624750d5a8
commit 99887cd05d
5 changed files with 71 additions and 14 deletions

View file

@ -132,6 +132,10 @@ func (t *Transport) Dial() (net.Conn, error) {
return &SnowflakeConn{Stream: stream, sess: sess, pconn: pconn, snowflakes: snowflakes}, nil
}
func (t *Transport) SetRendezvousMethod(r RendezvousMethod) {
t.dialer.Rendezvous = r
}
type SnowflakeConn struct {
*smux.Stream
sess *smux.Session