39 lines
896 B
Go
39 lines
896 B
Go
package client
|
|
|
|
import (
|
|
"net"
|
|
|
|
"github.com/admin-else/strom/proto"
|
|
"github.com/admin-else/strom/proto_base"
|
|
)
|
|
|
|
// Connect connects to the given address note that it does not resolve the SRV records it's just a raw net.Dial
|
|
func Connect(addr string) (ret *proto.Conn, err error) {
|
|
s, err := Status(addr)
|
|
if err != nil {
|
|
return
|
|
}
|
|
ret, err = ConnectVersionLess(addr)
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = ret.SetProtocolVersion(s.Version.Protocol)
|
|
return
|
|
}
|
|
|
|
// ConnectVersionLess connects to the given address without setting the version you should set it
|
|
// when using this function.
|
|
func ConnectVersionLess(addr string) (ret *proto.Conn, err error) {
|
|
ret = &proto.Conn{}
|
|
ret.SetState(proto_base.Handshaking)
|
|
ret.SetCompressionThreshold(-1)
|
|
ret.Actor = proto_base.Client
|
|
ret.Conn, err = net.Dial("tcp", addr)
|
|
if err != nil {
|
|
return
|
|
}
|
|
ret.R = ret.Conn
|
|
ret.W = ret.Conn
|
|
return
|
|
}
|